gpt4 book ai didi

python - numpy 重复元素

转载 作者:太空狗 更新时间:2023-10-30 01:02:43 28 4
gpt4 key购买 nike

也许这是一个简单的问题,但是有没有一种快速复制数组中元素的方法?对于 3D,它应该像这样工作:

1 2 3
4 5 6
7 8 9

1 1 2 2 3 3
1 1 2 2 3 3
4 4 5 5 6 6
4 4 5 5 6 6
7 7 8 8 9 9
7 7 8 8 9 9

我尝试了 3 个嵌套的 for 循环,但这真的很慢。

最佳答案

>>> a = np.array([[1, 2, 3],
[4, 5, 6],
[7, 8, 9]])
>>> np.repeat(np.repeat(a, 2, 0), 2, 1)

array([[1, 1, 2, 2, 3, 3],
[1, 1, 2, 2, 3, 3],
[4, 4, 5, 5, 6, 6],
[4, 4, 5, 5, 6, 6],
[7, 7, 8, 8, 9, 9],
[7, 7, 8, 8, 9, 9]])

关于python - numpy 重复元素,我们在Stack Overflow上找到一个类似的问题: https://stackoverflow.com/questions/13969002/

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