python - Pandas - 图像到 DataFrame

Question

我想将 RGB 图像转换为 DataFrame，以便获得每个像素的坐标及其 RGB 值。

         x   y   red  green  blue
0        0   0   154      0     0
1        1   0   149    111     0
2        2   0   153      0     5
3        0   1   154      0     9
4        1   1   154     10    10
5        2   1   154      0     0

我可以很容易地将 RGB 提取到 DataFrame 中

colourImg = Image.open("test.png")
colourPixels = colourImg.convert("RGB")
colourArray = np.array(colourPixels.getdata())

df = pd.DataFrame(colourArray, columns=["red","green","blue"])

但我不知道如何获取其中的 X 和 Y 坐标。我可以编写一个循环，但是在大图像上需要很长时间。

Answer 1

尝试使用np.indices不幸的是它最终得到一个坐标是第一个维度的数组，但你可以做一些np.moveaxis来解决这个问题.

colourImg = Image.open("test.png")
colourPixels = colourImg.convert("RGB")
colourArray = np.array(colourPixels.getdata()).reshape(colourImg.size + (3,))
indicesArray = np.moveaxis(np.indices(colourImg.size), 0, 2)
allArray = np.dstack((indicesArray, colourArray)).reshape((-1, 5))


df = pd.DataFrame(allArray, columns=["y", "x", "red","green","blue"])

这不是最漂亮的，但它似乎有效(编辑:修复了 x,y 的错误方式)。