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 列，并将每一个放在每个插槽中，并在图例区域显示标签？

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

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

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()