scipy.interpolate 中的 bisplrep 和 bisplev 問題

Question

我收到來自scipy.interpolate模塊的bisplrep和bisplev的一些非常奇怪的行為。 I try to reproduce the example from the scipy homepage ( https://docs.scipy.org/doc/scipy/reference/tutorial/interpolate.html ). 我有一些 function f(x,y)計算我一些 z 值，然后我使用bisplrep進行插值。 如果我使用bisplev和 plot 重新計算數據，則結果值顯示旋轉了大約 90°。 即使我交換 x 和 y 值 plot 仍然旋轉。 有人可以告訴我我在這里做錯了什么嗎？ 以下代碼應該足以重現錯誤。 我正在使用最新版本的 scipy，錯誤發生在 Jupyter、Spyder 和 IDLE 中。

import matplotlib.pyplot as plt
import numpy as np
from scipy import interpolate

def f(x, y):
    return x**2 + y**2

x, y = np.linspace(0, 5, 15), np.linspace(-2*np.pi, 2*np.pi, 15)
xx, yy = np.meshgrid(x, y)
zz = f(xx, yy)

tck = interpolate.bisplrep(xx, yy, zz)

plt.pcolor(xx, yy, zz)

x_new, y_new = np.linspace(0, 5, 100), np.linspace(-2*np.pi, 2*np.pi, 100)

z_new = interpolate.bisplev(x_new, y_new, tck)

plt.figure()    
plt.pcolor(x_new, y_new, z_new)

plt.figure()    
plt.pcolor(y_new, x_new, z_new)

plt.show()

Answer 1

我對網格使用了另一個定義。 現在，它應該可以工作了：

import matplotlib.pyplot as plt
import numpy as np
from scipy import interpolate

def f(x, y):
    return x**2 + y**2

#x, y = np.linspace(0, 5, 15), np.linspace(-2*np.pi, 2*np.pi, 15)
xx, yy = np.mgrid[0:5:15j, -2*np.pi:2*np.pi:15j]
zz = f(xx, yy)

tck = interpolate.bisplrep(xx, yy, zz)

plt.pcolor(xx, yy, zz)

#x_new, y_new = np.linspace(0, 5, 100), np.linspace(-2*np.pi, 2*np.pi, 100)
xx_new, yy_new = np.mgrid[0:5:100j, -2*np.pi:2*np.pi:100j]

zz_new = interpolate.bisplev(xx_new[:,0], yy_new[0,:], tck)

plt.figure()    
plt.pcolor(xx_new, yy_new, zz_new)

plt.figure()    
plt.pcolor(yy_new, xx_new, zz_new)

plt.show()