将pygame表面缓冲区字节转换为numpy数组

Question

有人知道为什么我用pygame的get_buffer方法创建的这个数组的R、G、B值颠倒了吗？我想创建一个具有颜色值的数组，其顺序与我将它们放入的顺序相同——类似于[8, 16, 32, 0]。我做错了什么吗？还是pygame存储像素数据的方式有问题？

>>> import pygame
>>> import pygame.gfxdraw
>>> import numpy as np
>>> background_colour = (1, 1, 1)
>>> width, height = (256, 256)
>>> screen = pygame.Surface((width, height))
>>> pygame.draw.rect(screen, (8, 16, 32), (0, 0, 100, 100), 0)
<rect(0, 0, 100, 100)>
>>> s = screen.get_buffer()
>>> x = np.fromstring(s.raw, dtype='b').reshape(height, width, 4)
>>> x[0, 0]
array([32, 16,  8,  0], dtype=int8)

我试过了，但我失去了R值：

>>> y = x[:, :, 3:0:-1]
>>> y[0, 0]
array([ 0,  8, 16], dtype=int8)

(我使用的是numpy 1.8.2版本，所以我没有np.flip(。

Answer 1

我意识到有更好的方法可以做到这一点。pygame.surfarray模块有各种方法，可以为您实际创建numpy数组！

>>> x3 = pygame.surfarray.pixels3d(screen)
>>> x3.shape
(256, 256, 3)
>>> x3[0, 0]
array([ 8, 16, 32], dtype=uint8)