基于集合中的成员资格替换熊猫数据框中的值

Question

我想替换 pandas 数据框中的值（如果它们包含在集合中）。这是一个小示例：

lset = frozenset((0,1))
data = {'col1': [3, 2, 1, 0]}
df = pd.DataFrame.from_dict(data)

我希望将 lset 中包含的 col1 的所有值替换为 -5。结果相当于：

data = {'col1': [3, 2, -5, -5]}
df = pd.DataFrame.from_dict(data)

我尝试了

df.loc[df['col1'] in lset, 'col1'] = -5

，但收到了 TypeError: unhashable type: 'Series'

Answer 1

使用ISIN检查

df.loc[df['col1'].isin(lset), 'col1'] = -5
df
Out[180]: 
   col1
0     3
1     2
2    -5
3    -5

Answer 2

在条件 np.where 中使用 isin

df['col2']=np.where(df['col1'].isin(lset),-5,df['col1'])



   col1  col2
0     3     3
1     2     2
2     1    -5
3     0    -5