在 numpy/matplotlib 中解决插值数据中的零

Question

我有一些我有兴趣分析的二维范围内的数据。这些数据最初位于列表 x、y 和 z 中，其中 z[i] 是位于 (x[i],y[i]) 处的点的值。然后我使用

将这些数据插入到常规网格中

x=np.array(x)
y=np.array(y)
z=np.array(z)
xi=np.linspace(minx,maxx,100)
yi=np.linspace(miny,maxy,100)
zi=griddata(x,y,z,xi,yi)

然后我使用

绘制了 xi,yi,zi 数据

plt.contour(xi,yi,zi)
plt.pcolormesh(xi,yi,zi,cmap=plt.get_cmap('PRGn'),norm=plt.Normalize(-10,10),vmin=-10,vmax=10)

这产生了这个情节：

在此图中，您可以看到值为零的 S 形曲线（此外：数据的变化没有颜色栏中显示的那么快——这只是我将数据标准化为 - 10-10 当它实际上远远超出该范围时；我这样做是为了使零值区域显示得更好——也许还有更好的方法来做到这一点......）。

散点只是我拥有原始数据的点（是的，在这种情况下，我的数据已经在常规网格上）。我很好奇是否有一种好方法可以让我提取曲线为零的值并获得 x,y 对，如果绘制为一条线，将在颜色网格中追踪该零区域。我可以插值到一个非常精细的网格，然后强力搜索最接近零的值。但是有没有更自动的方式来做到这一点，或者更自动的方式来绘制这个“零线”？

还有一个次要问题：我正确使用了 griddata，对吗？我有这些简单的一维数组，尽管在其他地方人们在调用 griddata 之前会使用各种网格网格、加载文本等。

Answer 1

这是一个完整的例子：

import numpy as np
import matplotlib.pyplot as plt
y, x = np.ogrid[-1.5:1.5:200j, -1.5:1.5:200j]
f = (x**2 + y**2)**4 - (x**2 - y**2)**2

plt.figure(figsize=(9,4))
plt.subplot(121)
extent = [np.min(x), np.max(x), np.min(y), np.max(y)]
cs = plt.contour(f, extent=extent, levels=[0.1], 
     colors=["b", "r"], linestyles=["solid", "dashed"], linewidths=[2, 2])

plt.subplot(122)
# get the points on the lines
for c in cs.collections:
    data = c.get_paths()[0].vertices
    plt.plot(data[:,0], data[:,1], 
        color=c.get_color()[0],  linewidth=c.get_linewidth()[0])

plt.show()

这是输出：