python - Python/numpy点列表到黑白图像区域

Question

我正在尝试将连续的列表点(介于0和1之间)转换为黑白图像，以表示列表点下方/上方的区域。

plt.plot(points)
plt.ylabel('True val')
plt.show()
print("Points shape-->", points.shape)

我可以保存由matplotlib生成的图像，但是我认为这可能是一个令人讨厌的解决方法

最后，我想获得形状为(224,224)的图像，其中白色区域代表线下区域，黑色区域代表线之上...

image_area = np.zeros((points.shape[0],points.shape[0],))
# ¿?

任何想法或建议如何处理它是受欢迎的!谢谢专家

Answer 1

这是如何做的一个基本例子。由于 slice 需要整数，因此您可能必须首先缩放原始数据。

import numpy as np
import matplotlib.pyplot as plt

# your 2D image
image_data = np.zeros((224, 224))

# your points. Here I am just using a random list of points
points = np.random.choice(224, size=224)

# loop over each column in the image and set the values
# under "points" equal to 1
for col in range(len(image_data[0])):
    image_data[:points[col], col] = 1

# show the final image
plt.imshow(image_data, cmap='Greys')
plt.show()