如何获取数据帧列值并替换？

Question

id  name   country
1   AAA    Afghanistan
2   BBB    Australia

这是我的数据帧↑

如何让国家变得↓

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

我想从国家栏中得到3个单词的值并添加到新的国家栏中，怎么办？你知道吗

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])