使用matplotlib将RBG图像用作3D立方体的纹理

import matplotlib.pyplot as plt
from mpl_toolkits.mplot3d import Axes3D
import numpy as np
from PIL import Image

image = Image.open('image.jpg')
rgb_image = image.convert('RGB')

rgb_array = np.array(rgb_image)

fig = plt.figure()
ax = fig.add_subplot(111, projection='3d')

vertices = np.array([[0, 0, 0], [1, 0, 0], [1, 1, 0], [0, 1, 0], [0, 0, 1], [1, 0, 1], [1, 1, 1], [0, 1, 1]])
texture_coords = np.array([[0, 0], [1, 0], [1, 1], [0, 1], [0, 0], [1, 0], [1, 1], [0, 1]])

# 绘制前面
ax.add_collection3d(plt.Polygon(vertices[[0, 1, 2, 3]], alpha=0.5))
# 绘制后面
ax.add_collection3d(plt.Polygon(vertices[[4, 5, 6, 7]], alpha=0.5))
# 绘制左侧面
ax.add_collection3d(plt.Polygon(vertices[[0, 3, 7, 4]], alpha=0.5))
# 绘制右侧面
ax.add_collection3d(plt.Polygon(vertices[[1, 2, 6, 5]], alpha=0.5))
# 绘制顶部面
ax.add_collection3d(plt.Polygon(vertices[[0, 1, 5, 4]], alpha=0.5))
# 绘制底部面
ax.add_collection3d(plt.Polygon(vertices[[2, 3, 7, 6]], alpha=0.5))

ax.plot_surface(texture_coords[:, 0], texture_coords[:, 1], np.zeros((8, 1)), facecolors=rgb_array / 255.0)

ax.set_xlim([0, 1])
ax.set_ylim([0, 1])
ax.set_zlim([0, 1])

plt.show()