python - Pandas 将 df.count() 结果的最后 n 行求和为一行

Question

我正在寻找一种方法来生成数据框的漂亮摘要统计信息。考虑以下示例:

>> df = pd.DataFrame({"category":['u','v','w','u','y','z','y','z','x','x','y','z','x','z','x']})

>> df['category'].value_counts()
z    4
x    4
y    3
u    2
v    1
w    1

>> ??
           count  pct
z              4  27%
x              4  27%
y              3  20%
Other (3)      4  27%

结果将最后 n=3 行的值计数相加，删除它们，然后将它们作为一行添加到原始值计数中。将所有内容都作为百分比也很好。任何想法如何实现这个？干杯!

Answer 1

对于具有百分比的 DataFrame，请使用 Series.iloc通过索引，通过 Series.to_frame 创建 DataFrame , 添加由百分比填充的新行和新列:

s = df['category'].value_counts()

n= 3
out = s.iloc[:-n].to_frame('count')
out.loc['Other ({n})'] = s.iloc[-n:].sum()
out['pct'] = out['count'].div(out['count'].sum()).apply(lambda x: f"{x:.0%}")
print (out)
           count  pct
z              4  27%
x              4  27%
y              3  20%
Other (3)      4  27%