如何基于列表中的值更新/创建熊猫中的列

发布于 2021-01-29 15:03:52

所以,这是我的数据框

import pandas as pd
cols = ['Name','Country','Income']
vals = [['Steve','USA',40000],['Matt','UK',40000],['John','USA',40000],['Martin','France',40000],]
x = pd.DataFrame(vals,columns=cols)

我还有另一个清单:

europe = ['UK','France']

如果x.Country位于欧洲,我想创建一个新列“ Continent”

关注者
0
被浏览
54
1 个回答
  • 面试哥
    面试哥 2021-01-29
    为面试而生,有面试问题,就找面试哥。

    或者您可以isin直接 使用

    x['New Column']='Not Europe'
    x.loc[x.Country.isin(europe),'New Column']='Europe'
    
    Out[612]: 
         Name Country  Income  New Column
    0   Steve     USA   40000  Not Europe
    1    Matt      UK   40000      Europe
    2    John     USA   40000  Not Europe
    3  Martin  France   40000      Europe
    


知识点
面圈网VIP题库

面圈网VIP题库全新上线,海量真题题库资源。 90大类考试,超10万份考试真题开放下载啦

去下载看看