将numpy数组转换为图片

Question

所以我有一个字符串，我用numpy数组中的1-5之间的数字表示。现在，我想通过首先向下重复数字字符串将其转换为图片形式，以使图片变得足够宽以可见（因为单个字符串将给出一线图片）。我的主要问题是如何将数字数组转换为图片？

Answer 1

这是使用matplotlib可视化的最小工作示例：

import numpy as np
import matplotlib.pyplot as plt

# generate 256 by 1 vector of values 1-5
img = np.random.randint(1,6, 256)
# transpose for visualization
img = np.expand_dims(img, 1).T

# force aspect ratio
plt.imshow(img, aspect=100)
# or, alternatively use aspect='auto'
plt.show()

您只需设置aspect的{​​{1}}选项即可强制绘制图形的长宽比