python-3.x - 通过np.select的 "default"参数输入空值

Question

尝试将值写入给定特定条件的列，默认为 Null 值，代码如下:

import pandas as pd
import numpy as np
df = pd.DataFrame({'col': list('ABCDE')})

cond1 = df['col'].eq('A')
cond2 = df['col'].isin(['B', 'E'])

df['new_col'] = np.select([cond1, cond2], ['foo', 'bar'], default=np.NaN)

但它在列中将 'nan' 作为字符串值。

df['new_col'].unique()
#array(['foo', 'bar', 'nan'], dtype=object)

有没有办法从这段代码中直接将其更改为 null？

Answer 1

找到了正确的解决方案，它使用None作为默认值:

df['new_col'] = np.select([cond1, cond2], ['foo', 'bar'], default=None)