如何在Python中读取给定像素的RGB值?

时间:2020-03-06 14:46:16  来源:igfitidea点击:

如果我使用open(" image.jpg")打开图像,假设我具有像素的坐标,如何获得像素的RGB值?

那么,我该如何做相反的事情呢?从空白图形开始,"写入"具有特定RGB值的像素?

如果我不必下载任何其他库,我会希望。

解决方案

图像处理是一个复杂的主题,最好使用库。我可以推荐gdmodule,它可以从Python内轻松访问许多不同的图像格式。

最好使用Python图像库来执行此操作,恐怕这是单独下载的。

执行所需操作的最简单方法是通过Image对象上的load()方法,该方法返回一个像素访问对象,我们可以像数组一样对其进行操作:

from PIL import Image

im = Image.open('dead_parrot.jpg') # Can be many different formats.
pix = im.load()
print im.size  # Get the width and hight of the image for iterating over
print pix[x,y]  # Get the RGBA Value of the a pixel of an image
pix[x,y] = value  # Set the RGBA Value of the image (tuple)
im.save('alive_parrot.png')  # Save the modified pixels as .png

或者,查看ImageDraw,它提供了更丰富的API用于创建图像。

在wiki.wxpython.org上有一篇非常不错的文章,名为"使用图像"。本文提到了使用wxWidgets(wxImage),PIL或者PythonMagick的可能性。我个人使用过PIL和wxWidgets,它们都使图像处理相当容易。

PyPNG轻量级PNG解码器/编码器

尽管该问题暗示了JPG,但我希望我的回答对某些人有用。

以下是使用PyPNG模块读取和写入PNG像素的方法:

import png, array

point = (2, 10) # coordinates of pixel to be painted red

reader = png.Reader(filename='image.png')
w, h, pixels, metadata = reader.read_flat()
pixel_byte_width = 4 if metadata['alpha'] else 3
pixel_position = point[0] + point[1] * w
new_pixel_value = (255, 0, 0, 0) if metadata['alpha'] else (255, 0, 0)
pixels[
  pixel_position * pixel_byte_width :
  (pixel_position + 1) * pixel_byte_width] = array.array('B', new_pixel_value)

output = open('image-with-red-dot.png', 'wb')
writer = png.Writer(w, h, **metadata)
writer.write_array(output, pixels)
output.close()

PyPNG是一个单独的纯Python模块,长度不足4000行,包括测试和注释。

PIL是一个更全面的映像库,但它也要重得多。

我们可以使用pygame的surfarray模块。该模块具有一种称为pixel3d(surface)的3d像素阵列返回方法。我在下面显示了用法:

from pygame import surfarray, image, display
import pygame
import numpy #important to import

pygame.init()
image = image.load("myimagefile.jpg") #surface to render
resolution = (image.get_width(),image.get_height())
screen = display.set_mode(resolution) #create space for display
screen.blit(image, (0,0)) #superpose image on screen
display.flip()
surfarray.use_arraytype("numpy") #important!
screenpix = surfarray.pixels3d(image) #pixels in 3d array:
#[x][y][rgb]
for y in range(resolution[1]):
    for x in range(resolution[0]):
        for color in range(3):
            screenpix[x][y][color] += 128
            #reverting colors
screen.blit(surfarray.make_surface(screenpix), (0,0)) #superpose on screen
display.flip() #update display
while 1:
    print finished

希望对我们有所帮助。最后一句话:屏幕在screenpix的生命周期内被锁定。