为了描述我的问题,我为您提供了一个小的数据集作为示例:请想象以下数据集:
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()))
另一个问题是所有条形都是相同的颜色,是否有一种方法可以使唯一标签获得不同的颜色而无需事先手动定义颜色?
所需的输出是:
解决方法:
您只在一个系列中绘制一次,这就是为什么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()
必须有一些巧妙的方法来解决您的问题,但是现在这已经超出了我的头脑.
标签:pandas,matplotlib,histogram,legend,python