如何为 Seaborn 条形图中的色调级别分配不同的调色板？

Question

我正在使用 Seaborn 在 Python 中创建条形图。 我有一个 2（注意力：分散与专注）X 3（解决方案：1、2、3）设计。 对于划分，我希望条形颜色为深红色、深绿色、深蓝色。 为了集中注意力，我希望条形颜色为红色、绿色、蓝色。

此代码不起作用，因为调色板仅适用于 hue 参数：

import seaborn as sns
import matplotlib.pyplot as plt

data = sns.load_dataset('attention')

palette = ["darkred", "red", "darkgreen", "green", "darkblue", "blue"]
palette = ["darkred", "darkgreen", "darkblue", "red", "green", "blue"]

sns.set(style='ticks', context='notebook', font_scale=1.2)
fig, ax = plt.subplots(figsize=(7, 5), dpi=96)
my_plot = sns.barplot(x="solutions", y="score", hue="attention",
                      palette=palette, capsize=.1, ax=ax, data=data)

plt.show()

这段代码几乎有效：

import seaborn as sns
import matplotlib.pyplot as plt

data = sns.load_dataset('attention')

sns.set(style='ticks', context='notebook', font_scale=1.5)
fig, ax = plt.subplots(figsize=(7, 5), dpi=96)
sns.barplot(x="solutions", y="score",
            palette=['red', 'green', 'blue'], capsize=.1, ax=ax,
            data=data.query('attention=="focused"'))
sns.barplot(x="solutions", y="score",
            palette=['darkred', 'darkgreen', 'darkblue'], capsize=.1, ax=ax,
            data=data.query('attention=="divided"'))
sns.despine(top=True, right=True)
plt.tight_layout()
plt.show()

但有两个问题：

1. 无图例
2. 条没有分开。

必须（应该？）以一种优雅的方式用 seaborn 创造这个人物。第一种方法是理想的：使用指定图形的参数一次调用 barplot。第二种方法没问题：多次调用 barplot。理想的租赁方法是直接调用 matplotlib bar()，但在这一点上，我会满足于任何导致所需数字的方法。

请注意，据我所知，这些名称相似的问题似乎并未解决此问题所寻求的结果：

1. seaborn barplot: vary color with x and hue
2. Set color-palette in Seaborn Grouped Barplot depending on values

Answer 1

您可以使用灰色和黑色作为图例的色调颜色来创建条形图。然后，循环创建条形并更改它们的颜色。

import seaborn as sns
import matplotlib.pyplot as plt

data = sns.load_dataset('attention')

sns.set(style='ticks', context='notebook', font_scale=1.2)
fig, ax = plt.subplots(figsize=(7, 5), dpi=96)
sns.barplot(x="solutions", y="score", hue="attention", hue_order=['divided', 'focused'],
            palette={'divided': 'grey', 'focused': 'black'}, capsize=.1, ax=ax, data=data)
for bar_group, desaturate_value in zip(ax.containers, [0.5, 1]):
    for bar, color in zip(bar_group, ['red', 'green', 'blue']):
        bar.set_facecolor(sns.desaturate(color, desaturate_value))
plt.show()

要将所有变体放入图例中，可以使用TupleHandler，例如（使用colors=['crimson', 'limegreen', 'dodgerblue']）：

from matplotlib.legend_handler import HandlerTuple

ax.legend(handles=[tuple(bar_group) for bar_group in ax.containers],
          labels=[bar_group.get_label() for bar_group in ax.containers],
          title=ax.legend_.get_title().get_text(),
          handlelength=4, handler_map={tuple: HandlerTuple(ndivide=None, pad=0.1)})

这是另一个例子：

data = sns.load_dataset('tips')

sns.set(style='ticks', context='notebook', font_scale=1.2)
fig, ax = plt.subplots(figsize=(12, 5))
sns.barplot(x="day", y="tip", hue="smoker",
            capsize=.1, ax=ax, data=data)
for bar_group, desaturate_value in zip(ax.containers, [0.5, 1]):
    for bar, color in zip(bar_group, plt.cm.Set2.colors):
        bar.set_facecolor(sns.desaturate(color, desaturate_value))

ax.legend(handles=[tuple(bar_group) for bar_group in ax.containers],
          labels=[bar_group.get_label() for bar_group in ax.containers],
          title=ax.legend_.get_title().get_text(),
          handlelength=4, handler_map={tuple: HandlerTuple(ndivide=None, pad=0.1)})
sns.despine()
plt.tight_layout()
plt.show()