基于多个条件改变Python中列的值

Question

我试图使用基于“或”条件的Pandas在Python中插入一个新列，但我在代码上遇到了问题。我想做的是：

如果“监管机构”一栏说FDIC、美联储或财政部，那么我想要一个新的专栏“备案”说“是”，否则，说“不是”。这就是我写的。我的数据是df200。

    df200["Filing"] = np.where(df200["Regulatory Body"]=="FDIC","Yes","No")

在这段代码中是否有一个“或”条件来适应其他两个变量？

谢谢!

Answer 1

是。使用pd.Series.isin

bodies = {'FDIC', 'Fed', 'Treasury'}

df200['Filing'] = np.where(df200['Regulatory Body'].isin(bodies), 'Yes', 'No')

或者，将pd.Series.map与从pd.Series.isin接收的布尔数组一起使用。

df200['Filing'] = df200['Regulatory Body'].isin(bodies).map({True: 'Yes', False: 'No'})