主页 / 问题 / 1439786
Accepted
Zhihar
Asked:2022-08-16 17:27:33 +0000 UTC2022-08-16 17:27:33 +0000 UTC

python：写入具有给定第一步值的列表

  • 772

需要arr将值value从位置start写入列表step

告诉我如何以最有效的方式做到这一点。

我用这种方式：

arr[start::step] = [value] * (len(arr) - start - 1)

但是有没有更有效的方法？

python python-3.x

1 个回答

  1. Best Answer

    对于足够大的数组，Numpy它似乎加速得很好。

    简单Python：

    arr = list(range(10000))

start = 10
step = 5
value = 42

%timeit arr[start::step] = [value] * ((len(arr) - start - 1 + step)//step)

# 16.6 µs ± 1.08 µs per loop (mean ± std. dev. of 7 runs, 10000 loops each)

    您不需要Numpy乘以该值：

    import numpy as np

arr = np.arange(10000)

start = 10
step = 5
value = 42

%timeit arr[start::step] = value

# 2.87 µs ± 580 ns per loop (mean ± std. dev. of 7 runs, 100000 loops each)
    • 3

相关问题