我正在编写一个用于在 matplotlib 中制作 3D 散点图的接口,并且我想从 python 脚本访问数据。对于二维散点图,我知道过程是:
import numpy as np
from matplotlib import pyplot as plt
fig = plt.figure()
ax = fig.add_subplot(111)
h = ax.scatter(x,y,c=c,s=15,vmin=0,vmax=1,cmap='hot')
data = h.get_offsets()
Run Code Online (Sandbox Code Playgroud)
通过上面的代码,我知道 data 将是一个(N,2)填充了我的数据的 numpy 数组(x,y)。当我尝试对 3D 数据执行相同的操作时:
import numpy as np
from matplotlib import pyplot as plt
from mpl_toolkits.mplot3d import Axes3D
fig = plt.figure()
ax = Axes3D(fig)
h = ax.scatter(x,y,z,c=c,s=15,cmap='hot',vmin=0,vmax=1)
data = h.get_offsets()
Run Code Online (Sandbox Code Playgroud)
结果data变量仍然是(N,2)numpy 数组而不是(N,3)numpy 数组。的内容data不再与我的任何输入数据匹配;我假设其中data填充了 3D …