Помогите разобраться с DataFrame на питоне

В меня есть набор данных, где хранятся данные мужчин и женщин. В задаче не было дано, какое описание признака пола соответствует gender=1 и gender=2. Пол можно определить по параметру роста, по разумным меркам мужчины всегда выше женщин.

Моё решение этой проблемы:

import numpy as np
import pandas as pd

df = pd.read_csv('mlbootcamp5_train.csv',sep=';',index_col='id')
print('Інформація для пяти перших користувачів: ')
print(df[['age', 'gender', 'height']].head())

mean1 = df.loc[df['gender']==1, 'height'].mean()
print("Середній зріст осіб за першою гендерною ознакою: {0:.2f}".format(mean1))
mean2 = df.loc[df['gender']==2, 'height'].mean()
print("Середній зріст осіб за другою гендерною ознакою: {0:.2f}".format(mean2))

print('Кількість всіх осіб: ', len(df))
print('Кількість осіб за першою гендерною ознакою: ', len(df.loc[df['gender']==1]))
print('Кількість осіб за другою гендерною ознакою: ', len(df.loc[df['gender']==2]))

max_height = df.loc[df['gender']==1, 'height'].max()
print('Найвижчий зріст для першої гендерної ознаки: ', max_height)
min_height = df.loc[df['gender']==1, 'height'].min()
print('Найменший зріст для першої гендерної ознаки: ', min_height)

max_height2 = df.loc[df['gender']==2, 'height'].max()
print('Найвижчий зріст для другої гендерної ознаки: ', max_height2)
min_height2 = df.loc[df['gender']==2, 'height'].min()
print('Найменший зріст для другої генедерної ознаки: ', min_height2)

Мне кажется, что я неправильно определил максимальный и минимальный рост для каждого пола. По результатам вышло, что 1 - женщины, а 2 - мужчины. Не знаю насколько в меня правильное решение, можно ли как-то по другому это решить?


Ответы (0 шт):