Pandas pivot_table按值列出组列

时间:2017-06-07 18:06:30

标签: python pandas pivot-table

我正在尝试将数值用作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')

这是结果的图像 pivot_table

如何根据我的列将我的值分组到范围内?

换句话说,我如何计算所有人口国家...< 100,100-200,> 300?

1 个答案:

答案 0 :(得分:4)

使用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