我有一个包含5个数字的数组:
A = [10, 20, 40, 80, 110]
我需要创建一个长度为10的新数组。
额外的数字可以是两个A
之间的平均数。
例如: 编辑 B = [10,15,20,30,40,60,80,95,110]
是否可以使用scipy或numpy函数?
答案 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]