Я пытаюсь написать условный оператор для полей из импортированного CSV (data_dict), однако мой текущий код с использованием np.where, похоже, не работает. Я пытаюсь определить возраст (data_dict ['Age']) людей в зависимости от того, мужчина это или женщина, от (data_dict ['Gender']). Как бы я подошел к решению этого? Пожалуйста, смотрите мой код ниже. Большое спасибо.
Пример индекса данных, возраст, год, Crash_Month, Crash_Day, Crash_Time, Road_User, Gender, Crash_Type, Injury_Severity, Crash_LGA, Crash_Area_Type 1,37,2000,1,1,4: 30: 59, PEDESTRIAN, MALE, UNKNOWN, 1, MARIBYRNONG, MELBOURNE 2,22,2000,1,1,0: 07: 35, ВОДИТЕЛЬ, MALE, ADJACENT DIRECTION, 1, YARRA, MELBOURNE 3,47,2000,1,1,4 : 51: 37, ДРАЙВЕР, ЖЕНЩИНА, НАПРАВЛЕНИЕ НАПРАВЛЕНИЯ, 0, ЯРРА, МЕЛЬБУРН 4,70,2000,1,1,4: 27: 56, ДРАЙВЕР, МУЖЧИНА, НАПРАВЛЕНИЕ НАПРАВЛЕНИЯ, 1, БАНЬЮЛЬ, МЕЛЬБУРН
Ожидаемый результат Возраст мужчин: [37,22,70, ...] Возраст женщин: [47, ...]
Текущий результат Возраст мужчин: [] Возраст женщин: []
gender1 = np.array(data_dict['Gender'])
age1 = np.array(data_dict['Age'])
age_females = age1[np.where(gender1 == 'Female')]
age_males = age1[np.where(gender1 == 'Male')]