python - 为什么我不能将数组分配给 pandas DataFrame 列？

Question

我正在使用 KNeighborsClassifier 对一些值进行分类，如下所示:

arr = classifier_3NN.predict(testData_df)
len(arr)
10960

我想将这个数组分配给 DataFrame 中的一个列，我已经检查过并且它的大小相同:

len(df[df['events']=='NaN']['events'])
10960

When I do the following command the array values are not in the column as expected:

df[df['events']=='NaN']['events'] = arr

谁能看出我做错了什么？

Answer 1

我想你需要isnull用 ix 检查 NaN 值:

df.ix[df['events'].isnull(), 'events'] = arr

但如果需要用 arr 替换 NaN 值，最好使用 fillna通过 Series 从 arr 创建:

arr = [5,7,9]

df = pd.DataFrame({'events':[np.nan, 1, np.nan]})
print (df)
   events
0     NaN
1     1.0
2     NaN

df['events'] = df['events'].fillna(pd.Series(arr, index=df.index))
print (df)
   events
0     5.0
1     1.0
2     9.0