我有一个二维ndarray。每个元素都是整数。我以这种方式显示矩阵:
plt.figure()
plt.imshow(img, cmap='Greys')
在这里,img 是一个二维数组。现在我想强调一些关于图片的点:例如,我希望像素 (x,y) 是红色的(我不想改变真实的像素值,只是用红色显示图像)。
有什么建议吗?
import numpy as np
import matplotlib.pyplot as plt
im = np.random.randint(0, 255, (16, 16))
I = np.dstack([im, im, im])
x = 5
y = 5
I[x, y, :] = [1, 0, 0]
plt.imshow(I, interpolation='nearest' )
plt.imshow(im, interpolation='nearest', cmap='Greys')
我任意选择了x
和y
的值为5(原图大小为16x16)。
最后一行显示im
未被改变,倒数第二行给出了想要的结果。
输出: