Question

我想在数据框中创建一个有条件地填充值的列。基本上我的数据框就像这样

  Origin     X
0 Guatemala  x
1 China      x
2 Kenya      x
3 Venezuela  x
4 Bangladesh x

我现在要做的是创建一个额外的列“大陆”，它会根据国家/地区添加大陆。我的结果看起来像这样：

 Origin      X  Continent
0 Guatemala  x  South america
1 China      x  Asia
2 Kenya      x  Africa
3 Venezuela  x  South america
4 Bangladesh x  Asia

我尝试了以下代码来满足我的需求：

def GetContinents(x):
    if x['Origin']== 'Thailand' or 'Indonesia' or 'China' or 'Japan' or 'Bangladesh':
        return 'Asia'
    elif x['Origin']== 'Boliva' or 'Guatemala' or 'Venezuela' or 'Mexico' or 'Argentinia':
        return 'South America'
    elif x['Origin']== 'Guinea Bissau' or 'Egypt' or 'Zaire' or 'Kenya':
        return 'Africa'
    else:
        return 'unknown'

df['Continent']= df.apply(GetContinents, axis=1)

这个神秘地用“亚洲”填充“大陆”的所有栏目。

df['Continent'] = np.where(df['Origin'] == 'Bangladesh', 'Asia', 'unknown')

这个可以正常工作，它将“亚洲”填入正确的列，而其他所有人都不知道，但当我尝试制作像df['Continent'] = np.where(df['Origin'] == 'Bangladesh' or 'China', 'Asia', 'unknown')这样的东西时，我得到了一个错误。

基本上我的问题是：如何用不同的值填充我的if条件？

Answer 1

您可以为每个大陆构建列表，并apply func：

In [35]:
asia = ['Thailand','Indonesia','China','Japan','Bangladesh']
south_america = ['Boliva' , 'Guatemala' , 'Venezuela' , 'Mexico' , 'Argentinia']
africa = [ 'Guinea Bissau' , 'Egypt' , 'Zaire' , 'Kenya']
def find_continent(x):
    if x in asia:
        return 'Asia'
    elif x in south_america:
        return 'South America'
    elif x in africa:
        return 'Africa'
    else:
        return 'Unknown'
df['Continent'] = df['Origin'].apply(find_continent)
df

Out[35]:
       Origin  X      Continent
0   Guatemala  x  South America
1       China  x           Asia
2       Kenya  x         Africa
3   Venezuela  x  South America
4  Bangladesh  x           Asia

或者如果您有更大的df，那么您可以使用isin进行连续调用并使用loc屏蔽行：

In [38]:
df.loc[df['Origin'].isin(asia),'Continent'] = 'Asia'
df.loc[df['Origin'].isin(south_america),'Continent'] = 'South America'
df.loc[df['Origin'].isin(africa),'Continent'] = 'Africa'
df['Continent'] = df['Continent'].fillna('Unknown')
df

Out[38]:
       Origin  X      Continent
0   Guatemala  x  South America
1       China  x           Asia
2       Kenya  x         Africa
3   Venezuela  x  South America
4  Bangladesh  x           Asia

至于为什么你的尝试不起作用：

if x['Origin']== 'Thailand' or 'Indonesia' or 'China' or 'Japan' or 'Bangladesh'

这会返回True，因为or 'Indonesia'始终为True所以所有行都设置为亚洲。

你应该把它改成这样：

if x['Origin'] in ('Thailand' , 'Indonesia' , 'China' , 'Japan' , 'Bangladesh'):

参见相关内容：How do I test one variable against multiple values?

使用np.where会很好，但是你没有屏蔽行，所以你不断覆盖行，所以只有最后一个操作仍然存在。

如何将条件列添加到pandas df

1 个答案: