熊猫透视数据框

Question

我有一个数据框架，如下所示，我想进行数据透视，以便可以得出每年排名前三的国家的GDP。 因此，我需要以一种数据透视的方式进行数据透视，以使我每年能获得3个GDP最高的3个国家

有任何想法吗？

 Year  Country   GDP
 01    USA      100
 01    UK        80
 01    Japan     50
 01    China     75
 02    USA       90
 02    UK        65
 02    Japan     70
 02    China     80

Answer 1

你想要这样的东西吗？

In [19]: df
Out[19]: 
   Year Country  GDP
0     1     USA  100
1     1      UK   80
2     1   Japan   50
3     1   China   75
4     2     USA   90
5     2      UK   65
6     2   Japan   70
7     2   China   80

In [20]: df.sort_values(['Year', 'GDP'], ascending=[True, False]).groupby('Year').head(3)
Out[20]: 
   Year Country  GDP
0     1     USA  100
1     1      UK   80
3     1   China   75
4     2     USA   90
7     2   China   80
6     2   Japan   70