python - 如何创建数字的累积和数组

Question

如何创建基于 x 的递增数字序列？我可能有 x 的数组，其中有 100,000 个值，所以我不能假设 x 是什么。

x = np.array([0,2,9,6,1,3,6,2,8,6,10])

所需数组：

print(y)
np.array([1,2,3,4,5,6,7,8,9,10,11])

标签： pythonarraysnumpy

Answer 1

我无法对我的代表发表评论，但我认为我们需要更多信息。你本质上是在寻找与长度相对应x但从 1 开始的数字范围吗？如果是这样，你可以这样做：

import numpy as np
x = np.array([0,2,9,6,1,3,6,2,8,6,10])
y = np.array([i+1 for i in range(len(x))])
print(y)

编辑：

@obchardon 在评论中提出了一个很好的观点。以下是您将如何使用np.arange()：

import numpy as np
x = np.array([0,2,9,6,1,3,6,2,8,6,10])
y = np.arange(1,len(x)+1)
print(y)