在 Numpy 数组中添加以前和当前的索引 Python
Adding previous and current index in Numpy Arrays Python
如何在不使用 for 循环的情况下编写一个添加到数组的前一个索引的函数。因此,对于预期输出中的第二个值,它将是先前索引 (2
) 加上当前索引 (5
),结果是 7
。我怎么会做这种事?
import numpy as np
A = np.array([2,5,44,-12,3,-5])
预期输出:
[2,7,51,39,42,37]
给你:
np.cumsum(A)
打印:
array([ 2, 7, 51, 39, 42, 37], dtype=int32)
如何在不使用 for 循环的情况下编写一个添加到数组的前一个索引的函数。因此,对于预期输出中的第二个值,它将是先前索引 (2
) 加上当前索引 (5
),结果是 7
。我怎么会做这种事?
import numpy as np
A = np.array([2,5,44,-12,3,-5])
预期输出:
[2,7,51,39,42,37]
给你:
np.cumsum(A)
打印:
array([ 2, 7, 51, 39, 42, 37], dtype=int32)