项目场景:
提示:这里简述项目相关背景:
k-means聚类MNIST数据集,将聚类数据转为图像报错
问题描述
提示:这里描述项目中遇到的问题:
ValueError: cannot reshape array of size 2370032 into shape (28,28)
fix, axes = plt.subplots(4, 5, figsize=(5, 4),subplot_kw={'xticks': (), 'yticks': ()})
for cluster,ax in zip(clusters,axes.ravel()):
cluster = np.array(cluster)
image = cluster.reshape([28,28])
ax.imshow(image, cmap="gray")
原因分析:
提示:这里填写问题的分析:
1.网上查找原因是前后大小不匹配,没法转换。没看明白怎么改的,重新看代码,尝试搞清楚2370032这个大小是怎么转换过来的。
解决方案:
查看了cluster的具体内容,发现是[[[a]]],示例代码的数据是l类似[a,b],所以在变量类型相同的情况下,会报错。想办法取到a,再转换a就可以。