gpt4 book ai didi

python - 4-D Numpy 数组的交换排序

转载 作者:行者123 更新时间:2023-12-01 04:46:44 34 4
gpt4 key购买 nike

我有一个维度为 96x96x3x1000 的 4-D Numpy 数组 - 这些数组对应于我导入的图像数据集:每个 96X96 像素 1000 个图像以及每个像素的 RGB 值。

但是,我需要迭代每个图像的展平数组,即。只有一个二维数组[1000][96*96*3]。我设法通过首先执行

来转换给定的数组
    a.reshape(-1,a.size[3])

然后使用循环将每一列分配给图像。我想问是否有更简单/切片的方法来交换 ndarray 的顺序?

谢谢

最佳答案

您可以使用numpy.swapaxes更改轴的顺序

a.reshape(-1,1000).swapaxes(0,1)

或者简单地转置它

a.reshape(-1,1000).T

您还可以使用numpy.transpose更改轴的顺序,然后应用reshape

a.transpose([3,0,1,2]).reshape(1000,-1)

关于python - 4-D Numpy 数组的交换排序,我们在Stack Overflow上找到一个类似的问题: https://stackoverflow.com/questions/29258291/

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