Untitled
unknown
plain_text
5 months ago
1.4 kB
7
Indexable
def make_dict(row): data = dict() for seconds, vehicle_id, event_id, ac_state, dc_state, train_speed in zip( row["seconds_to_incident_sequence"], row["vehicles_sequence"], row["events_sequence"], row["dj_ac_state_sequence"], row["dj_dc_state_sequence"], row["train_kph_sequence"], ): if seconds not in data.keys(): data[seconds] = { vehicle_id: { event_id: { "train_speed": train_speed, "ac_state": ac_state, "dc_state": dc_state, } } } elif vehicle_id not in data[seconds].keys(): data[seconds] = { vehicle_id: { event_id: { "train_speed": train_speed, "ac_state": ac_state, "dc_state": dc_state, } } } elif event_id not in data[seconds][vehicle_id].keys(): data[seconds][vehicle_id] = { event_id: { "train_speed": train_speed, "ac_state": ac_state, "dc_state": dc_state, } } else: print("problem") return data
Editor is loading...
Leave a Comment