python - 如果条件不适用于 Titanic Kaggle 数据集

Question

我将这个泰坦尼克号数据集用作来自 Kaggle 的 titanic_df，我在其中创建了一个新列 titanic_df['person'] 并输入值作为子项，如果 passenger 是低于 16 岁或乘客的性别(如果他/她高于 16 岁)。但是 if 条件未被检查并且 ['person'] 列获得乘客的性别为它的值(value)观。

这是我做的:

titanic_df= pd.read_csv('train.csv')

for age in titanic_df['Age']:

        if age < 16 :

          titanic_df['person']= 'child' 

        if age > 16 :

          titanic_df['person'] = titanic_df['Sex']

print titanic_df[:12]

Answer 1

我将回应@fuglede 所说的关于在每次迭代时通过行分配整个列的内容。但是，下面是使用 numpy.where 完成任务的矢量化方法。

借用@fuglede的示例数据

df['Person'] = np.where(df.Age.values < 16, 'Child', df.Sex.values)
print(df)

      Sex  Age  Person
0    Male   15   Child
1  Female   20  Female
2    Male   50    Male