matplotlib 使用颜色条在2d图中绘制3d点(x,y,z)

7kjnsjlb  于 2023-05-01  发布在  其他
关注(0)|答案(2)|浏览(164)

我已经计算了很多(~5000)3d点(x,y,z)在一个相当复杂的方式,所以我没有这样的函数z = f(x,y)。我可以使用

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值。我知道有一个简单的解决方案使用斧头。如果我的z是一个矩阵,但这里我只有一个向量。
旋转到xy平面时附加plot_trisurf结果。这就是我想实现的,而不必旋转3D图。在这里,我的变量surface_points是一个np。数组大小为5024 x 3。

ppcbkaq5

ppcbkaq51#

我在一个代码中遇到了同样的问题,我用这种方法解决了:

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()

31moq8wy

31moq8wy2#

可以使用散点图显示z颜色在x-y轴上的投影。

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()

因为你似乎有一个中空的3D形状,你可以把投影分成两部分,就像你把形状分成两部分一样。

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()

相关问题