如何解决seaborn绘制柱状图时相同数值柱子颜色深度不一致的问题
问题原因
你之前使用的data.argsort().argsort()生成的是每个元素的位置次序号,即使数值相同也会分配到不同的序号,所以会匹配到调色盘里不同位置的颜色,导致同值不同色的问题。
解决方法
改用pandas的rank()方法指定密集排序规则,让相同数值得到相同排名,同时调整调色盘长度和唯一值数量匹配即可,修改后代码如下:
import numpy as np, matplotlib.pyplot as plt, seaborn as sns import pandas as pd fig = plt.figure(1) sns.set(style="whitegrid", color_codes=True) data = departments.value_counts()[:10] # 取唯一计数值的数量作为调色盘长度,保证每个数值对应唯一颜色 unique_vals = data.unique() pal = sns.color_palette("Blues_d", len(unique_vals)) # dense规则排序保证相同数值排名一致,减1适配0起始的索引规则 rank = data.rank(method='dense', ascending=False).astype(int) - 1 sns.barplot(y=data.index, x=data, palette=np.array(pal)[rank]) plt.title("Top departments", y=1.02) fig.savefig('Top_departments', bbox_inches='tight', dpi=300)
修改后数值相同的柱子就会自动匹配到相同深浅的蓝色,比如计数为8的3个部门、计数为7的3个部门都会分别使用统一的色调。
内容的提问来源于stack exchange,提问作者anikaM
相关产品推荐
相关产品推荐

