如何替换不在列表中的 Pandas Dataframe 中的所有值？ [复制]

Question

我有一个值列表。如何替换不在给定值列表中的 Dataframe 列中的所有值？

例如，

>>> df = pd.DataFrame(['D','ND','D','garbage'], columns=['S'])
>>> df
      S
0    D
1    ND
2    D
3  garbage

>>> allowed_vals = ['D','ND']

我想用“无”替换数据框 S 列中不在列表 allowed_vals 中的所有值。我该怎么做？

Answer 1

您可以使用isin 来检查allowed_list 中的成员资格，~ 来否定它，然后.loc 来修改系列：

>>> df.loc[~df["S"].isin(allowed_vals), "S"] = "None"
>>> df
      S
0     D
1    ND
2     D
3  None

因为

>>> df["S"].isin(allowed_vals)
0     True
1     True
2     True
3    False
Name: S, dtype: bool

如果你想修改整个框架（不仅仅是S列），你可以制作一个框架大小的蒙版：

>>> df
         S   T
0        D   D
1       ND   A
2        D  ND
3  garbage   A
>>> df[~df.isin(allowed_vals)] = "None"
>>> df
      S     T
0     D     D
1    ND  None
2     D    ND
3  None  None