Как добавить цветовую карту в 3D-график matplotlib

Первая проблема здесь заключается в том, что количество записей в аргументе facecolors должно совпадать с количеством граней, иначе похоже, что matplotlib возвращается к выбору первого цвета в списке — здесь это черный.

Вторая проблема заключается в том, что вы пытаетесь создать прямоугольный параллелепипед всего с 6 гранями, а затем раскрасить его с помощью гладкой карты цветов. Многоугольники в matplotlib могут быть окрашены только одним цветом каждый. Итак, вы должны разделить свой кубоид.

Вот моя попытка решить эти две проблемы! Я создаю набор точек от 0 до 10, а затем создаю вершины в этих x-позициях. Я также строю массив x-позиций непосредственно из граней, взяв среднее значение x-позиции каждой из четырех вершин, составляющих каждую грань.

import matplotlib.pyplot as plt
import numpy as np
from mpl_toolkits.mplot3d.art3d import Poly3DCollection

# Create a 3D figure
fig = plt.figure()
ax = fig.add_subplot(111, projection='3d')

# Define the vertices of the rectangular rod
N_sub = 50  # Set higher for better resolution; lower for better performance!
vertices = np.concatenate([
    np.array([[x, -0.5, -0.5],
              [x, -0.5, 0.5],
              [x, 0.5, 0.5],
              [x, 0.5, -0.5]])
    for x in np.linspace(0, 10, N_sub + 1)], axis=0)

# Define the faces of the rectangular rod
offset = (N_sub) * 4
faces = np.concatenate([
    [[0, 1, 2, 3]],
    *[np.array([[3, 0, 4, 7],
                [0, 1, 5, 4],
                [1, 2, 6, 5],
                [2, 3, 7, 6]]) + 4 * x for x in range(0, N_sub)],
    [[0 + offset, 1 + offset, 2 + offset, 3 + offset]]
], axis=0)

# Map the x-coordinate of each face to a color in the colormap
x = np.array([np.mean([vertices[idx, 0] for idx in face]) for face in faces])
norm = plt.Normalize(x.min(), x.max())
cmap = plt.get_cmap("gnuplot")
colors = cmap(norm(x))

# Create a Poly3DCollection object with the vertices, faces, and colors
rect = Poly3DCollection(vertices[faces], alpha=0.8, facecolors=colors)

# Add the rectangular rod to the plot
ax.add_collection3d(rect)

# Set the limits of the x, y, and z axes
ax.set_xlim([0, 10])
ax.set_ylim([-1, 1])
ax.set_zlim([-1, 1])

# Add labels to the axes
ax.set_xlabel('X')
ax.set_ylabel('Y')
ax.set_zlabel('Z')

# Show the plot
plt.show()

Надеюсь это поможет! Однако я не уверен, что смогу воспроизвести черные края. Я попытался сделать второй прямоугольный параллелепипед, как ваш оригинальный, с шестью гранями и facecolor = "none", но это выдало ошибку.

PS: я сохранил вам импорт с помощью plt.get_cmap.