Выбор элементов из списка двух массивов - PullRequest
0 голосов
/ 27 апреля 2018

У меня есть следующий список из двух массивов:

l=[array([16, 19]), array([24, 17])]

Как создать новые списки, чтобы иметь

#[16,24] and [19,17]

Я пробовал l[0] и l[1], но те просто дают мне [array([16, 19])] и [array([24, 17])] соответственно.

Ответы [ 3 ]

0 голосов
/ 27 апреля 2018

Как я уже говорил в комментариях, это операция транспонирования.

import numpy as np
l=[np.array([16, 19]), np.array([24, 17])]

a, b = np.transpose(l)
print(a)
print(b)

# [16 24]
# [19 17]
0 голосов
/ 27 апреля 2018

Вы можете использовать комбинацию zip и map:

list(map(list, zip(*l)))
# [[16, 24], [19, 17]]
0 голосов
/ 27 апреля 2018

Использовать .tolist() метод от numpy

Ex:

import numpy as np
l=[np.array([16, 19]), np.array([24, 17])]
print([i.tolist() for i in l])
...