{
"cells": [
{
"cell_type": "code",
"execution_count": 1,
"metadata": {},
"outputs": [],
"source": [
"import os\n",
"import pandas as pd\n",
"import numpy as np\n",
"import sys\n",
"import copy\n",
"sys.path.append('../../Scripts')\n",
"from preprocess_daily import DailyPreprocessor\n",
"from generate_signals import SignalGenerator"
]
},
{
"cell_type": "code",
"execution_count": 2,
"metadata": {},
"outputs": [
{
"name": "stdout",
"output_type": "stream",
"text": [
"1. 데이터를 준비합니다.\n",
"1-1. 네트워크가 로드되었습니다.\n",
"1-2. 테이블들이 로드되었습니다.\n",
"2. 신호이력 테이블을 변환합니다.\n",
"3. 이동류정보 테이블을 변환합니다.\n",
"4. 통합 테이블을 생성합니다.\n"
]
}
],
"source": [
"self = SignalGenerator()\n",
"self.prepare_data() # 1 \n",
"self.process_history() # 2\n",
"self.process_movement() # 3\n",
"self.make_histids() # 4\n",
"self.set_timepoints() # 5-1\n",
"self.assign_red_yellow() # 5-2"
]
},
{
"cell_type": "code",
"execution_count": 3,
"metadata": {},
"outputs": [
{
"name": "stdout",
"output_type": "stream",
"text": [
"['i0', 'i1', 'i2', 'i3', 'i6', 'i7', 'i8', 'i9']\n",
"['u00', 'u20', 'u30', 'u31', 'u32', 'u60']\n",
"['c30']\n"
]
}
],
"source": [
"print(self.parent_ids)\n",
"print(self.uturn_ids)\n",
"print(self.coord_ids)"
]
},
{
"cell_type": "code",
"execution_count": 4,
"metadata": {},
"outputs": [
{
"data": {
"text/html": [
"
\n",
"\n",
"
\n",
" \n",
" \n",
" | \n",
" node_id | \n",
" start_unix | \n",
" phase | \n",
" duration | \n",
" state | \n",
"
\n",
" \n",
" \n",
" \n",
" 0 | \n",
" i0 | \n",
" 1704418330 | \n",
" 1g_1g | \n",
" 39 | \n",
" gGGrgrrrgGGGGrgrr | \n",
"
\n",
" \n",
" 1 | \n",
" i0 | \n",
" 1704418330 | \n",
" 1y_1y | \n",
" 4 | \n",
" gyyrgrrrgyyyyrgrr | \n",
"
\n",
" \n",
" 2 | \n",
" i0 | \n",
" 1704418330 | \n",
" 2r_2r | \n",
" 1 | \n",
" grrrgrrrgrrrrrgrr | \n",
"
\n",
" \n",
" 3 | \n",
" i0 | \n",
" 1704418330 | \n",
" 2g_2g | \n",
" 40 | \n",
" grrGgrrrgrrrrGgrr | \n",
"
\n",
" \n",
" 4 | \n",
" i0 | \n",
" 1704418330 | \n",
" 2y_2y | \n",
" 4 | \n",
" grrygrrrgrrrrygrr | \n",
"
\n",
" \n",
" 5 | \n",
" i0 | \n",
" 1704418330 | \n",
" 3r_3r | \n",
" 1 | \n",
" grrrgGGrgrrrrrgrr | \n",
"
\n",
" \n",
" 6 | \n",
" i0 | \n",
" 1704418330 | \n",
" 3g_3g | \n",
" 28 | \n",
" grrrgGGGgrrrrrgrr | \n",
"
\n",
" \n",
" 7 | \n",
" i0 | \n",
" 1704418330 | \n",
" 3g_3y | \n",
" 4 | \n",
" grrrgGGygrrrrrgrr | \n",
"
\n",
" \n",
" 8 | \n",
" i0 | \n",
" 1704418330 | \n",
" 3g_4r | \n",
" 1 | \n",
" grrrgGGrgrrrrrgGr | \n",
"
\n",
" \n",
" 9 | \n",
" i0 | \n",
" 1704418330 | \n",
" 3g_4g | \n",
" 17 | \n",
" grrrgGGrgrrrrrgGr | \n",
"
\n",
" \n",
" 10 | \n",
" i0 | \n",
" 1704418330 | \n",
" 3y_4g | \n",
" 4 | \n",
" grrrgyyrgrrrrrgGr | \n",
"
\n",
" \n",
" 11 | \n",
" i0 | \n",
" 1704418330 | \n",
" 4r_4g | \n",
" 1 | \n",
" grrrgrrrgrrrrrgGr | \n",
"
\n",
" \n",
" 12 | \n",
" i0 | \n",
" 1704418330 | \n",
" 4g_4g | \n",
" 32 | \n",
" grrrgrrrgrrrrrgGG | \n",
"
\n",
" \n",
" 13 | \n",
" i0 | \n",
" 1704418330 | \n",
" 4y_4y | \n",
" 4 | \n",
" grrrgrrrgrrrrrgyy | \n",
"
\n",
" \n",
" 14 | \n",
" i0 | \n",
" 1704418510 | \n",
" 1r_1r | \n",
" 1 | \n",
" grrrgrrrgrrrrrgrr | \n",
"
\n",
" \n",
" 15 | \n",
" i0 | \n",
" 1704418510 | \n",
" 1g_1g | \n",
" 38 | \n",
" gGGrgrrrgGGGGrgrr | \n",
"
\n",
" \n",
" 16 | \n",
" i0 | \n",
" 1704418510 | \n",
" 1y_1y | \n",
" 4 | \n",
" gyyrgrrrgyyyyrgrr | \n",
"
\n",
" \n",
" 17 | \n",
" i0 | \n",
" 1704418510 | \n",
" 2r_2r | \n",
" 1 | \n",
" grrrgrrrgrrrrrgrr | \n",
"
\n",
" \n",
" 18 | \n",
" i0 | \n",
" 1704418510 | \n",
" 2g_2g | \n",
" 40 | \n",
" grrGgrrrgrrrrGgrr | \n",
"
\n",
" \n",
" 19 | \n",
" i0 | \n",
" 1704418510 | \n",
" 2y_2y | \n",
" 4 | \n",
" grrygrrrgrrrrygrr | \n",
"
\n",
" \n",
" 20 | \n",
" i0 | \n",
" 1704418510 | \n",
" 3r_3r | \n",
" 1 | \n",
" grrrgGGrgrrrrrgrr | \n",
"
\n",
" \n",
" 21 | \n",
" i0 | \n",
" 1704418510 | \n",
" 3g_3g | \n",
" 28 | \n",
" grrrgGGGgrrrrrgrr | \n",
"
\n",
" \n",
" 22 | \n",
" i0 | \n",
" 1704418510 | \n",
" 3g_3y | \n",
" 4 | \n",
" grrrgGGygrrrrrgrr | \n",
"
\n",
" \n",
" 23 | \n",
" i0 | \n",
" 1704418510 | \n",
" 3g_4r | \n",
" 1 | \n",
" grrrgGGrgrrrrrgGr | \n",
"
\n",
" \n",
" 24 | \n",
" i0 | \n",
" 1704418510 | \n",
" 3g_4g | \n",
" 17 | \n",
" grrrgGGrgrrrrrgGr | \n",
"
\n",
" \n",
" 25 | \n",
" i0 | \n",
" 1704418510 | \n",
" 3y_4g | \n",
" 4 | \n",
" grrrgyyrgrrrrrgGr | \n",
"
\n",
" \n",
" 26 | \n",
" i0 | \n",
" 1704418510 | \n",
" 4r_4g | \n",
" 1 | \n",
" grrrgrrrgrrrrrgGr | \n",
"
\n",
" \n",
" 27 | \n",
" i0 | \n",
" 1704418510 | \n",
" 4g_4g | \n",
" 32 | \n",
" grrrgrrrgrrrrrgGG | \n",
"
\n",
" \n",
" 28 | \n",
" i0 | \n",
" 1704418510 | \n",
" 4y_4y | \n",
" 4 | \n",
" grrrgrrrgrrrrrgyy | \n",
"
\n",
" \n",
" 29 | \n",
" i0 | \n",
" 1704418690 | \n",
" 1r_1r | \n",
" 1 | \n",
" grrrgrrrgrrrrrgrr | \n",
"
\n",
" \n",
" 30 | \n",
" i0 | \n",
" 1704418690 | \n",
" 1g_1g | \n",
" 38 | \n",
" gGGrgrrrgGGGGrgrr | \n",
"
\n",
" \n",
" 31 | \n",
" i0 | \n",
" 1704418690 | \n",
" 1y_1y | \n",
" 4 | \n",
" gyyrgrrrgyyyyrgrr | \n",
"
\n",
" \n",
" 32 | \n",
" i0 | \n",
" 1704418690 | \n",
" 2r_2r | \n",
" 1 | \n",
" grrrgrrrgrrrrrgrr | \n",
"
\n",
" \n",
" 33 | \n",
" i0 | \n",
" 1704418690 | \n",
" 2g_2g | \n",
" 40 | \n",
" grrGgrrrgrrrrGgrr | \n",
"
\n",
" \n",
" 34 | \n",
" i0 | \n",
" 1704418690 | \n",
" 2y_2y | \n",
" 4 | \n",
" grrygrrrgrrrrygrr | \n",
"
\n",
" \n",
" 35 | \n",
" i0 | \n",
" 1704418690 | \n",
" 3r_3r | \n",
" 1 | \n",
" grrrgGGrgrrrrrgrr | \n",
"
\n",
" \n",
" 36 | \n",
" i0 | \n",
" 1704418690 | \n",
" 3g_3g | \n",
" 28 | \n",
" grrrgGGGgrrrrrgrr | \n",
"
\n",
" \n",
" 37 | \n",
" i0 | \n",
" 1704418690 | \n",
" 3g_3y | \n",
" 22 | \n",
" grrrgGGygrrrrrgrr | \n",
"
\n",
" \n",
"
\n",
"
"
],
"text/plain": [
" node_id start_unix phase duration state\n",
"0 i0 1704418330 1g_1g 39 gGGrgrrrgGGGGrgrr\n",
"1 i0 1704418330 1y_1y 4 gyyrgrrrgyyyyrgrr\n",
"2 i0 1704418330 2r_2r 1 grrrgrrrgrrrrrgrr\n",
"3 i0 1704418330 2g_2g 40 grrGgrrrgrrrrGgrr\n",
"4 i0 1704418330 2y_2y 4 grrygrrrgrrrrygrr\n",
"5 i0 1704418330 3r_3r 1 grrrgGGrgrrrrrgrr\n",
"6 i0 1704418330 3g_3g 28 grrrgGGGgrrrrrgrr\n",
"7 i0 1704418330 3g_3y 4 grrrgGGygrrrrrgrr\n",
"8 i0 1704418330 3g_4r 1 grrrgGGrgrrrrrgGr\n",
"9 i0 1704418330 3g_4g 17 grrrgGGrgrrrrrgGr\n",
"10 i0 1704418330 3y_4g 4 grrrgyyrgrrrrrgGr\n",
"11 i0 1704418330 4r_4g 1 grrrgrrrgrrrrrgGr\n",
"12 i0 1704418330 4g_4g 32 grrrgrrrgrrrrrgGG\n",
"13 i0 1704418330 4y_4y 4 grrrgrrrgrrrrrgyy\n",
"14 i0 1704418510 1r_1r 1 grrrgrrrgrrrrrgrr\n",
"15 i0 1704418510 1g_1g 38 gGGrgrrrgGGGGrgrr\n",
"16 i0 1704418510 1y_1y 4 gyyrgrrrgyyyyrgrr\n",
"17 i0 1704418510 2r_2r 1 grrrgrrrgrrrrrgrr\n",
"18 i0 1704418510 2g_2g 40 grrGgrrrgrrrrGgrr\n",
"19 i0 1704418510 2y_2y 4 grrygrrrgrrrrygrr\n",
"20 i0 1704418510 3r_3r 1 grrrgGGrgrrrrrgrr\n",
"21 i0 1704418510 3g_3g 28 grrrgGGGgrrrrrgrr\n",
"22 i0 1704418510 3g_3y 4 grrrgGGygrrrrrgrr\n",
"23 i0 1704418510 3g_4r 1 grrrgGGrgrrrrrgGr\n",
"24 i0 1704418510 3g_4g 17 grrrgGGrgrrrrrgGr\n",
"25 i0 1704418510 3y_4g 4 grrrgyyrgrrrrrgGr\n",
"26 i0 1704418510 4r_4g 1 grrrgrrrgrrrrrgGr\n",
"27 i0 1704418510 4g_4g 32 grrrgrrrgrrrrrgGG\n",
"28 i0 1704418510 4y_4y 4 grrrgrrrgrrrrrgyy\n",
"29 i0 1704418690 1r_1r 1 grrrgrrrgrrrrrgrr\n",
"30 i0 1704418690 1g_1g 38 gGGrgrrrgGGGGrgrr\n",
"31 i0 1704418690 1y_1y 4 gyyrgrrrgyyyyrgrr\n",
"32 i0 1704418690 2r_2r 1 grrrgrrrgrrrrrgrr\n",
"33 i0 1704418690 2g_2g 40 grrGgrrrgrrrrGgrr\n",
"34 i0 1704418690 2y_2y 4 grrygrrrgrrrrygrr\n",
"35 i0 1704418690 3r_3r 1 grrrgGGrgrrrrrgrr\n",
"36 i0 1704418690 3g_3g 28 grrrgGGGgrrrrrgrr\n",
"37 i0 1704418690 3g_3y 22 grrrgGGygrrrrrgrr"
]
},
"execution_count": 4,
"metadata": {},
"output_type": "execute_result"
}
],
"source": [
"node_id = self.parent_ids[0]\n",
"self.SIGTABLE[self.SIGTABLE.node_id==node_id]"
]
},
{
"cell_type": "code",
"execution_count": 8,
"metadata": {},
"outputs": [
{
"name": "stdout",
"output_type": "stream",
"text": [
"1. 데이터를 준비합니다.\n",
"1-1. 네트워크가 로드되었습니다.\n",
"1-2. 테이블들이 로드되었습니다.\n",
"2. 신호이력 테이블을 변환합니다.\n",
"3. 이동류정보 테이블을 변환합니다.\n",
"4. 통합 테이블을 생성합니다.\n"
]
}
],
"source": [
"self = SignalGenerator()\n",
"self.prepare_data() # 1 \n",
"self.process_history() # 2\n",
"self.process_movement() # 3\n",
"self.make_histids() # 4\n",
"self.set_timepoints() # 5-1"
]
},
{
"cell_type": "code",
"execution_count": 14,
"metadata": {},
"outputs": [
{
"data": {
"text/html": [
"\n",
"\n",
"
\n",
" \n",
" \n",
" | \n",
" inter_no | \n",
" node_id | \n",
" start_unix | \n",
" phas_A | \n",
" phas_B | \n",
" move_A | \n",
" move_B | \n",
" duration | \n",
" state_A | \n",
" state_B | \n",
" phase_sumo | \n",
"
\n",
" \n",
" \n",
" \n",
" 86 | \n",
" 206 | \n",
" i7 | \n",
" 1704418700 | \n",
" 1 | \n",
" 1 | \n",
" 8 | \n",
" 4 | \n",
" 45 | \n",
" rrrggGG | \n",
" GGrggrr | \n",
" 0 | \n",
"
\n",
" \n",
" 87 | \n",
" 206 | \n",
" i7 | \n",
" 1704418700 | \n",
" 2 | \n",
" 2 | \n",
" 17 | \n",
" 18 | \n",
" 53 | \n",
" rrrggrr | \n",
" rrrggrr | \n",
" 1 | \n",
"
\n",
" \n",
" 88 | \n",
" 206 | \n",
" i7 | \n",
" 1704418700 | \n",
" 3 | \n",
" 3 | \n",
" 8 | \n",
" 4 | \n",
" 26 | \n",
" rrrggGG | \n",
" GGrggrr | \n",
" 2 | \n",
"
\n",
" \n",
" 89 | \n",
" 206 | \n",
" i7 | \n",
" 1704418700 | \n",
" 4 | \n",
" 4 | \n",
" 17 | \n",
" 18 | \n",
" 26 | \n",
" rrrggrr | \n",
" rrrggrr | \n",
" 3 | \n",
"
\n",
" \n",
" 90 | \n",
" 206 | \n",
" i7 | \n",
" 1704418850 | \n",
" 1 | \n",
" 1 | \n",
" 8 | \n",
" 4 | \n",
" 45 | \n",
" rrrggGG | \n",
" GGrggrr | \n",
" 0 | \n",
"
\n",
" \n",
" 91 | \n",
" 206 | \n",
" i7 | \n",
" 1704418850 | \n",
" 2 | \n",
" 2 | \n",
" 17 | \n",
" 18 | \n",
" 53 | \n",
" rrrggrr | \n",
" rrrggrr | \n",
" 1 | \n",
"
\n",
" \n",
" 92 | \n",
" 206 | \n",
" i7 | \n",
" 1704418850 | \n",
" 3 | \n",
" 3 | \n",
" 8 | \n",
" 4 | \n",
" 26 | \n",
" rrrggGG | \n",
" GGrggrr | \n",
" 2 | \n",
"
\n",
" \n",
" 93 | \n",
" 206 | \n",
" i7 | \n",
" 1704418850 | \n",
" 4 | \n",
" 4 | \n",
" 17 | \n",
" 18 | \n",
" 26 | \n",
" rrrggrr | \n",
" rrrggrr | \n",
" 3 | \n",
"
\n",
" \n",
" 94 | \n",
" 206 | \n",
" i7 | \n",
" 1704419000 | \n",
" 1 | \n",
" 1 | \n",
" 8 | \n",
" 4 | \n",
" 45 | \n",
" rrrggGG | \n",
" GGrggrr | \n",
" 0 | \n",
"
\n",
" \n",
" 95 | \n",
" 206 | \n",
" i7 | \n",
" 1704419000 | \n",
" 2 | \n",
" 2 | \n",
" 17 | \n",
" 18 | \n",
" 53 | \n",
" rrrggrr | \n",
" rrrggrr | \n",
" 1 | \n",
"
\n",
" \n",
" 96 | \n",
" 206 | \n",
" i7 | \n",
" 1704419000 | \n",
" 3 | \n",
" 3 | \n",
" 8 | \n",
" 4 | \n",
" 26 | \n",
" rrrggGG | \n",
" GGrggrr | \n",
" 2 | \n",
"
\n",
" \n",
" 97 | \n",
" 206 | \n",
" i7 | \n",
" 1704419000 | \n",
" 4 | \n",
" 4 | \n",
" 17 | \n",
" 18 | \n",
" 26 | \n",
" rrrggrr | \n",
" rrrggrr | \n",
" 3 | \n",
"
\n",
" \n",
" 98 | \n",
" 206 | \n",
" i7 | \n",
" 1704419150 | \n",
" 1 | \n",
" 1 | \n",
" 8 | \n",
" 4 | \n",
" 45 | \n",
" rrrggGG | \n",
" GGrggrr | \n",
" 0 | \n",
"
\n",
" \n",
" 99 | \n",
" 206 | \n",
" i7 | \n",
" 1704419150 | \n",
" 2 | \n",
" 2 | \n",
" 17 | \n",
" 18 | \n",
" 53 | \n",
" rrrggrr | \n",
" rrrggrr | \n",
" 1 | \n",
"
\n",
" \n",
" 100 | \n",
" 206 | \n",
" i7 | \n",
" 1704419150 | \n",
" 3 | \n",
" 3 | \n",
" 8 | \n",
" 4 | \n",
" 26 | \n",
" rrrggGG | \n",
" GGrggrr | \n",
" 2 | \n",
"
\n",
" \n",
" 101 | \n",
" 206 | \n",
" i7 | \n",
" 1704419150 | \n",
" 4 | \n",
" 4 | \n",
" 17 | \n",
" 18 | \n",
" 26 | \n",
" rrrggrr | \n",
" rrrggrr | \n",
" 3 | \n",
"
\n",
" \n",
"
\n",
"
"
],
"text/plain": [
" inter_no node_id start_unix phas_A phas_B move_A move_B duration \\\n",
"86 206 i7 1704418700 1 1 8 4 45 \n",
"87 206 i7 1704418700 2 2 17 18 53 \n",
"88 206 i7 1704418700 3 3 8 4 26 \n",
"89 206 i7 1704418700 4 4 17 18 26 \n",
"90 206 i7 1704418850 1 1 8 4 45 \n",
"91 206 i7 1704418850 2 2 17 18 53 \n",
"92 206 i7 1704418850 3 3 8 4 26 \n",
"93 206 i7 1704418850 4 4 17 18 26 \n",
"94 206 i7 1704419000 1 1 8 4 45 \n",
"95 206 i7 1704419000 2 2 17 18 53 \n",
"96 206 i7 1704419000 3 3 8 4 26 \n",
"97 206 i7 1704419000 4 4 17 18 26 \n",
"98 206 i7 1704419150 1 1 8 4 45 \n",
"99 206 i7 1704419150 2 2 17 18 53 \n",
"100 206 i7 1704419150 3 3 8 4 26 \n",
"101 206 i7 1704419150 4 4 17 18 26 \n",
"\n",
" state_A state_B phase_sumo \n",
"86 rrrggGG GGrggrr 0 \n",
"87 rrrggrr rrrggrr 1 \n",
"88 rrrggGG GGrggrr 2 \n",
"89 rrrggrr rrrggrr 3 \n",
"90 rrrggGG GGrggrr 0 \n",
"91 rrrggrr rrrggrr 1 \n",
"92 rrrggGG GGrggrr 2 \n",
"93 rrrggrr rrrggrr 3 \n",
"94 rrrggGG GGrggrr 0 \n",
"95 rrrggrr rrrggrr 1 \n",
"96 rrrggGG GGrggrr 2 \n",
"97 rrrggrr rrrggrr 3 \n",
"98 rrrggGG GGrggrr 0 \n",
"99 rrrggrr rrrggrr 1 \n",
"100 rrrggGG GGrggrr 2 \n",
"101 rrrggrr rrrggrr 3 "
]
},
"metadata": {},
"output_type": "display_data"
},
{
"data": {
"text/html": [
"\n",
"\n",
"
\n",
" \n",
" \n",
" | \n",
" node_id | \n",
" start_unix | \n",
" phase | \n",
" duration | \n",
" state | \n",
"
\n",
" \n",
" \n",
" \n",
" 0 | \n",
" i7 | \n",
" 1704418700 | \n",
" 1g_1g | \n",
" 41 | \n",
" GGrggGG | \n",
"
\n",
" \n",
" 1 | \n",
" i7 | \n",
" 1704418700 | \n",
" 1y_1y | \n",
" 4 | \n",
" yyrggyy | \n",
"
\n",
" \n",
" 2 | \n",
" i7 | \n",
" 1704418700 | \n",
" 2r_2r | \n",
" 0 | \n",
" rrrggrr | \n",
"
\n",
" \n",
" 3 | \n",
" i7 | \n",
" 1704418700 | \n",
" 2r_2g | \n",
" 0 | \n",
" rrrggrr | \n",
"
\n",
" \n",
" 4 | \n",
" i7 | \n",
" 1704418700 | \n",
" 2g_2r | \n",
" 0 | \n",
" rrrggrr | \n",
"
\n",
" \n",
" ... | \n",
" ... | \n",
" ... | \n",
" ... | \n",
" ... | \n",
" ... | \n",
"
\n",
" \n",
" 66 | \n",
" i7 | \n",
" 1704419150 | \n",
" 2y_2y | \n",
" 4 | \n",
" rrrggrr | \n",
"
\n",
" \n",
" 67 | \n",
" i7 | \n",
" 1704419150 | \n",
" 3g_3r | \n",
" 0 | \n",
" rrrggGG | \n",
"
\n",
" \n",
" 68 | \n",
" i7 | \n",
" 1704419150 | \n",
" 3g_3g | \n",
" 0 | \n",
" GGrggGG | \n",
"
\n",
" \n",
" 69 | \n",
" i7 | \n",
" 1704419150 | \n",
" 3r_3r | \n",
" 0 | \n",
" rrrggrr | \n",
"
\n",
" \n",
" 70 | \n",
" i7 | \n",
" 1704419150 | \n",
" 3r_3g | \n",
" 22 | \n",
" GGrggrr | \n",
"
\n",
" \n",
"
\n",
"
71 rows × 5 columns
\n",
"
"
],
"text/plain": [
" node_id start_unix phase duration state\n",
"0 i7 1704418700 1g_1g 41 GGrggGG\n",
"1 i7 1704418700 1y_1y 4 yyrggyy\n",
"2 i7 1704418700 2r_2r 0 rrrggrr\n",
"3 i7 1704418700 2r_2g 0 rrrggrr\n",
"4 i7 1704418700 2g_2r 0 rrrggrr\n",
".. ... ... ... ... ...\n",
"66 i7 1704419150 2y_2y 4 rrrggrr\n",
"67 i7 1704419150 3g_3r 0 rrrggGG\n",
"68 i7 1704419150 3g_3g 0 GGrggGG\n",
"69 i7 1704419150 3r_3r 0 rrrggrr\n",
"70 i7 1704419150 3r_3g 22 GGrggrr\n",
"\n",
"[71 rows x 5 columns]"
]
},
"metadata": {},
"output_type": "display_data"
}
],
"source": [
"NODE_ID = 'i7'\n",
"sigtable = self.sigtable[self.sigtable.node_id==NODE_ID]\n",
"display(sigtable)\n",
"\n",
"self.SIGTABLE = []\n",
"for node_id in self.node_ids:\n",
" sig = self.sigtable.query('node_id==@node_id')\n",
" for i, row in sig.iterrows():\n",
" inter_no = row.inter_no\n",
" phas_A = row.phas_A\n",
" phas_B = row.phas_B\n",
" start_unix = row.start_unix\n",
" prow = self.load_prow(inter_no, start_unix)[1].iloc[0]\n",
" red_A = prow[f'red_A{phas_A}']\n",
" yellow_A = prow[f'yellow_A{phas_A}']\n",
" red_B = prow[f'red_B{phas_B}']\n",
" yellow_B = prow[f'yellow_B{phas_B}']\n",
" sig.loc[i, ['red_A', 'red_B', 'yellow_A', 'yellow_B']] = red_A, red_B, yellow_A, yellow_B\n",
" sig = sig.astype({'red_A': int, 'red_B': int, 'yellow_A': int, 'yellow_B': int,\n",
" 'phas_A':str, 'phas_B':str})\n",
" sig = sig.drop(['move_A','move_B'], axis=1)\n",
"\n",
" sig_A = sig[['start_unix', 'phas_A', 'duration', 'state_A', 'red_A', 'yellow_A']].reset_index(drop=True)\n",
" sig_B = sig[['start_unix', 'phas_B', 'duration', 'state_B', 'red_B', 'yellow_B']].reset_index(drop=True)\n",
"\n",
" csig_A = self.cumulate(sig_A, 'A')\n",
" csig_B = self.cumulate(sig_B, 'B')\n",
"\n",
" SIG = pd.merge(csig_A, csig_B, on=['start_time', 'start_unix'], how='outer')\n",
" SIG = SIG.sort_values(by='start_time').reset_index(drop=True)\n",
" SIG[['phas_A', 'state_A']] = SIG[['phas_A', 'state_A']].fillna(method='ffill')\n",
" SIG[['phas_B', 'state_B']] = SIG[['phas_B', 'state_B']].fillna(method='ffill')\n",
" SIG['phase'] = SIG['phas_A'] + \"_\" + SIG['phas_B']\n",
" SIG['node_id'] = node_id\n",
" SIG = SIG[['node_id', 'start_unix', 'start_time', 'phase', 'state_A', 'state_B']]\n",
" SIG['duration'] = SIG['start_time'].shift(-1) - SIG['start_time']\n",
" SIG = SIG[:-1]\n",
" SIG['duration'] = SIG['duration'].astype(int)\n",
" for row in SIG.itertuples():\n",
" state = ''\n",
" for a, b, in zip(row.state_A, row.state_B):\n",
" if a == 'r':\n",
" state += b\n",
" elif b == 'r':\n",
" state += a\n",
" elif a == b:\n",
" state += a\n",
" else:\n",
" raise ValueError(f\"예상되지 않은 조합 발생: a={a}, b={b}\")\n",
" SIG.at[row.Index, 'state'] = state\n",
" SIG = SIG.drop(columns=['start_time', 'state_A', 'state_B'])\n",
" self.SIGTABLE.append(SIG)\n",
"self.SIGTABLE = pd.concat(self.SIGTABLE)\n",
"\n",
"SIGTABLE = self.SIGTABLE[self.SIGTABLE.node_id==NODE_ID]\n",
"display(SIGTABLE)"
]
},
{
"cell_type": "code",
"execution_count": 47,
"metadata": {},
"outputs": [
{
"data": {
"text/html": [
"\n",
"\n",
"
\n",
" \n",
" \n",
" | \n",
" start_unix | \n",
" phas_A | \n",
" duration | \n",
" state_A | \n",
" red_A | \n",
" yellow_A | \n",
"
\n",
" \n",
" \n",
" \n",
" 0 | \n",
" 1704418700 | \n",
" 1 | \n",
" 45 | \n",
" rrrggGG | \n",
" 0 | \n",
" 4 | \n",
"
\n",
" \n",
" 1 | \n",
" 1704418700 | \n",
" 2 | \n",
" 53 | \n",
" rrrggrr | \n",
" 0 | \n",
" 4 | \n",
"
\n",
" \n",
" 2 | \n",
" 1704418700 | \n",
" 3 | \n",
" 26 | \n",
" rrrggGG | \n",
" 0 | \n",
" 4 | \n",
"
\n",
" \n",
" 3 | \n",
" 1704418700 | \n",
" 4 | \n",
" 26 | \n",
" rrrggrr | \n",
" 0 | \n",
" 4 | \n",
"
\n",
" \n",
" 4 | \n",
" 1704418850 | \n",
" 1 | \n",
" 45 | \n",
" rrrggGG | \n",
" 0 | \n",
" 4 | \n",
"
\n",
" \n",
"
\n",
"
"
],
"text/plain": [
" start_unix phas_A duration state_A red_A yellow_A\n",
"0 1704418700 1 45 rrrggGG 0 4\n",
"1 1704418700 2 53 rrrggrr 0 4\n",
"2 1704418700 3 26 rrrggGG 0 4\n",
"3 1704418700 4 26 rrrggrr 0 4\n",
"4 1704418850 1 45 rrrggGG 0 4"
]
},
"metadata": {},
"output_type": "display_data"
},
{
"data": {
"text/html": [
"\n",
"\n",
"
\n",
" \n",
" \n",
" | \n",
" start_unix | \n",
" phas_B | \n",
" duration | \n",
" state_B | \n",
" red_B | \n",
" yellow_B | \n",
"
\n",
" \n",
" \n",
" \n",
" 0 | \n",
" 1704418700 | \n",
" 1 | \n",
" 45 | \n",
" GGrggrr | \n",
" 0 | \n",
" 4 | \n",
"
\n",
" \n",
" 1 | \n",
" 1704418700 | \n",
" 2 | \n",
" 53 | \n",
" rrrggrr | \n",
" 0 | \n",
" 4 | \n",
"
\n",
" \n",
" 2 | \n",
" 1704418700 | \n",
" 3 | \n",
" 26 | \n",
" GGrggrr | \n",
" 0 | \n",
" 4 | \n",
"
\n",
" \n",
" 3 | \n",
" 1704418700 | \n",
" 4 | \n",
" 26 | \n",
" rrrggrr | \n",
" 0 | \n",
" 4 | \n",
"
\n",
" \n",
" 4 | \n",
" 1704418850 | \n",
" 1 | \n",
" 45 | \n",
" GGrggrr | \n",
" 0 | \n",
" 4 | \n",
"
\n",
" \n",
"
\n",
"
"
],
"text/plain": [
" start_unix phas_B duration state_B red_B yellow_B\n",
"0 1704418700 1 45 GGrggrr 0 4\n",
"1 1704418700 2 53 rrrggrr 0 4\n",
"2 1704418700 3 26 GGrggrr 0 4\n",
"3 1704418700 4 26 rrrggrr 0 4\n",
"4 1704418850 1 45 GGrggrr 0 4"
]
},
"metadata": {},
"output_type": "display_data"
},
{
"data": {
"text/html": [
"\n",
"\n",
"
\n",
" \n",
" \n",
" | \n",
" start_time | \n",
" phas_A | \n",
" state_A | \n",
" start_unix | \n",
"
\n",
" \n",
" \n",
" \n",
" 0 | \n",
" 0 | \n",
" 1g | \n",
" rrrggGG | \n",
" 1704418700 | \n",
"
\n",
" \n",
" 1 | \n",
" 41 | \n",
" 1y | \n",
" rrrggyy | \n",
" 1704418700 | \n",
"
\n",
" \n",
" 2 | \n",
" 45 | \n",
" 2r | \n",
" rrrggrr | \n",
" 1704418700 | \n",
"
\n",
" \n",
" 3 | \n",
" 45 | \n",
" 2g | \n",
" rrrggrr | \n",
" 1704418700 | \n",
"
\n",
" \n",
" 4 | \n",
" 94 | \n",
" 2y | \n",
" rrrggrr | \n",
" 1704418700 | \n",
"
\n",
" \n",
"
\n",
"
"
],
"text/plain": [
" start_time phas_A state_A start_unix\n",
"0 0 1g rrrggGG 1704418700\n",
"1 41 1y rrrggyy 1704418700\n",
"2 45 2r rrrggrr 1704418700\n",
"3 45 2g rrrggrr 1704418700\n",
"4 94 2y rrrggrr 1704418700"
]
},
"metadata": {},
"output_type": "display_data"
},
{
"data": {
"text/html": [
"\n",
"\n",
"
\n",
" \n",
" \n",
" | \n",
" start_time | \n",
" phas_B | \n",
" state_B | \n",
" start_unix | \n",
"
\n",
" \n",
" \n",
" \n",
" 0 | \n",
" 0 | \n",
" 1g | \n",
" GGrggrr | \n",
" 1704418700 | \n",
"
\n",
" \n",
" 1 | \n",
" 41 | \n",
" 1y | \n",
" yyrggrr | \n",
" 1704418700 | \n",
"
\n",
" \n",
" 2 | \n",
" 45 | \n",
" 2r | \n",
" rrrggrr | \n",
" 1704418700 | \n",
"
\n",
" \n",
" 3 | \n",
" 45 | \n",
" 2g | \n",
" rrrggrr | \n",
" 1704418700 | \n",
"
\n",
" \n",
" 4 | \n",
" 94 | \n",
" 2y | \n",
" rrrggrr | \n",
" 1704418700 | \n",
"
\n",
" \n",
"
\n",
"
"
],
"text/plain": [
" start_time phas_B state_B start_unix\n",
"0 0 1g GGrggrr 1704418700\n",
"1 41 1y yyrggrr 1704418700\n",
"2 45 2r rrrggrr 1704418700\n",
"3 45 2g rrrggrr 1704418700\n",
"4 94 2y rrrggrr 1704418700"
]
},
"metadata": {},
"output_type": "display_data"
},
{
"data": {
"text/html": [
"\n",
"\n",
"
\n",
" \n",
" \n",
" | \n",
" node_id | \n",
" start_unix | \n",
" phase | \n",
" duration | \n",
" state | \n",
"
\n",
" \n",
" \n",
" \n",
" 0 | \n",
" u60 | \n",
" 1704418700 | \n",
" 1g_1g | \n",
" 41 | \n",
" GGrggGG | \n",
"
\n",
" \n",
" 1 | \n",
" u60 | \n",
" 1704418700 | \n",
" 1y_1y | \n",
" 4 | \n",
" yyrggyy | \n",
"
\n",
" \n",
" 2 | \n",
" u60 | \n",
" 1704418700 | \n",
" 2r_2r | \n",
" 0 | \n",
" rrrggrr | \n",
"
\n",
" \n",
" 3 | \n",
" u60 | \n",
" 1704418700 | \n",
" 2r_2g | \n",
" 0 | \n",
" rrrggrr | \n",
"
\n",
" \n",
" 4 | \n",
" u60 | \n",
" 1704418700 | \n",
" 2g_2r | \n",
" 0 | \n",
" rrrggrr | \n",
"
\n",
" \n",
"
\n",
"
"
],
"text/plain": [
" node_id start_unix phase duration state\n",
"0 u60 1704418700 1g_1g 41 GGrggGG\n",
"1 u60 1704418700 1y_1y 4 yyrggyy\n",
"2 u60 1704418700 2r_2r 0 rrrggrr\n",
"3 u60 1704418700 2r_2g 0 rrrggrr\n",
"4 u60 1704418700 2g_2r 0 rrrggrr"
]
},
"metadata": {},
"output_type": "display_data"
}
],
"source": [
"NODE_ID = 'i7'\n",
"sig = self.sigtable[self.sigtable.node_id==NODE_ID]\n",
"# display(sig.head())\n",
"\n",
"for i, row in sig.iterrows():\n",
" inter_no = row.inter_no\n",
" phas_A = row.phas_A\n",
" phas_B = row.phas_B\n",
" start_unix = row.start_unix\n",
" prow = self.load_prow(inter_no, start_unix)[1].iloc[0]\n",
" red_A = prow[f'red_A{phas_A}']\n",
" yellow_A = prow[f'yellow_A{phas_A}']\n",
" red_B = prow[f'red_B{phas_B}']\n",
" yellow_B = prow[f'yellow_B{phas_B}']\n",
" sig.loc[i, ['red_A', 'red_B', 'yellow_A', 'yellow_B']] = red_A, red_B, yellow_A, yellow_B\n",
"\n",
"sig = sig.astype({'red_A': int, 'red_B': int, 'yellow_A': int, 'yellow_B': int,\n",
" 'phas_A':str, 'phas_B':str})\n",
"sig = sig.drop(['move_A','move_B'], axis=1)\n",
"\n",
"# display(sig.head())\n",
"sig_A = sig[['start_unix', 'phas_A', 'duration', 'state_A', 'red_A', 'yellow_A']].reset_index(drop=True)\n",
"sig_B = sig[['start_unix', 'phas_B', 'duration', 'state_B', 'red_B', 'yellow_B']].reset_index(drop=True)\n",
"\n",
"display(sig_A.head())\n",
"display(sig_B.head())\n",
"\n",
"csig_A = self.cumulate(sig_A, 'A')\n",
"csig_B = self.cumulate(sig_B, 'B')\n",
"\n",
"display(csig_A.head())\n",
"display(csig_B.head())\n",
"SIG = pd.merge(csig_A, csig_B, on=['start_time', 'start_unix'], how='outer')\n",
"SIG = SIG.sort_values(by='start_time').reset_index(drop=True)\n",
"SIG[['phas_A', 'state_A']] = SIG[['phas_A', 'state_A']].fillna(method='ffill')\n",
"SIG[['phas_B', 'state_B']] = SIG[['phas_B', 'state_B']].fillna(method='ffill')\n",
"SIG['phase'] = SIG['phas_A'] + \"_\" + SIG['phas_B']\n",
"SIG['node_id'] = node_id\n",
"SIG = SIG[['node_id', 'start_unix', 'start_time', 'phase', 'state_A', 'state_B']]\n",
"SIG['duration'] = SIG['start_time'].shift(-1) - SIG['start_time']\n",
"SIG = SIG[:-1]\n",
"SIG['duration'] = SIG['duration'].astype(int)\n",
"for row in SIG.itertuples():\n",
" state = ''\n",
" for a, b, in zip(row.state_A, row.state_B):\n",
" if a == 'r':\n",
" state += b\n",
" elif b == 'r':\n",
" state += a\n",
" elif a == b:\n",
" state += a\n",
" else:\n",
" raise ValueError(f\"예상되지 않은 조합 발생: a={a}, b={b}\")\n",
" SIG.at[row.Index, 'state'] = state\n",
"SIG = SIG.drop(columns=['start_time', 'state_A', 'state_B'])\n",
"\n",
"display(SIG.head())"
]
},
{
"cell_type": "code",
"execution_count": 50,
"metadata": {},
"outputs": [
{
"data": {
"text/html": [
"\n",
"\n",
"
\n",
" \n",
" \n",
" | \n",
" node_id | \n",
" start_unix | \n",
" phase | \n",
" duration | \n",
" state | \n",
"
\n",
" \n",
" \n",
" \n",
" 0 | \n",
" u60 | \n",
" 1704418700 | \n",
" 1g_1g | \n",
" 41 | \n",
" GGrggGG | \n",
"
\n",
" \n",
" 1 | \n",
" u60 | \n",
" 1704418700 | \n",
" 1y_1y | \n",
" 4 | \n",
" yyrggyy | \n",
"
\n",
" \n",
" 2 | \n",
" u60 | \n",
" 1704418700 | \n",
" 2r_2r | \n",
" 0 | \n",
" rrrggrr | \n",
"
\n",
" \n",
" 3 | \n",
" u60 | \n",
" 1704418700 | \n",
" 2r_2g | \n",
" 0 | \n",
" rrrggrr | \n",
"
\n",
" \n",
" 4 | \n",
" u60 | \n",
" 1704418700 | \n",
" 2g_2r | \n",
" 0 | \n",
" rrrggrr | \n",
"
\n",
" \n",
" 5 | \n",
" u60 | \n",
" 1704418700 | \n",
" 2g_2g | \n",
" 49 | \n",
" rrrggrr | \n",
"
\n",
" \n",
" 6 | \n",
" u60 | \n",
" 1704418700 | \n",
" 2y_2y | \n",
" 4 | \n",
" rrrggrr | \n",
"
\n",
" \n",
" 7 | \n",
" u60 | \n",
" 1704418700 | \n",
" 3r_3r | \n",
" 0 | \n",
" rrrggrr | \n",
"
\n",
" \n",
" 8 | \n",
" u60 | \n",
" 1704418700 | \n",
" 3r_3g | \n",
" 0 | \n",
" GGrggrr | \n",
"
\n",
" \n",
" 9 | \n",
" u60 | \n",
" 1704418700 | \n",
" 3g_3r | \n",
" 0 | \n",
" rrrggGG | \n",
"
\n",
" \n",
"
\n",
"
"
],
"text/plain": [
" node_id start_unix phase duration state\n",
"0 u60 1704418700 1g_1g 41 GGrggGG\n",
"1 u60 1704418700 1y_1y 4 yyrggyy\n",
"2 u60 1704418700 2r_2r 0 rrrggrr\n",
"3 u60 1704418700 2r_2g 0 rrrggrr\n",
"4 u60 1704418700 2g_2r 0 rrrggrr\n",
"5 u60 1704418700 2g_2g 49 rrrggrr\n",
"6 u60 1704418700 2y_2y 4 rrrggrr\n",
"7 u60 1704418700 3r_3r 0 rrrggrr\n",
"8 u60 1704418700 3r_3g 0 GGrggrr\n",
"9 u60 1704418700 3g_3r 0 rrrggGG"
]
},
"metadata": {},
"output_type": "display_data"
},
{
"data": {
"text/html": [
"\n",
"\n",
"
\n",
" \n",
" \n",
" | \n",
" node_id | \n",
" start_unix | \n",
" phase | \n",
" duration | \n",
" state | \n",
"
\n",
" \n",
" \n",
" \n",
" 0 | \n",
" u60 | \n",
" 1704418700 | \n",
" 1g_1g | \n",
" 41 | \n",
" GGrggGG | \n",
"
\n",
" \n",
" 1 | \n",
" u60 | \n",
" 1704418700 | \n",
" 1y_1y | \n",
" 4 | \n",
" yyrggyy | \n",
"
\n",
" \n",
" 5 | \n",
" u60 | \n",
" 1704418700 | \n",
" 2g_2g | \n",
" 49 | \n",
" rrrggrr | \n",
"
\n",
" \n",
" 6 | \n",
" u60 | \n",
" 1704418700 | \n",
" 2y_2y | \n",
" 4 | \n",
" rrrggrr | \n",
"
\n",
" \n",
" 10 | \n",
" u60 | \n",
" 1704418700 | \n",
" 3g_3g | \n",
" 22 | \n",
" GGrggGG | \n",
"
\n",
" \n",
" 11 | \n",
" u60 | \n",
" 1704418700 | \n",
" 3y_3y | \n",
" 4 | \n",
" yyrggyy | \n",
"
\n",
" \n",
" 15 | \n",
" u60 | \n",
" 1704418700 | \n",
" 4g_4g | \n",
" 22 | \n",
" rrrggrr | \n",
"
\n",
" \n",
" 16 | \n",
" u60 | \n",
" 1704418700 | \n",
" 4y_4y | \n",
" 4 | \n",
" rrrggrr | \n",
"
\n",
" \n",
" 20 | \n",
" u60 | \n",
" 1704418850 | \n",
" 1g_1g | \n",
" 41 | \n",
" GGrggGG | \n",
"
\n",
" \n",
" 21 | \n",
" u60 | \n",
" 1704418850 | \n",
" 1y_1y | \n",
" 4 | \n",
" yyrggyy | \n",
"
\n",
" \n",
"
\n",
"
"
],
"text/plain": [
" node_id start_unix phase duration state\n",
"0 u60 1704418700 1g_1g 41 GGrggGG\n",
"1 u60 1704418700 1y_1y 4 yyrggyy\n",
"5 u60 1704418700 2g_2g 49 rrrggrr\n",
"6 u60 1704418700 2y_2y 4 rrrggrr\n",
"10 u60 1704418700 3g_3g 22 GGrggGG\n",
"11 u60 1704418700 3y_3y 4 yyrggyy\n",
"15 u60 1704418700 4g_4g 22 rrrggrr\n",
"16 u60 1704418700 4y_4y 4 rrrggrr\n",
"20 u60 1704418850 1g_1g 41 GGrggGG\n",
"21 u60 1704418850 1y_1y 4 yyrggyy"
]
},
"metadata": {},
"output_type": "display_data"
}
],
"source": [
"display(SIG.head(10))\n",
"display(SIG[SIG.duration!=0].head(10))"
]
},
{
"cell_type": "code",
"execution_count": 53,
"metadata": {},
"outputs": [],
"source": [
"reds = [f'red_{alph}{i}' for alph in ['A','B'] for i in range(1,9)]\n",
"yels = [f'yellow_{alph}{i}' for alph in ['A','B'] for i in range(1,9)]"
]
},
{
"cell_type": "code",
"execution_count": 62,
"metadata": {},
"outputs": [
{
"data": {
"text/html": [
"\n",
"\n",
"
\n",
" \n",
" \n",
" | \n",
" inter_no | \n",
" start_hour | \n",
" start_minute | \n",
" dura_A1 | \n",
" dura_A2 | \n",
" dura_A3 | \n",
" dura_A4 | \n",
" dura_A5 | \n",
" dura_A6 | \n",
" dura_A7 | \n",
" ... | \n",
" red_A4 | \n",
" red_B4 | \n",
" red_A5 | \n",
" red_B5 | \n",
" red_A6 | \n",
" red_B6 | \n",
" red_A7 | \n",
" red_B7 | \n",
" red_A8 | \n",
" red_B8 | \n",
"
\n",
" \n",
" \n",
" \n",
" 24 | \n",
" 206 | \n",
" 0 | \n",
" 0 | \n",
" 33 | \n",
" 35 | \n",
" 26 | \n",
" 26 | \n",
" 0 | \n",
" 0 | \n",
" 0 | \n",
" ... | \n",
" 0 | \n",
" 0 | \n",
" 0 | \n",
" 0 | \n",
" 0 | \n",
" 0 | \n",
" 0 | \n",
" 0 | \n",
" 0 | \n",
" 0 | \n",
"
\n",
" \n",
" 25 | \n",
" 206 | \n",
" 7 | \n",
" 0 | \n",
" 44 | \n",
" 44 | \n",
" 26 | \n",
" 26 | \n",
" 0 | \n",
" 0 | \n",
" 0 | \n",
" ... | \n",
" 0 | \n",
" 0 | \n",
" 0 | \n",
" 0 | \n",
" 0 | \n",
" 0 | \n",
" 0 | \n",
" 0 | \n",
" 0 | \n",
" 0 | \n",
"
\n",
" \n",
" 26 | \n",
" 206 | \n",
" 9 | \n",
" 0 | \n",
" 45 | \n",
" 53 | \n",
" 26 | \n",
" 26 | \n",
" 0 | \n",
" 0 | \n",
" 0 | \n",
" ... | \n",
" 0 | \n",
" 0 | \n",
" 0 | \n",
" 0 | \n",
" 0 | \n",
" 0 | \n",
" 0 | \n",
" 0 | \n",
" 0 | \n",
" 0 | \n",
"
\n",
" \n",
" 27 | \n",
" 206 | \n",
" 18 | \n",
" 30 | \n",
" 46 | \n",
" 62 | \n",
" 26 | \n",
" 26 | \n",
" 0 | \n",
" 0 | \n",
" 0 | \n",
" ... | \n",
" 0 | \n",
" 0 | \n",
" 0 | \n",
" 0 | \n",
" 0 | \n",
" 0 | \n",
" 0 | \n",
" 0 | \n",
" 0 | \n",
" 0 | \n",
"
\n",
" \n",
"
\n",
"
4 rows × 53 columns
\n",
"
"
],
"text/plain": [
" inter_no start_hour start_minute dura_A1 dura_A2 dura_A3 dura_A4 \\\n",
"24 206 0 0 33 35 26 26 \n",
"25 206 7 0 44 44 26 26 \n",
"26 206 9 0 45 53 26 26 \n",
"27 206 18 30 46 62 26 26 \n",
"\n",
" dura_A5 dura_A6 dura_A7 ... red_A4 red_B4 red_A5 red_B5 red_A6 \\\n",
"24 0 0 0 ... 0 0 0 0 0 \n",
"25 0 0 0 ... 0 0 0 0 0 \n",
"26 0 0 0 ... 0 0 0 0 0 \n",
"27 0 0 0 ... 0 0 0 0 0 \n",
"\n",
" red_B6 red_A7 red_B7 red_A8 red_B8 \n",
"24 0 0 0 0 0 \n",
"25 0 0 0 0 0 \n",
"26 0 0 0 0 0 \n",
"27 0 0 0 0 0 \n",
"\n",
"[4 rows x 53 columns]"
]
},
"execution_count": 62,
"metadata": {},
"output_type": "execute_result"
}
],
"source": [
"zeros = (self.plan[reds + yels]==0)\n",
"self.plan[zeros.any(axis=1)]"
]
}
],
"metadata": {
"kernelspec": {
"display_name": "sts",
"language": "python",
"name": "sts"
},
"language_info": {
"codemirror_mode": {
"name": "ipython",
"version": 3
},
"file_extension": ".py",
"mimetype": "text/x-python",
"name": "python",
"nbconvert_exporter": "python",
"pygments_lexer": "ipython3",
"version": "3.8.10"
}
},
"nbformat": 4,
"nbformat_minor": 2
}