熊貓透視數據框

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