numpy 带有标签和球体的3D散点图

t2a7ltrp 于 2023-04-30 发布在其他

关注(0)|答案(1)|浏览(180)

我试图用一些数据点（x，y，z，半径）绘制散点图，这是我到目前为止的结果：

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

x = np.random.rand(20)
y = np.random.rand(20)
z = np.random.rand(20)
r = np.random.rand(20)

plt.rc('text', usetex=True)
plt.rcParams['text.latex.preamble']=[r"\usepackage{amsmath}"]

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

ax.scatter(x, y, z, s=np.pi*r**2*100, c='blue', alpha=0.75)

ax.set_xlabel(r'$x$ $\left[\frac{\text{Mpc}}{h}\right]$')
ax.set_ylabel(r'$y$ $\left[\frac{\text{Mpc}}{h}\right]$')
ax.set_zlabel(r'$z$ $\left[\frac{\text{Mpc}}{h}\right]$')

#plt.savefig('spheres.png')

plt.show()

如何改进此图，使x和y标签与控制点不重叠？
是否有可能在这个3D图中制作球体而不是区域？

numpy

来源：https://stackoverflow.com/questions/24123659/scatter-plot-3d-with-labels-and-spheres

1条答案

按热度按时间

c0vxltue1#

您可以创建使用球体而不是圆形标记的打印，方法是在每个位置绘制一个标记，如here所述。下面是一个例子：

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

def drawSphere(xCenter, yCenter, zCenter, r):
    #draw sphere
    u, v = np.mgrid[0:2*np.pi:20j, 0:np.pi:10j]
    x=np.cos(u)*np.sin(v)
    y=np.sin(u)*np.sin(v)
    z=np.cos(v)
    # shift and scale sphere
    x = r*x + xCenter
    y = r*y + yCenter
    z = r*z + zCenter
    return (x,y,z)

x = 10*np.random.rand(20)
y = 10*np.random.rand(20)
z = 10*np.random.rand(20)
r = np.random.rand(20)

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

# draw a sphere for each data point
for (xi,yi,zi,ri) in zip(x,y,z,r):
    (xs,ys,zs) = drawSphere(xi,yi,zi,ri)
    ax.plot_wireframe(xs, ys, zs, color="r")

plt.show()

要解决标签定位问题，请尝试添加第二行，如here所述。

赞(0）回复(0）举报 2023-04-30

我来回答

numpy 带有标签和球体的3D散点图

1条答案

相关问题

热门标签

最新问答