Постройте трехмерные точки (x, y, z) на двухмерном графике с помощью цветной полосы
Я вычислил много (~ 5000) трехмерных точек (x, y, z) довольно сложным способом, поэтому у меня нет такой функции, что z = f (x, y). Я могу построить 3D-поверхность, используя
import numpy as np
import matplotlib.pyplot as plt
from matplotlib import cm
X = surface_points[:,0]
Y = surface_points[:,1]
Z = surface_points[:,2]
fig = plt.figure()
ax = fig.add_subplot(projection='3d')
surf = ax.plot_trisurf(X, Y, Z, cmap=cm.coolwarm, vmin=np.nanmin(Z), vmax=np.nanmax(Z))
Я хотел бы построить это также в 2d, с цветовой полосой, указывающей значение z. Я знаю, что есть простое решение с использованием ax.contour, если мой z является матрицей, но здесь у меня есть только вектор.
Прикрепление результата plot_trisurf при повороте на плоскость xy. Это то, чего я хотел бы добиться, не вращая 3D-график. В этом моя переменная surface_points представляет собой np.array размером 5024 x 3.
Ответы 2
У меня были те же проблемы в одном из моих кодов, я решил это так:
import numpy as np
from scipy.interpolate import griddata
import matplotlib.pylab as plt
from matplotlib import cm
N = 10000
surface_points = np.random.rand(N,3)
X = surface_points[:,0]
Y = surface_points[:,1]
Z = surface_points[:,2]
nx = 10*int(np.sqrt(N))
xg = np.linspace(X.min(), X.max(), nx)
yg = np.linspace(Y.min(), Y.max(), nx)
xgrid, ygrid = np.meshgrid(xg, yg)
ctr_f = griddata((X, Y), Z, (xgrid, ygrid), method='linear')
fig = plt.figure()
ax = fig.add_subplot(1, 1, 1)
ax.contourf(xgrid, ygrid, ctr_f, cmap=cm.coolwarm)
plt.show()
... и я только что научился загружать цифры в stackoverflow ;-)
Вы можете использовать точечную диаграмму для отображения проекции вашего цвета z на ось xy.
import numpy as np
import matplotlib.pyplot as plt
from matplotlib import cm
N = 10000
surface_points = np.random.rand(N,3)
X = surface_points[:,0]
Y = surface_points[:,1]
Z = surface_points[:,2]
# fig = plt.figure()
# ax = fig.add_subplot(projection='3d')
# surf = ax.plot_trisurf(X, Y, Z, cmap=cm.coolwarm, vmin=np.nanmin(Z), vmax=np.nanmax(Z))
fig = plt.figure()
cmap = cm.get_cmap('coolwarm')
color = cmap(Z)[..., :3]
plt.scatter(X,Y,c=color)
plt.show()
Поскольку кажется, что у вас есть полая трехмерная фигура, вы можете разделить проекцию на две части, как если бы вы вырезали фигуру из двух частей.
fig = plt.figure()
plt.subplot(121)
plt.scatter(X[Z<0.5],Y[Z<0.5],c=color[Z<0.5])
plt.title('down part')
plt.subplot(122)
plt.scatter(X[Z>=0.5],Y[Z>=0.5],c=color[Z>+0.5])
plt.title('top part')
plt.show()
Большое спасибо за ответ, ответ ниже дал мне именно то, что я имел в виду, но это также было очень полезно.
Потрясающе! Большое спасибо! С этой небольшой модификацией цветовой панели proj = ax.contourf(xgrid, ygrid, ctr_f, cmap=cm.coolwarm) fig.colorbar(proj, shrink=0.5, aspect=5) я получил именно то, на что надеялся.