使用 pyplot 在 Python 2.7 上进行曲面绘图

Question

我是Python的新手。我一直在尝试绘制一个包含 3 列和 1024 个数据点的数据文件。运行代码时，出现以下错误：

Traceback (most recent call last):
File "plot-data.py", line 27, in <module>
linewidth=0, antialiased=False)
File "/home/ritajit/.local/lib/python2.7/site-packages/mpl_toolkits/mplot3d/axes3d.py", line 1624, in plot_surface
X, Y, Z = np.broadcast_arrays(X, Y, Z)
File "/home/ritajit/.local/lib/python2.7/site-packages/numpy/lib/stride_tricks.py", line 249, in broadcast_arrays
shape = _broadcast_shape(*args)
File "/home/ritajit/.local/lib/python2.7/site-packages/numpy            /lib/stride_tricks.py", line 184, in _broadcast_shape
b = np.broadcast(*args[:32])
ValueError: shape mismatch: objects cannot be broadcast to a single shape

我的代码看起来像这样

import numpy as np
import matplotlib.pyplot as plt
from mpl_toolkits.mplot3d import Axes3D
from matplotlib.mlab import griddata
import matplotlib.cm as cm    
from pylab import rcParams
rcParams['figure.figsize'] = 9, 9

## 3D surface_plot
fig = plt.figure()
axes = fig.add_subplot(111, projection='3d') #gca = get current axis
data = np.loadtxt('2D-data.txt')
x = data[:,0]
y = data[:,1]
z = data[:,2]
xi = np.unique(x)
yi = np.unique(y)
xv, yv = np.meshgrid(x,y)
Z = griddata(x, y, z, xi, yi, interp='linear')
# surface_plot with color grading and color bar
p = axes.plot_surface(xv,yv,Z, rstride=4, cstride=4, cmap=cm.RdBu, 
linewidth=0, antialiased=False)
fig.colorbar(p, shrink=0.5)
axes.set_xlabel('$x$',fontsize=15)
axes.set_ylabel('$y$',fontsize=15)
axes.set_zlabel('$z$',fontsize=15)
plt.tight_layout()
fig.savefig("surface.pdf")
plt.show()

我无法解决这个问题。
我做错了什么？
有没有其他方法可以绘制 3D 数据文件？

我的数据文件中的几行：

1 2 1.30884
2 2 1.30925
3 2 1.30974
4 2 1.30841
5 2 1.30864
6 2 1.30795

第 1、2、3 列分别为 x，y，z

Answer 1

这里有三个主要问题：

• 您需要对唯一值进行网格，而不是原始值

  xi = np.unique(x)
  yi = np.unique(y)
  xv, yv = np.meshgrid(xi,yi)
  

• 您需要对网格化值进行插值

  griddata(x, y, z, xv, yv)
  

• 你需要绘制Z，而不是z

  p = axes.plot_surface(xv,yv,Z)
  

总的来说，看起来您可以通过重塑数据列来实现几乎相同的目标(但小数据摘录不足以判断这一点(。

最后，matplotlib.mlab.griddata将在下一个版本中弃用。作为替代方案，请考虑scipy.interpolate.griddata.另请查看不规则间隔数据的等值线图示例。