построить интерполированную поверхность в python - PullRequest
Новая
       86

построить интерполированную поверхность в python

0 голосов
/ 03 августа 2020

У меня есть матрица 4x4 (test4x4), которую я хочу интерполировать в матрицу 8x8 (test8x8). Я использовал interpolate.interp2d для интерполяции, но когда я рисую его (test8x8), это не похоже на график test4x4. Где я ошибаюсь? 

import numpy as np
from scipy import interpolate
import matplotlib.pyplot as plt
from mpl_toolkits.mplot3d import Axes3D

mymin,mymax = 0,3
X = np.linspace(mymin,mymax,4)
Y = np.linspace(mymin,mymax,4)

x,y = np.meshgrid(X,Y)

test4x4 = np.array([[ 1.2514318 ,  1.25145821,  1.25148472,  1.25151133],
   [ 1.25087456,  1.25090105,  1.25092764,  1.25095435],
   [ 1.25031581,  1.25034238,  1.25036907,  1.25039586],
   [ 1.24975557,  1.24978222,  1.24980898,  1.24983587]])

f = interpolate.interp2d(x,y,test4x4,kind='cubic')

# use linspace so your new range also goes from 0 to 3, with 8 intervals
Xnew = np.linspace(mymin,mymax,8)
Ynew = np.linspace(mymin,mymax,8)

test8x8 = f(Xnew,Ynew)

print('test8x8=',test8x8)

plot1=plt.figure(1)
plt.title('test 4X4')
fig1 = plt.figure(1)
ax1 = fig1.gca(projection='3d')
ax1.plot_surface(x.T,y.T, test4x4, alpha = 1, rstride=1, cstride=1, linewidth=0.5, antialiased=True, zorder = 0.5)
plt.xlabel('x')
plt.ylabel('y')
plt.grid()

ax1.plot_surface(Xnew.T, Ynew.T, test8x8, alpha = 1, rstride=1, cstride=1, linewidth=0.5, antialiased=True, zorder = 0.5)
plt.grid()

plt.show()

1

Думаю, я смогу решить эту проблему, надо использовать x1,y1 = np.meshgrid(Xnew,Ynew).

1 Ответ

1 голос
/ 03 августа 2020

Для 8x8 вам также понадобится сетка: 

...
Xnew = np.linspace(mymin,mymax,8)
Ynew = np.linspace(mymin,mymax,8)
xx, yy = np.meshgrid(Xnew, Ynew)  #You need this

И используйте эту сетку для построения 

ax1.plot_surface(xx.T, yy.T, test8x8, alpha=0.5, rstride=1, cstride=1, \
                   linewidth=0.5, antialiased=True, zorder = 10)

Используйте alpha=0.5 в обоих plot_surface(), чтобы вы может видеть обе поверхности.

Чтобы разделить 2 поверхности более отчетливо, вы можете попробовать вторую .plot_surface() как 

ax1.plot_surface(xx.T, yy.T, 0.0001+test8x8, alpha=0.5, rstride=1, cstride=1, \
                   linewidth=0.5, antialiased=True, zorder = 10)

Значение 0,0001 увеличивает вторую поверхность (в направлении оси Z). ).

...