如何将一个（1，28，28）的ndarray 以图像的方式输出

最新推荐文章于 2023-07-23 11:37:50 发布

原创最新推荐文章于 2023-07-23 11:37:50 发布 · 715 阅读

1 ·

CC 4.0 BY-SA版权

文章标签：

#python #numpy #机器学习

深度学习专栏收录该内容

2 篇文章

订阅专栏

这段代码演示如何用NumPy生成一个形状为(1,28,28)的随机数组，然后通过squeeze函数转换为(28,28)，以便用Matplotlib的imshow以灰度模式显示图像。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

import matplotlib.pyplot as plt
import numpy as np

# 创建形状为（1，28，28）的随机NumPy数组
image = np.random.rand(1, 28, 28)

# 将形状从（1，28，28）转换为（28，28）
image = np.squeeze(image)

# 显示图像
plt.imshow(image, cmap='gray')
plt.show()

在此示例中，我们首先使用NumPy的random.rand()函数创建了一个形状为（1，28，28）的随机NumPy数组。我们然后使用NumPy的squeeze()函数将形状从（1，28，28）转换为（28，28），以便我们可以将其显示为图像。最后，我们使用Matplotlib的imshow()函数将NumPy数组显示为图像，并使用show()函数显示图像。

请注意，由于我们使用了gray颜色映射，因此图像将以灰度形式显示。如果您想要显示彩色图像，则需要选择适当的颜色映射。