python - 使用零和五创建一个 numpy 数组 (10x1)

Question

我无法弄清楚如何创建一个 10x1 numpy 数组，其中前 3 个元素的数字为 5，而其他 7 个元素的数字为 0。关于如何有效地执行此操作有什么想法吗？

Answer 1

最简单的似乎是:

import numpy as np

the_array = np.array([5]*3 + [0]*7)

这种简单的方法是否对您的目的有一些特定的不利影响？

当然还有很多选择，比如

the_array = np.zeros((10,))
the_array[:3] = 5

如果您需要多次重复此特定操作，那么速度上的微小差异很重要，您可以对各种方法进行基准测试，看看哪里可以节省纳秒左右的时间。但是这个前提不太可能，我不建议针对这个特定问题这样做，即使我是 timeit:-) 的忠实粉丝。