使用matplotlib和numpy将图像和标签打印为条形图

Question

所以我必须把数据集中的图像分布绘制成条形图。我试过几种方法来做这件事，但都没有用。在

我有两个纽比阵列：

X_列车形状（20000、32、32、3） 游车-外形（20000）

labels-用于标记字符串的标签索引词典。尺寸50

所以X_train包含图像，y_train包含相应的标签索引

我需要根据50个标签画一张X峎火车的柱状图。显示每个标签的图像数量分布。在

我是否应该首先根据y峈train中的相应索引对X峎train数组中的图像进行分组？这和matplotlib.bar文件API调用？在

或者我应该使用numpy直方图API。在

非常感谢任何帮助。在

Answer 1

一种方法是将histogram与一些附加参数一起使用。你可以用类似的东西

In [55]: y
Out[55]: array([0, 0, 1, 2, 1])

In [54]: plt.hist(y, align='mid', range=(np.min(y), np.max(y)+1), bins=50)
In [55]: plt.xlabel("labels")
In [56]: plt.ylabel("image counts")
In [57]: plt.show()

图中显示标签0&1出现2次，2出现一次。 从y_train获取标签并根据其计数绘制。请随意根据您的标签更改垃圾箱的数量。在