如何获取dataframe列值并替换？

Question

id  name   country
1   AAA    Afghanistan
2   BBB    Australia

这是我的数据框↑

如何让国家成为↓

id  name   country       Newcountry
1   AAA    Afghanistan   Afg
2   BBB    Australia     Aus

我想从国家/地区列值中获取3个单词并添加到Newcountry列，该怎么办？

Answer 1

你可以：

In [7]: df['Newcountry'] = df['country'].str[:3]

In [8]: df
Out[8]: 
   id name      country Newcountry
0   1  AAA  Afghanistan        Afg
1   2  BBB    Australia        Aus

Answer 2

您可以使用apply()。

df['Newcountry'] = df['country'].apply(lambda x:x[:3])

或您可以使用属性访问。

df['Newcountry'] = df1.country.apply(lambda x:x[:3])