Pandas pivot_table 按值分组列

Question

我正在尝试将数值用作 Pandas pivot_table 上的列。问题在于，由于每个数字大多是唯一的，因此生成的 pivot_table 作为聚合数据的方式并不是很有用。

这是我目前所拥有的（假数据示例）：

import pandas as pd   

df = pd.DataFrame({'Country': ['US', 'Brazil', 'France', 'Germany'], 
                       'Continent': ['Americas', 'Americas', 'Europe', 'Europe'], 
                       'Population': [321, 207, 80, 66]})


pd.pivot_table(df, index='Continent', columns='Population', aggfunc='count')

这是结果的图像 .

如何根据列将值分组到范围内？

换句话说，我如何计算所有人口... 300 的国家/地区？

Answer 1

使用 pd.cut：

df = df.assign(PopGroup=pd.cut(df.Population,bins=[0,100,200,300,np.inf],labels=['<100','100-200','200-300','>300']))

输出：

  Continent  Country  Population PopGroup
0  Americas       US         321     >300
1  Americas   Brazil         207  200-300
2    Europe   France          80     <100
3    Europe  Germany          66     <100

pd.pivot_table(df, index='Continent', columns='PopGroup',values=['Country'], aggfunc='count')

输出：

        Country          
PopGroup  200-300 <100 >300
Continent                  
Americas      1.0  NaN  1.0
Europe        NaN  2.0  NaN