python - 使用 numpy 或其他库在 python 中进行列表扩充

Question

我想从

[1, 2, 3, 4, 5]

到

[1, 1, 2, 2, 3, 3, 4, 4, 5, 5]

如果我想同样增加 n 次（比如 100 或 500 次），我该怎么做？我不想用常规循环来做，而是使用像 numpy 这样的库。有什么帮助吗？

非常感谢。

Answer 1

你可以用 numpy's 做到这一点np.repeat：

import numpy as np
a = np.array([1, 2, 3, 4, 5])
np.repeat(a,2)
# array([1, 1, 2, 2, 3, 3, 4, 4, 5, 5])