gpt4 book ai didi

python - 在 Python 中创建数组的最快方法

转载 作者:太空宇宙 更新时间:2023-11-04 05:59:17 26 4
gpt4 key购买 nike

<分区>

我想用 Python 创建一个 3D 数组,填充 -1。

我测试了这些方法:

import numpy as np
l = 200
b = 100
h = 30

%timeit grid = [[[-1 for x in range(l)] for y in range(b)] for z in range(h)]
1 loops, best of 3: 458 ms per loop

%timeit grid = -1 * np.ones((l, b, h), dtype=np.int)
10 loops, best of 3: 35.5 ms per loop

%timeit grid = np.zeros((l, b, h), dtype=np.int) - 1
10 loops, best of 3: 31.7 ms per loop

%timeit grid = -1.0 * np.ones((l, b, h), dtype=np.float32)
10 loops, best of 3: 42.1 ms per loop

%%timeit
grid = np.empty((l,b,h))
grid.fill(-1.0)
100 loops, best of 3: 13.7 ms per loop

很明显,最后一个是最快的。有人有更快的方法或至少更少的内存占用吗?因为它在 RaspberryPi 上运行。

26 4 0
Copyright 2021 - 2024 cfsdn All Rights Reserved 蜀ICP备2022000587号
广告合作:1813099741@qq.com 6ren.com