[英]Double the length of a python numpy array with interpolated new values
我有一个5个数字的数组:
A = [10, 20, 40, 80, 110]
我需要创建一个长度为10的新数组。
额外的数字可以是两个A
之间的平均数。
例如:编辑B = [10,15,20,30,40,60,80,95,110]
是否可以使用scipy或numpy功能?
使用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.]
使用这个答案 :
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.])
你不会加倍,因为你只是两者之间的平均值。 正如@LevLevitsky在评论中指出的那样,你也失踪了40
。
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]
声明:本站的技术帖子网页,遵循CC BY-SA 4.0协议,如果您需要转载,请注明本站网址或者原文地址。任何问题请咨询:yoyou2525@163.com.