Как преобразовать массив numpy объектов в массив, содержащий элементы каждого объекта? - PullRequest
0 голосов
/ 30 апреля 2019

У меня есть массив пустых объектов с формой (60000,), и каждый из 60000 элементов является массивом (32,32,3). Мой вопрос заключается в том, как преобразовать массив (60000,) в массив (32,32,3,60000).

1 Ответ

0 голосов
/ 30 апреля 2019

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)
Добро пожаловать на сайт PullRequest, где вы можете задавать вопросы и получать ответы от других членов сообщества.
...