ZeroDivisionError：表面图代码中的浮点除法零

Question

我有这个代码来生成一个曲面图。但它给出了零除法错误。我无法弄清楚出了什么问题。谢谢你。

import pylab, csv
import numpy
from mayavi.mlab import *

def getData(fileName):
    try:
        data = csv.reader(open(fileName,'rb'))
    except:
        print 'File not found'
    else:
        data = [[float(row[0]), float(row[1]),float(row[2])] for row in data]
        x = [row[0] for row in data]
        y = [row[1] for row in data]
        z = [row[2] for row in data]
    return (x, y, z)

def plotData(fileName):
    xVals, yVals, zVals = getData(fileName)
    xVals = pylab.array(xVals)   
    yVals = pylab.array(yVals)
    zVals = (pylab.array(zVals)*10**3)
    x, y = numpy.mgrid[-0.5:0.5:0.001, -0.5:0.5:0.001]  
    s = surf(x, y, zVals)
    return s

plotData('data')   

Answer 1

如果我正确理解了代码，那么mayavi.mlab.surf 中的zVals 有问题。

根据函数的文档，s 是高程矩阵，一个 2D 数组，其中沿第一个数组轴的索引表示 x 位置，沿第二个数组轴的索引表示 y 位置。您的文件阅读器似乎返回一维向量而不是数组。

但是，这可能不是最困难的问题。您的文件似乎包含 x、y 和 z 坐标的三元组。仅当文件中的 x 和 y 坐标形成规则方形网格时，您才能使用 mayavi.mlab.surf。如果是这种情况，那么您只需要恢复该网格并形成所有三个部分的漂亮二维数组。如果这些点在文件中的顺序是已知的，那就容易了，否则就比较棘手了。

也许您想从mayavi.mlab.points3d(xVals, yVals, zVals) 开始。这将使您对数据有一个整体印象。 （或者，如果您已经了解更多关于您的数据的信息，您可以通过编辑您的问题并添加更多信息来给我们一个提示！）

---

只是为了让您了解可能略带 Python 风格的编写方式，您的代码被重写（并替换 surf）如下：

import mayavi.mlab as ml
import numpy

def plot_data(filename):
    data = numpy.loadtxt(filename)
    xvals = data[:,0]
    yvals = data[:,1]
    zvals = data[:,2] * 1000.
    return ml.points3d(x, y, z)

plot_data('data')   

（基本更改：numpy.loadtxt 的使用，此处去掉 pylab 命名空间，没有 import *，没有 CamelCase 变量或函数名称。更多信息，请参阅 PEP 8。）

---

如果您只需要查看曲面的形状，并且文件中的数据是逐行排序的，并且每行中的数据点数相同（即固定的列数），那么您可以使用：

import mayavi.mlab as ml
import numpy
importt matplotlib.pyplot as plt

# whatever you have as the number of points per row
columns = 13

data = numpy.loadtxt(filename)

# draw the data points into a XY plane to check that they really for a rectangular grid:
plt.plot(data[:,0], data[:,1])

# draw the surface
zvals = data[:,2].reshape(-1,columns)
ml.surf(zvals, warp_scale='auto')

如您所见，此代码允许您检查您的值是否真的在正确的网格中。它不会检查它们的顺序是否正确，但至少您可以看到它们形成了一个漂亮的网格。此外，您必须手动输入列数。关键字warp_scale 负责表面缩放，使其看起来合理。