在Python中從csv打印年度流行電影

Question

我有一個Excel格式的電影數據集合。 它具有年份，標題，受歡迎程度的列。 我的目標是創建一個年度流行電影的數據框架。 目前，我只能創建年份和人氣等級。 我也想添加電影標題。 有任何想法嗎？

df=pd.DataFrame(data)
xd=data.groupby(['release_year']).max()['popularity']
xf=pd.DataFrame(xd)
xd.head(100)

輸出是這樣的

1960     2.610362
1961     2.631987
1962     3.170651

我也想要電影名稱。 提前感謝:)

Answer 1

您只需要轉換索引。

假設這是您的數據：

release_year, popularity, movie
1999, 5, a
1999, 4, c
2000, 3, b
2000, 4, d

請執行下列操作：

import pandas as pd
data= pd.read_csv('data.csv')
idx = data.groupby(['release_year'])['popularity'].transform(max) == data['popularity']

data['popularity']為：

release_year popularity movie
1999  5 a
2000 4 b