如何基于给定NumPy数组生成对应长度的连续递增数值序列

解决方案 #

你需要生成与输入numpy数组x长度一致、从1开始递增的连续整数序列，针对10万级别的数组规模，直接使用numpy内置方法即可高效实现，无性能压力。

实现代码

import numpy as np

# 你的输入数组示例
x = np.array([0,2,9,6,1,3,6,2,8,6,10])
# 生成目标递增序列
y = np.arange(start=1, stop=x.size + 1)

输出验证

执行print(y)即可得到你期望的结果：

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

说明

• x.size是numpy数组的内置属性，读取数组元素总数的速度比通用的len(x)更快，处理超大数组时优势更明显
• np.arange直接生成numpy数组，不需要额外类型转换，10万级数据量下生成耗时仅为微秒级，完全满足你的使用要求

内容的提问来源于stack exchange，提问作者Starbucks