熊貓為另一列分配不同的值，取決於另一列中的值

Question

我有以下df ，

days
NaN
70
29

我喜歡做

df['short_days'] = np.where(df.days < 30, 'Yes', 'No')

但是當它為NaN ， df.short_days = 'Not Applicable' ，因此結果將如下所示：

days    short_days
NaN     Not Applicable
70      No
29      Yes

Answer 1

只需執行fillna即可用您想要的任何東西填充nan值

import numpy as np

df['short_days'] = np.where(df.days < 30, 'Yes', 'No')
df['short_days'].fillna('Not Applicable',inplace=True)

days    short_days
NaN     Not Applicable
70      No
29      Yes

Answer 2

使用np.select更優雅，可以在此處定義default值：

df['short_days'] = np.select([df.days<30, df.days>=30], 
                             ['Yes', 'No'], 
                             default='Not Applicable')

輸出量

   days      short_days
0   NaN  Not Applicable
1  70.0              No
2  29.0             Yes