将哈希模式添加到 seaborn 条形图中

Question

我有一个使用 seaborn 创建的条形图。例如，可以按如下方式创建图：

import pandas as pd
import numpy as np
import matplotlib.pyplot as plt
import seaborn as sns
data1 = pd.DataFrame(np.random.rand(17,3), columns=['A','B','C']).assign(Location=1)
data2 = pd.DataFrame(np.random.rand(17,3)+0.2, columns=['A','B','C']).assign(Location=2)
data3 = pd.DataFrame(np.random.rand(17,3)+0.4, columns=['A','B','C']).assign(Location=3)
cdf = pd.concat([data1, data2, data3])
mdf = pd.melt(cdf, id_vars=['Location'], var_name=['Letter'])
ax = sns.barplot(x="Location", y="value", hue="Letter", data=mdf, errwidth=0)  
ax.legend(loc='upper center', bbox_to_anchor=(0.5, 1.2), ncol=3, fancybox=True, shadow=True)
plt.show()

这给出了以下情节 我想按如下方式自定义图表：

1. 去除面部颜色（将其设置为白色）
2. 向图像添加哈希模式以区分组 这是如何实现的？

Answer 1

去除面部颜色很容易，只需执行ax.set_facecolor('w')，尽管这会使网格线不可见。我建议sns.set_style('whitegrid')在绘图之前使用，你会得到一个白色的背景，只有灰色的水平网格线。

至于不同的has模式，这对seaborn来说有点棘手，但可以做到。您可以将hatch关键字参数传递给 barplot，但它会应用于每个条形图，这并不能真正帮助您区分它们。不幸的是，在这里传递字典不起作用。相反，您可以在构建条形后迭代条形以应用影线。您必须计算位置的数量，但这对于 Pandas 来说非常简单。事实证明，seaborn 在移动到下一个色调之前实际上会绘制每个色调，因此在您的示例中，它将绘制所有蓝色条，然后绘制所有绿色条，然后绘制所有红色条，因此逻辑非常简单：

num_locations = len(mdf.Location.unique())
hatches = itertools.cycle(['/', '//', '+', '-', 'x', '\\', '*', 'o', 'O', '.'])
for i, bar in enumerate(ax.patches):
    if i % num_locations == 0:
        hatch = next(hatches)
    bar.set_hatch(hatch)

所以完整的脚本是

import itertools
import pandas as pd
import numpy as np
import matplotlib.pyplot as plt
import seaborn as sns
sns.set_style('whitegrid')
data1 = pd.DataFrame(np.random.rand(17, 3), columns=['A', 'B', 'C']).assign(Location=1)
data2 = pd.DataFrame(np.random.rand(17, 3) + 0.2, columns=['A', 'B', 'C']).assign(Location=2)
data3 = pd.DataFrame(np.random.rand(17, 3) + 0.4, columns=['A', 'B', 'C']).assign(Location=3)
cdf = pd.concat([data1, data2, data3])
mdf = pd.melt(cdf, id_vars=['Location'], var_name=['Letter'])
ax = sns.barplot(x="Location", y="value", hue="Letter", data=mdf, errwidth=0)

num_locations = len(mdf.Location.unique())
hatches = itertools.cycle(['/', '//', '+', '-', 'x', '\\', '*', 'o', 'O', '.'])
for i, bar in enumerate(ax.patches):
    if i % num_locations == 0:
        hatch = next(hatches)
    bar.set_hatch(hatch)

ax.legend(loc='upper center', bbox_to_anchor=(0.5, 1.1), ncol=3, fancybox=True, shadow=True)

plt.show()

我得到了输出

设置舱口和可用不同舱口的参考：http : //matplotlib.org/examples/pylab_examples/hatch_demo.html

注意：我根据bbox_to_anchor图例调整了您的图例，因为它部分位于我计算机上的图形之外。