Изменить цвет в contour3D

Question

Я написал этот простой код для трехмерного графика двух функций. Как я могу изменить цвет моих участков?

from mpl_toolkits import mplot3d
import matplotlib.pyplot as plt

def f1(x,y):
    return 454.586-x-y
def f2(x,y):
    return 293.53-x
import numpy as np
x = np.linspace(70, 6, 300)
y = np.linspace(70, 6, 300)
X, Y = np.meshgrid(x, y)
Z = f1(X, Y)
ZZ = f2(X, Y)
fig = plt.figure()
ax = plt.axes(projection='3d')
ax.contour3D(X, Y, Z, 50, cmap='binary')
ax.contour3D(X, Y, ZZ, 50, cmap='binary')
ax.set_xlabel('x')
ax.set_ylabel('y')
ax.set_zlabel('z')
plt.show()

Answer 1

Вы можете изменить цвет в аргументе " cmap " функции contour3D .

Например:

from mpl_toolkits import mplot3d
import matplotlib.pyplot as plt

def f1(x,y):
    return 454.586-x-y
def f2(x,y):
    return 293.53-x
import numpy as np
x = np.linspace(70, 6, 300)
y = np.linspace(70, 6, 300)
X, Y = np.meshgrid(x, y)
Z = f1(X, Y)
ZZ = f2(X, Y)
fig = plt.figure()
ax = plt.axes(projection='3d')
ax.contour3D(X, Y, Z, 50, cmap='viridis')
ax.contour3D(X, Y, ZZ, 50, cmap='plasma')
ax.set_xlabel('x')
ax.set_ylabel('y')
ax.set_zlabel('z')
plt.show()

Здесь у вас есть список цветов