如何将numpy对象数组转换为包含每个对象元素的数组?

时间:2019-04-30 18:35:26

标签: python numpy

我有一个形状为(60000,)的numpy对象数组,每个60000个元素都是(32,32,3)数组。我的问题是如何将(60000)数组转换为(32,32,3,60000)数组。

1 个答案:

答案 0 :(得分:0)

import numpy as np

class Obj():
    def __init__(self,i):
        self.i = i

l = np.array([np.array([Obj(i) for i in range(64*3)]).reshape(8,8,3)
 for _ in range(100)])

print(l.shape)
#Output: (100, 8, 8, 3)

print(np.transpose(l,(1,2,3,0)).shape)
#Output: (8, 8, 3, 100)