python - 如何在 Python 的 Matplotlib 中绘制嵌套饼图？

Question

我有一个关于在 Python 中的 Matplotlib 中绘制嵌套饼图的问题。我写了一些代码来处理这个过程，但我有一个与设计和标签有关的问题

我想画一种这种嵌套的饼图。（从嵌套的最上层到最里面是SEX，ALIGN用覆盖它们的计数）

这是我的数据框，如下所示。

ALIGN   SEX count
2   Bad Characters  Male Characters 1542
5   Good Characters Male Characters 1419
3   Good Characters Female Characters   714
0   Bad Characters  Female Characters   419
8   Neutral Characters  Male Characters 254
6   Neutral Characters  Female Characters   138
1   Bad Characters  Genderless Characters   9
4   Good Characters Genderless Characters   4
7   Neutral Characters  Genderless Characters   3
9   Reformed Criminals  Male Characters 2

这是我与显示嵌套饼图相关的代码片段，如下所示。

fig, ax = plt.subplots(figsize=(24,12))
size = 0.3

ax.pie(dc_df_ALIGN_SEX.groupby('SEX')['count'].sum(), radius=1,
       labels=dc_df_ALIGN_SEX['SEX'].drop_duplicates(),
       autopct='%1.1f%%',
       wedgeprops=dict(width=size, edgecolor='w'))

ax.pie(dc_df_ALIGN_SEX['count'], radius=1-size, labels = dc_df_ALIGN_SEX["ALIGN"],
       wedgeprops=dict(width=size, edgecolor='w'))

ax.set(aspect="equal", title='Pie plot with `ax.pie`')
plt.show()

如何设计 4 行 4 列并将每一个放在每个插槽中并在图例区域显示标签？

标签： pythonpandasmatplotlibpie-chart

Answer 1

由于问题已更改，因此我发布了一个新答案。

首先，我稍微简化了您的 DataFrame：

import pandas as pd

df = pd.DataFrame([['Bad', 'Male', 1542],
                   ['Good', 'Male', 1419],
                   ['Good', 'Female', 714],
                   ['Bad', 'Female', 419],
                   ['Neutral', 'Male', 254],
                   ['Neutral', 'Female', 138], 
                   ['Bad', 'Genderless', 9], 
                   ['Good', 'Genderless', 4],
                   ['Neutral', 'Genderless', 3], 
                   ['Reformed', 'Male', 2]])
df.columns = ['ALIGN', 'SEX', 'n']

对于外环中的数字，我们可以使用简单的groupby，就像您所做的那样：

outer = df.groupby('SEX').sum()

但是对于内环中的数字，我们需要按两个分类变量进行分组，这会产生一个 MultiIndex：

inner = df.groupby(['SEX', 'ALIGN']).sum()
inner

                     n
SEX         ALIGN   
Female      Bad      419
            Good     714
            Neutral  138
Genderless  Bad        9
            Good       4
            Neutral    3
Male        Bad     1542
            Good    1419
            Neutral  254
            Reformed   2

我们可以使用它的方法从 MultiIndex 中提取适当的标签get_level_values()：

inner_labels = inner.index.get_level_values(1)

现在您可以将上述值转换​​为一维数组并将它们插入您的绘图调用中：

import matplotlib.pyplot as plt
import numpy as np

fig, ax = plt.subplots(figsize=(24,12))
size = 0.3

ax.pie(outer.values.flatten(), radius=1,
       labels=outer.index,
       autopct='%1.1f%%',
       wedgeprops=dict(width=size, edgecolor='w'))

ax.pie(inner.values.flatten(), radius=1-size, 
       labels = inner_labels,
       wedgeprops=dict(width=size, edgecolor='w'))

ax.set(aspect="equal", title='Pie plot with `ax.pie`')
plt.show()