python - 有没有更好的方法来获取rgb矩阵？

Question

我正在尝试创建一个 python 程序，它获取一些图像并将其转换为 ASCII 艺术。
该项目取自Robert Heaton他在网页上建议了一些编程项目来培养您的技能。

嗯，在某一时刻，我必须从每个像素获取 RGB 值并将它们存储在矩阵中，我认为这可以用比我更好的方式来完成。这是我的代码:

def extractPixels(img=None):
    '''
    This function will receive a Image object and return
    a 2D matrix containing pixels information
    '''
    if(type(img) == None or not(Image.isImageType(img))):
        raise TypeArgumentError("You have to pass a Image object")

    dataMatrix = []
    auxList = []
    for i in range(0, img.width, 1):
        for j in range(0, img.height, 1):
            auxList.append(img.getpixel((i,j)))
        dataMatrix.append(auxList)
        auxList = []

    return dataMatrix

我正在使用Pillow图像处理库。

这段代码

img.getpixel(i,j)

将返回每个像素的元组(R，G，B)。

Answer 1

无需执行任何显式循环...您可以直接将图像转换为 numpy 数组

import numpy
from PIL import Image
img = numpy.uint8(Image.open("myimage.jpg"))
h, w, _ = img.shape

然后

r, g, b = img[y][x]