在 for 循环中绘制 groupby

Question

我在 for 循环内绘制 groupby 时遇到问题。 我正在尝试为数据中的每个分类特征生成一个 groupby 条形图，但是当我运行循环时，什么也没有发生。

这是代码：

for col in df.select_dtypes(include = ['object']):
    plt.figure(figsize = (12, 7))
    df.groupby(col).col2.mean().plot(kind = 'bar')
    plt.show();

有人知道为什么这没有产生任何情节吗？

Answer 1

您可以创建需要分组的所需列的列表，然后遍历这些列，例如：

假设您总是希望在 dataframe 中的其他列上按平均值对 col3 进行分组，因此您首先返回除 col3 之外的所有其他 dataframe 的列表，然后循环遍历该列表：

neededCols = [c for c in df if c is not 'col3']

for col in neededCols:
    plt.figure(figsize = (12, 7))
    df.groupby(col).col3.mean().plot(kind = 'bar')
    plt.show()