python - 如何计算两年之间的百分比变化并插入 Pandas 中的新数据帧？

Question

我有一个巨大的数据框，看起来像这样:

year    country     population  
1971    Afghanistan 11500000    
1972    Afghanistan 11800000    
1973    Afghanistan 12100000    
1974    Afghanistan 12400000    
1975    Afghanistan 12700000    

我想创建一个新的 DataFrame 来计算每十年按国家分组的人口百分比差异

country      1971-1980   1981-1990 1991-2000 2001-2010
Afghanistan  --          --        --        --
Australia    --          --        --        --

需要一些帮助来了解如何做到这一点。任何帮助，将不胜感激。

Answer 1

您可以创建十年列，然后使用 DataFrame.pivot_table 与 sum并添加 DataFrame.pct_change :

d = df['year'] // 10 * 10
df['dec'] = (d + 1).astype(str) + '-' + (d + 10).astype(str)

另一个想法 cut :

bins = range(df['year'].min(), df['year'].max() + 10, 10)
labels = [f'{i}-{j-1}' for i, j in zip(bins[:-1], bins[1:])] 

df['dec'] = pd.cut(df.year, bins=bins, labels=labels, include_lowest=True)

df1 = (df.pivot_table(index='country', 
                    columns='dec', 
                    values='population', 
                    aggfunc='sum')
        .pct_change(axis=1))