python - 每个 ytick 有多个条形的水平条形图

Question

我不太明白您应该如何传递参数以在 matplotlib 中创建水平条形图。我只是想模仿我在这个示例代码中看到的... https://pythonspot.com/matplotlib-bar-chart/

以及我在这篇 stackoverflow 帖子中看到的内容 How to plot multiple horizontal bars in one chart with matplotlib

现在我的代码如下:

import numpy as np
x_locs = np.arange(len(total_vals))
t_label_lst = ['Digital Learning Apps', 'News, Events, Daily', 'News on School Events', 'STEM Extracurriculars & School Programs', 'Hiring, STEM Workforce', 'Women in STEM', 'Activities and Projects Outside of the Classroom', 'Ambiguous', 'Ambiguous, STEM in College', 'Next Generation of Engineers', 'News, Events, Daily', 'Educational Policy and Higher Education, Reform', 'STEM Activities, Building, Arts, and Design', 'Engaging students with STEM using programming and robotics', 'Black Leaders in STEM', 'Next Generation of Engineers', 'Ambiguous', 'Astronomy, NASA', 'STEM workshops and summer camps', 'Competitions, Team Credit', 'Ambiguous, Technology Hashtags', 'Google Education', 'Good Job Today! Crediting Daily Activities and Work', 'Engaging students with STEM using programming and robotics', 'Environmental Science', 'Teachers, Public Schools In STEM', 'Ambiguous', 'Edtech Companies', 'Ambiguous, PHD Conversation', 'Ambiguous', 'Engaging students with STEM using programming and robotics', 'Ambiguous, Virtual Reality and Personalized learning mention', 'Ambiguous', 'Ambiguous', 'Ambiguous, #Autism hashtag has disproportionate weight']

print(x_locs)
total_vals = [23668, 13186, 10752, 10002, 9558, 9126, 8138, 7389, 7006, 6965, 6859, 6621, 6538, 5700, 5110, 5069, 4419, 4025, 3943, 3866, 3761, 3697, 3543, 3294, 3067, 2928, 2511, 2491, 2353, 2312, 2229, 2175, 2021, 1921, 1787]
positive_vals = [9941, 9306, 7595, 5935, 5913, 7488, 5258, 4905, 4026, 5242, 5557, 3225, 3530, 3055, 3300, 3503, 2461, 2199, 2074, 2379, 1665, 2274, 2250, 1674, 1523, 1533, 1241, 859, 1504, 1419, 1132, 1082, 805, 753, 580]
neutral_vals = [13727, 3880, 3157, 4067, 3645, 1638, 2880, 2484, 2980, 1723, 1302, 3396, 3008, 2645, 1810, 1566, 1958, 1826, 1869, 1487, 2096, 1423, 1293, 1620, 1544, 1395, 1270, 1632, 849, 893, 1097, 1093, 1216, 1168, 1207]

rects1 = ax.barh(x_locs, total_vals, width=.15, color='r', label="total tweet count")
# rects2 = ax.barh(positive_vals, width=.2, color = 'b', label="positive tweet count")
# rects3 = ax.barh(neutral_vals, width=.2, color='yellow', label="neutral tweet count")
ax.set(yticks=x_locs, yticklabels=t_label_lst, ylim=[0, len(x_locs)])
plt.show()

但这只会导致程序崩溃并显示以下错误消息:“TypeError: barh() 为参数‘width’获取了多个值”我什至不确定如果我设置宽度参数为 .15。

total_vals 包含 35 个计数，或我试图在 x 轴上绘制的频率。就像这些是我试图让我的酒吧反射(reflect)在高度方面的值(value)观。 positive_vals 和 neutral_vals 也包含 35 个计数。

x_locs 是我根据我在示例代码中看到的内容创建的一个变量，但它应该只是数字 0-34。这只是应该指示我要显示的 35 个条中的每一个。

同样让我感到困惑的是，当我删除 x_loc 时会发生什么。所以如果我只是做

 rects1 = ax.barh(total_vals, width=.15, color='r', label="total tweet count")

我得到了这个 matplotlib 图，它似乎在图的 x 轴上绘制了我的“宽度”参数/使宽度参数成为 x 轴的限制。

我的 yticks 看起来它们实际上至少按照它们应该进行的顺序正确显示，但看起来宽度是在 x 轴上绘制或测量的内容......很明显，图表是空的.

我知道我的 yticklabels 现在长得可怕，我只是上传了这段代码，以便它可以重现。

此时我不太确定“y”参数实际上应该表示什么。文档说“每个条的 y 坐标”。我最初以为 y 参数应该只是我想显示的计数，但后来我看到“left”参数是“条形图左侧的 x 坐标”。

所以我把我的代码改成了

rects1 = ax.barh(y=x_locs, left=total_vals, width=.15, color='r', label="total tweet count")

这正确地改变了 y 轴，但它仍然给我一个空图。我不知道我的计数/条发生了什么，也不知道为什么它们根本不显示。

我实际上如何让条形图显示出来？我只是对这些水平条形图的参数实际工作方式有一些明显的困惑。

Answer 1

您实际上非常了解您的代码；您只是对 barh 的参数有些困惑。

首先，让我们弄清楚所需的参数是什么。

第一个位置参数是y，它指的是每个类别 的y 坐标。因此，x_locs 是用词不当。这些可以按运行顺序轻松创建:第一个类别位于 y=0，第二个类别位于 y=1，依此类推。

接下来，您传递一个变量，如 total_vals，还有 width，并得到一个错误。

考虑一个条由两个方面定义，位置 和大小。由于所有条形都与图的左侧脊柱对齐，因此它们的 x 坐标相同，并且它们将仅由它们的 y 坐标定义，我们已经完成了这一点。

对于条形图，我们通常希望一个维度是动态的(表示某个数量的维度)，而另一个维度对于所有条形图都是相同的。相同的是 height，因为它等于条形的垂直尺寸。

另外一个动态维度，那么，就是width，width代表total_vals，positive_vals 和 negative_vals。因此，您收到该错误是因为您试图指定每个条的长度应该两次。

现在，让我们回到y_locs。请记住，我们希望每个订单号有 3 个柱线，并排。这相当于说我们希望每个“类别”的条形图(total、negative 和positive)都进行轻微调整，否则它们会重叠。

例如，我们可以让 negative 的条保留在 y_loc 指定的位置，positive 的条向下移动 10像素，total 的像素为 20。这是一个实现细节；重要的是我们认识到需要这样的抵消。

将所有这些放在一起，我们得到:

import numpy as np
from matplotlib import pyplot as plt

fig, ax = plt.subplots(figsize=(6, 10))

t_label_lst = ['Digital Learning Apps', 'News, Events, Daily', 'News on School Events', 'STEM Extracurriculars & School Programs', 'Hiring, STEM Workforce', 'Women in STEM', 'Activities and Projects Outside of the Classroom', 'Ambiguous', 'Ambiguous, STEM in College', 'Next Generation of Engineers', 'News, Events, Daily', 'Educational Policy and Higher Education, Reform', 'STEM Activities, Building, Arts, and Design', 'Engaging students with STEM using programming and robotics', 'Black Leaders in STEM', 'Next Generation of Engineers', 'Ambiguous', 'Astronomy, NASA', 'STEM workshops and summer camps', 'Competitions, Team Credit', 'Ambiguous, Technology Hashtags', 'Google Education', 'Good Job Today! Crediting Daily Activities and Work', 'Engaging students with STEM using programming and robotics', 'Environmental Science', 'Teachers, Public Schools In STEM', 'Ambiguous', 'Edtech Companies', 'Ambiguous, PHD Conversation', 'Ambiguous', 'Engaging students with STEM using programming and robotics', 'Ambiguous, Virtual Reality and Personalized learning mention', 'Ambiguous', 'Ambiguous', 'Ambiguous, #Autism hashtag has disproportionate weight']

total_vals = [23668, 13186, 10752, 10002, 9558, 9126, 8138, 7389, 7006, 6965, 6859, 6621, 6538, 5700, 5110, 5069, 4419, 4025, 3943, 3866, 3761, 3697, 3543, 3294, 3067, 2928, 2511, 2491, 2353, 2312, 2229, 2175, 2021, 1921, 1787]
positive_vals = [9941, 9306, 7595, 5935, 5913, 7488, 5258, 4905, 4026, 5242, 5557, 3225, 3530, 3055, 3300, 3503, 2461, 2199, 2074, 2379, 1665, 2274, 2250, 1674, 1523, 1533, 1241, 859, 1504, 1419, 1132, 1082, 805, 753, 580]
neutral_vals = [13727, 3880, 3157, 4067, 3645, 1638, 2880, 2484, 2980, 1723, 1302, 3396, 3008, 2645, 1810, 1566, 1958, 1826, 1869, 1487, 2096, 1423, 1293, 1620, 1544, 1395, 1270, 1632, 849, 893, 1097, 1093, 1216, 1168, 1207]

bar_size = 0.25
padding = 0.25

y_locs = np.arange(len(total_vals)) * (bar_size * 3 + padding)

rects1 = ax.barh(y_locs, total_vals, align='edge', height=bar_size, color='r', label="total tweet count")
rects2 = ax.barh(y_locs + bar_size, positive_vals, align='edge', height=bar_size, color='b', label="positive tweet count")
rects3 = ax.barh(y_locs + 2 * bar_size, neutral_vals, align='edge', height=bar_size, color='yellow', label="neutral tweet count")
ax.set(yticks=x_locs, yticklabels=t_label_lst, ylim=[0 - padding, len(x_locs)])