gpt4 book ai didi

python - 如何将 numpy 数组值复制到更高维度

转载 作者:太空狗 更新时间:2023-10-29 20:15:21 27 4
gpt4 key购买 nike

我有一个二维的 (w,h) np 数组。我想创建一个值大于 1 的 3d 维度,并沿第 3 个维度复制它的值。我希望广播能做到这一点,但事实并非如此。我就是这样做的

arr = np.expand_dims(arr, axis=2)
arr = np.concatenate((arr,arr,arr), axis=2)

有没有更快的方法?

最佳答案

您可以所有暗淡向前推,引入一个单独的暗淡/新轴作为最后一个暗淡来创建一个3D数组,然后沿着那个重复3次np.repeat , 像这样 -

arr3D = np.repeat(arr[...,None],3,axis=2)

这是使用 np.tile 的另一种方法 -

arr3D = np.tile(arr[...,None],3)

关于python - 如何将 numpy 数组值复制到更高维度,我们在Stack Overflow上找到一个类似的问题: https://stackoverflow.com/questions/39463019/

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