python - Matplotlib，图例未出现在直方图中

Question

为了描述我的问题，我为您提供了一个小数据集作为示例:想象一下以下数据集:

import matplotlib.pyplot as plt
import pandas as pd
import numpy as np
df = pd.DataFrame({'name':['a', 'b', 'c', 'a', 'b', 'c'], 'val':[1,5,3,4,5,3]} )

我正在使用以下代码创建简单的直方图:

def plot_bar_x():
    index = np.arange(len(df['name']))
    plt.bar(index, df['val'])
    plt.legend(list(df['name'].unique()))
    plt.xticks(index, df['name'], fontsize=10, rotation=30)
    plt.show()
plot_bar_x()

但它给了我以下情节:

虽然我有 3 个唯一的名称，但我只看到“a”标签，但是我使用了这一行:plt.legend(list(df['name'].unique())) 另一个问题是所有的条都是相同的颜色，有没有办法在不事先手动定义颜色的情况下为独特的标签获得不同的颜色？

期望的输出是:

Answer 1

你只能在一个系列上绘制一次，这就是为什么 plt 只为图例选择一个标签。如果您没有很多名字，请尝试:

def plot_bar_x():
    index = np.arange(len(df['name']))
    plt.figure()
    for name in df.name.unique():
        tmp_df = df[df.name == name]
        plt.bar(tmp_df.index, tmp_df.val, label=name)
    plt.xticks(index, df['name'], fontsize=10, rotation=30)
    plt.legend()
    plt.show()

一定有一些聪明的方法可以解决您的问题，但现在我无法解决。