如何使用 pyglet 显示 numpy 数组？

我有一个尺寸为 (100*100) 的标签矩阵，存储为 numpy 数组，我想用 pyglet 显示该矩阵。

我最初的想法是使用这个矩阵通过函数 pyglet.image.ImageData() 形成一个新的 pyglet 图像。它需要图像数据的缓冲区作为输入，但是我不知道如何从 numpy 数组中获取正确格式的缓冲区。

有人有任何想法吗？

附：我当前的解决方案：

3d_label = numpy.empty([100,100,3])
3d_label[:,:,0] = label * 255            # value range of label is [0,1]
3d_label[:,:,1] = label * 255
3d_label[:,:,2] = label * 255
image_data = ctypes.string_at(id(3d_label.tostring())+20, 100*100*3)
image = pyglet.image.ImageData(100, 100, 'RGB', image_data, -100*3)

有没有更好的方法用 numpy 从 3 个 [100*100] 矩阵构造一个 [100*100*3] 矩阵？

我想你正在寻找的是np.dstack（或者更一般地说，np.concatenate):

label255=label*255
label3=numpy.dstack((label255,label255,label255))

由此可见dstack产生相同的数组（label3）作为您的构造label_3d:

import numpy as np

label=np.random.random((100,100))
label255=label*255
label3=np.dstack((label255,label255,label255))

label_3d = np.empty([100,100,3])
label_3d[:,:,0] = label * 255            # value range of label is [0,1]
label_3d[:,:,1] = label * 255
label_3d[:,:,2] = label * 255
print(np.all(label3==label_3d))
# True

附言。我不确定，但你尝试过使用label3.data代替ctypes.string_at(id(label3.tostring())+20, 100*100*3) ?