python - 使用 matplotlib 的分类变量的直方图

Question

我在 pandas 数据框中有一列，其中包含三个可能的分类值。当我尝试使用 matplotlib 中的 plt.hist(data['column']) 绘制它时，直方图条未与 x 轴刻度对齐，并且它们的间距不均匀。我该如何解决这个问题？

Answer 1

直方图用于绘制数值变量(连续或离散)的频率分布。分类变量的频率分布最好用条形图显示。为此，您首先需要使用 value_counts 计算每个类别的频率。然后你可以方便地直接用 pandas plot.bar 绘制它.或者如果您愿意，也可以使用 matplotlib，如下所示。

import numpy as np               # v 1.19.2
import pandas as pd              # v 1.2.3
import matplotlib.pyplot as plt  # v 3.3.4

data = pd.DataFrame(dict(column=np.repeat(['F', 'M', '--'], [11000, 13000, 3000])))

Pandas

data['column'].value_counts().plot.bar(rot=0)

matplotlib

categories = data['column'].value_counts().index
counts = data['column'].value_counts().values
plt.bar(categories, counts, width=0.5)