如何使用numpy从2D图像矩阵生成所有(x, y，像素值)元组的列表

Question

使用python和可能的numpy，我如何生成一个(x,y, pixelValue(x,y))格式的所有坐标列表?我可以使用以下语句实现for循环:

img = cv2.imread('messi5.jpg',0)
data = []
for x in range(img.shape[1]):
    for y in range(img.shape[0]):
        data.append((x, y, data[x, y]))

但是当我的图像包含~ 400万像素时，这样做需要非常长的时间。谢谢!

Answer 1

numpy.ndenumerate将立即创建生成器。如果你真的需要(???)to.

，你可以把它转换成一个列表。

>>> np.ndenumerate(np.eye(2))
<numpy.lib.index_tricks.ndenumerate at 0x20d485f8>
>>>for elem in np.ndenumerate(np.eye(2)):
...    print elem
...
((0L, 0L), 1.0)
((0L, 1L), 0.0)
((1L, 0L), 0.0)
((1L, 1L), 1.0)
>>>list(np.ndenumerate(np.eye(2)))
[((0L, 0L), 1.0), ((0L, 1L), 0.0), ((1L, 0L), 0.0), ((1L, 1L), 1.0)]