如何使用matplotlib中的分类数据创建饼图？ [英] How do I create a pie chart using categorical data in matplotlib?

问题描述

我有以下数据：

ID  Gender  Country  ...
1   Male    UK
2   Female  US
3   Male    NZ
4   Female  UK
...

性别只有2个选项，国家/地区只有3个选项。我想为性别和国家/地区创建一个单独的饼图，以显示每个选项在数据中出现的次数，但我对如何执行此操作感到困惑。

数据存储在 pandas 数据框中。

非常感谢您的任何帮助！

推荐答案

以下是一种使用pandas：

的方法

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

def label_function(val):
    return f'{val / 100 * len(df):.0f}
{val:.0f}%'

N = 50
df = pd.DataFrame({'country': np.random.choice(['UK', 'US', 'NZ'], N),
                   'gender': np.random.choice(['Male', 'Female'], N)})

fig, (ax1, ax2) = plt.subplots(ncols=2, figsize=(10, 5))

df.groupby('country').size().plot(kind='pie', autopct=label_function, textprops={'fontsize': 20},
                                  colors=['tomato', 'gold', 'skyblue'], ax=ax1)
df.groupby('gender').size().plot(kind='pie', autopct=label_function, textprops={'fontsize': 20},
                                 colors=['violet', 'lime'], ax=ax2)
ax1.set_ylabel('Per country', size=22)
ax2.set_ylabel('Per gender', size=22)plt.tight_layout()
plt.show()

ps：若要仅显示百分比，请使用autopct='%1.0f%%'。

这篇关于如何使用matplotlib中的分类数据创建饼图？的文章就介绍到这了，希望我们推荐的答案对大家有所帮助，也希望大家多多支持IT屋！