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]} )

我用以下代码创建简单的histogam：

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

必须有一些聪明的方法来解决你的问题，但现在已经超出了我的想法。

Answer 2

您可以使用df.plot()直接绘制DataFrame，然后使用一些预定义的字典传递每个条的颜色。然后你可以使用mpatches.Patch创建传说，如图所示here。您可以使用seaborn使灰色背景与网格。可以根据DataFrame中唯一项的数量生成颜色。

import matplotlib.pyplot as plt
import matplotlib.patches as mpatches
import pandas as pd
import numpy as np; np.random.seed(123)
import seaborn as sns
sns.set()

df = pd.DataFrame({'name':['a', 'b', 'c', 'a', 'b', 'c'], 'val':[1,5,3,4,5,3]} )
col_dict = {k:np.random.rand(3,) for k in df['name'].unique()}

ax = df.plot.bar(x='name', y='val', color=[col_dict[i] for i in df['name']])
labels = [mpatches.Patch(color=v, label=k) for k,v in col_dict.items()]
plt.legend(handles=labels, loc='best')    
plt.show()