使用step高效创建numpy N数组

Question

嗨我需要在一定范围内更快地创建N维numpy数组。原始vectors_number将超过10 ** 6

我的工作代码：

vectors_number = 12

data = []
for i in range(0, vectors_number, 4):
    data.append(np.arange(0+i, 4+i,1))

out = np.asarray(data)
print(out)

输出：

[[ 0  1  2  3]
 [ 4  5  6  7]
 [ 8  9 10 11]]

Answer 1

许多numpy示例都是从创建此类数组开始的。我鼓励你稍微浏览一下文档！

通过在1D中取整个范围然后重塑它来有效地完成它：

np.arange(vectors_number).reshape((-1, 4))