可以使用numpy库中的array方法将图像转换为数组,然后赋值给一个变量。示例如下:
import numpy as np
import matplotlib.pyplot as plt
# 生成图像数据
data = np.random.rand(5, 5)
plt.imshow(data)
# 将图像数据转换为数组
img_array = np.array(plt.imshow(data))
# 打印数组
print(img_array)
<matplotlib.image.AxesImage object at 0x0000027179F53670>
在上面的示例中,我们首先生成一个随机的5×5的数组,然后使用plt.imshow()方法将其打印成图像。接着,将这个图像数据转换为数组并赋值给img_array变量,最后使用print()方法输出这个变量。
需要注意的是,img_array变量中保存的是一个AxesImage对象,而非原始的图像数据数组。如果需要获取原始的图像数据数组,可以使用AxesImage对象的data属性:
img_array = np.array(plt.imshow(data).data)
这样就可以获取原始的图像数据数组了。