微信公众号搜"智元新知"关注
微信扫一扫可直接关注哦!

Python pandas – 如果项目在列表中,则为新列的值

我想在pandas dataframe中创建一个新列.第一列包含国家/地区的名称.该列表包含我感兴趣的国家(例如,在欧盟).新列应指示数据框中的国家/地区是否在列表中.

以下是代码的缩短版本:

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.

谢谢

解决方法:

使用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

版权声明:本文内容由互联网用户自发贡献,该文观点与技术仅代表作者本人。本站仅提供信息存储空间服务,不拥有所有权,不承担相关法律责任。如发现本站有涉嫌侵权/违法违规的内容, 请发送邮件至 [email protected] 举报,一经查实,本站将立刻删除。

相关推荐