我有一个圆形数据网格,例如数据是从0
到π
的角度给出的。在这些数据中,我还有另一个较小的网格。
这可能看起来像这样:
我想要做的是在红点上插入黑色数据。因此,我正在使用scipy.interpolate.griddata
。这将给我以下结果:
如您所见,当角度从“几乎0
”变为“几乎π
”时,会出现间断。
为消除这一点,我尝试在插值之前解开数据。根据这个答案(here)。我得到了这个(更好的)结果,但是令人惊讶的是,右边有一个我不理解的新矛盾。
所以我的问题是:如何使用np.unwrap
获得连续插值?还是有更好的方法呢?
以下是要复制的代码:
import numpy as np
from matplotlib import pyplot as plt
from scipy.interpolate import griddata
ax = plt.subplot()
ax.set_aspect(1)
# Simulate some given data.
x, y = np.meshgrid(np.linspace(-10, 10, 20), np.linspace(-10, 10, 20))
phi = np.arctan2(y, x) % (2 * np.pi)
data = np.arctan2(np.cos(phi), np.sin(phi)) % np.pi
# Plot data.
u = np.cos(data)
v = np.sin(data)
ax.quiver(x, y, u, v, headlength=0.01, headaxislength=0, pivot='middle', units='xy')
# Create a smaller grid within.
x1, y1 = np.meshgrid(np.linspace(-6, 5, 20), np.linspace(-4, 8, 25))
# ax.plot(x1, y1, '.', color='red', markersize=2)
# Prepare data.
data = np.unwrap(2 * data) / 2
# Interpolate data on grid.
interpolation = griddata((x.flatten(), y.flatten()), data.flatten(), (x1.flatten(), y1.flatten()))
# Plot interpolated data.
u1 = np.cos(interpolation)
v1 = np.sin(interpolation)
ax.quiver(x1, y1, u1, v1, headlength=0.01, headaxislength=0, pivot='middle', units='xy',
scale=3, width=0.03, color='red')
plt.show()
答案 0 :(得分:1)
要在circular quantities上正确运行,请在调用pom.xml
之前将角度转换为复数,然后再转换回角度:
griddata
2的因数会将您的[0,π)扩展到整个圆并再次返回。请注意,对c=np.exp(2j*data) # 0,pi -> 1
# …
a=np.angle(interpolation)/2
的调用中隐含的归一化对输入数据在输入数据的一个“网格单元”内变化太大的输入非常敏感。