使用插值的新值加倍python numpy数组的长度

时间:2013-05-05 19:09:47

标签: python arrays numpy scipy

我有一个包含5个数字的数组:

A = [10, 20, 40, 80, 110]

我需要创建一个长度为10的新数组。

额外的数字可以是两个A之间的平均数。

例如: 编辑     B = [10,15,20,30,40,60,80,95,110]

是否可以使用scipy或numpy函数?

3 个答案:

答案 0 :(得分:8)

使用numpy.interp

import numpy as np
Y = [10, 20, 40, 80, 110]
N = len(Y)
X = np.arange(0, 2*N, 2)
X_new = np.arange(2*N-1)       # Where you want to interpolate
Y_new = np.interp(X_new, X, Y) 
print(Y_new)

产量

[  10.   15.   20.   30.   40.   60.   80.   95.  110.]

答案 1 :(得分:5)

使用this answer

In [1]: import numpy as np

In [2]: a = np.array([10, 20, 40, 80, 110])

In [3]: b = a[:-1] + np.diff(a)/2

In [4]: c = np.empty(2 * a.size -1)

In [5]: c[::2] = a

In [6]: c[1::2] = b

In [7]: c
Out[7]: array([  10.,   15.,   20.,   30.,   40.,   60.,   80.,   95.,  110.])

答案 2 :(得分:2)

你不会加倍,因为你只是两者之间的平均值。你也错过了40,因为@LevLevitsky在评论中指出了这一点。

import numpy as np
A = np.array([10, 20, 40, 80, 110])
avg = (A[:-1] + A[1:])/2

B = []
for x1, x2 in zip(A, avg):
    B.append(x1)
    B.append(x2)
B.append(A[-1])

B = np.array(B)
print B

给出:

[ 10  15  20  30  40  60  80  95 110]