Python Pandas-如果项目在列表中，则为新列的值

Question

我想在pandas数据框中创建一个新列。第一列包含国家名称。该列表包含我感兴趣的国家（例如欧盟）。新列应指出数据框中的国家/地区是否在列表中。

下面是该代码的简化版本：

import pandas as pd
import numpy as np

EU = ["Austria","Belgium","Germany"]

df1 = pd.DataFrame(data={"Country":["USA","Germany","Russia","Poland"], "Capital":["Washington","Berlin","Moscow","Warsaw"]})

df1["EU"] = np.where(df1["Country"] in EU, "EU", "Other")

我得到的错误是：ValueError：系列的真值不明确。使用a.empty，a.bool（），a.item（），a.any（）或a.all（）

我不知道问题是什么以及如何解决。我想念什么？我在Windows上使用Anaconda。

谢谢

Answer 1

使用isin的检查成员：

df1["EU"] = np.where(df1["Country"].isin(EU), "EU", "Other")
print (df1)
      Capital  Country     EU
0  Washington      USA  Other
1      Berlin  Germany     EU
2      Moscow   Russia  Other
3      Warsaw   Poland     EU