为 groupby 中的独特结果生成饼图子图?

0 python matplotlib pie-chart

为数据框中的唯一值生成饼图的最佳方法是什么?

我有一个数据框,显示按县划分的服务数量。我想为每个县制作一组饼图,显示该县的服务数量。我尝试了多种不同的方法,但没有取得太大成功。

这是我的数据

打印(MDF组)

    County     Service    ServiceCt
0   Alamance    Literacy          1
1   Alamance   Technical          1
2   Alamance  Vocational          4
3    Chatham    Literacy          3
4    Chatham   Technical          2
5    Chatham  Vocational          1
6     Durham    Literacy          1
7     Durham   Technical          1
8     Durham  Vocational          1
9     Orange    Literacy          1
10      Wake    Literacy          2
11      Wake   Technical          2
Run Code Online (Sandbox Code Playgroud)

因此,阿拉曼斯将会有一张图表,其中包含识字、技术、职业的切片;Chatham、Durham 等的图表。切片大小将基于 ServiceCt。

我一直在尝试很多不同的方法,但我不确定最有效的方法是什么。我尝试过,但下面并没有真正按县进行细分,也没有生成任何图表。

for i, row in enumerate(mdfgroup.itertuples(),1):
       plt.figure()
       plt.pie(row.ServiceCt,labels=row.Service,
       startangle=90,frame=True, explode=0.2,radius=3)
plt.show()
Run Code Online (Sandbox Code Playgroud)

这会引发错误:

TypeError: len() of unsized object
Run Code Online (Sandbox Code Playgroud)

然后产生一个空白绘图框

(我还不能嵌入图像,所以这是链接) 空白绘图框

理想情况下,我希望它们都是次要情节,但在这个阶段我会采取一系列单独的情节。我发现的其他示例不处理键(县)的唯一值。

Imp*_*est 5

使用 matplotlib

一种常见的方法是迭代列的groupby。这里要迭代的列是"Country". 您可以首先创建一个子图网格,其中子图的数量至少与您拥有的独特国家/地区一样多。然后您可以同时迭代子图和组。
最后可能会有一些空的子图;这些可以设置为不可见。

import numpy as np
import pandas as pd
import matplotlib.pyplot as plt

df = pd.DataFrame({"Country" : list("AAACCCDDDOWW"),
                   "Service" : list("LTV")*4,
                   "ServiceCt" : list(map(int, "114321111122"))})

cols = 3
g = df.groupby("Country")
rows = int(np.ceil(len(g)/cols))

fig, axes = plt.subplots(ncols=cols, nrows=rows)

for (c, grp), ax in zip(g, axes.flat):
    ax.pie(grp.ServiceCt, labels=grp.Service)
    ax.set_title(c)

if len(g) < cols*rows:    
    for ax in axes.flatten()[len(g):]:
        ax.axis("off")
        
plt.show()
Run Code Online (Sandbox Code Playgroud)

在此输入图像描述

使用seaborn

这个案例实际上非常适合与seaborn 的FacetGrid.

import pandas as pd
import matplotlib.pyplot as plt
import seaborn as sns

df = pd.DataFrame({"Country" : list("AAACCCDDDOWW"),
                   "Service" : list("LTV")*4,
                   "ServiceCt" : list(map(int, "114321111122"))})

def pie(v, l, color=None):
    plt.pie(v, labels=l.values)
g = sns.FacetGrid(df, col="Country")
g.map(pie, "ServiceCt", "Service" )
        
plt.show()
Run Code Online (Sandbox Code Playgroud)

在此输入图像描述

使用熊猫

最后,我们可以使用 pandas 在一行中完成所有操作。

import pandas as pd
import matplotlib.pyplot as plt

df = pd.DataFrame({"Country" : list("AAACCCDDDOWW"),
                   "Service" : list("LTV")*4,
                   "ServiceCt" : list(map(int, "114321111122"))})


df.pivot("Service", "Country", "ServiceCt").plot.pie(subplots=True, legend=False)
        
plt.show()
Run Code Online (Sandbox Code Playgroud)

在此输入图像描述