2012-08-09 44 views
5

yo escribimos un código que realiza una interpolación spline:Python Mantenga los puntos de interpolación spline

x1 = [ 0., 13.99576991, 27.99153981, 41.98730972, 55.98307963, 69.97884954, 83.97461944, 97.97038935, 111.9661593, 125.9619292, 139.9576991, 153.953469 ] 
y1 = [ 1., 0.88675318, 0.67899118, 0.50012243, 0.35737022, 0.27081293, 0.18486778, 0.11043095, 0.08582272, 0.04946131, 0.04285015, 0.02901567] 

x = np.array(x1) 
y = np.array(y1) 

# Interpolate the data using a cubic spline to "new_length" samples 
new_length = 50 
new_x = np.linspace(x.min(), x.max(), new_length) 
new_y = sp.interpolate.interp1d(x, y, kind='cubic')(new_x) 

Pero en el nuevo conjunto de datos generados new_x y new_y los puntos originales se eliminan, sólo el primero y los últimos valores se mantienen . Me gustaría mantener los puntos originales.

+1

¿por qué no haces entonces 'new_y = sp.interpolate.interp1d (x, y, kind = 'cubic') (x)' –

Respuesta

6

derecho, linspace no generará cualquiera de los valores en x excepto los que se pasan a ella (x.min() y x.max()).

no tengo una gran respuesta ágil, pero aquí es una manera de hacerlo:

# Interpolate the data using a cubic spline to "new_length" samples 
new_length = 50 
interpolated_x = np.linspace(x.min(), x.max(), new_length - len(x) + 2) 
new_x = np.sort(np.append(interpolated_x, x[1:-1])) # include the original points 
new_y = sp.interpolate.interp1d(x, y, kind='cubic')(new_x) 

Este código utiliza:

  • np.linspace para crear tantos puntos extra, ya que necesitamos
  • np.append para combinar la matriz de puntos extra con los puntos originales de x
  • np.sort poner el conjunto combinado en orden
+0

¡Muchas gracias! Funcionó. – Hellfish

Cuestiones relacionadas