Как в numpy из матрицы с векторами из int сделать матрицу из булевых векторов?

Question

[notxleb]

Допустим есть такая матрица indexes = [[[1,2,3]], [[3,2,4]]]. Как получить матрицу [[[0,1,1,1,0]], [0,0,1,1,1]]] без использования for желательно.

Comments

[dmshar]

Допустим такая дивная матрица есть. А можете объяснить, что означает вот это?

[[1,2,3]]

и - главное - ЗАЧЕМ принята такая структура представления?

[Dimonchik]

dmshar, разве не тензор? ну или просто н-мерная

[dmshar]

Dimonchik, Так и я о том. ТС имел ввиду именно то, что каждый из элементов есть список но из одного элемента, который тоже есть список из трех элементов? Или там одни скобки - лишние? И это списки или numpy-массивы?

Answer 1

[Dimonchik]

гугли что-то вроде astype(bool)

Comments

[dmshar]

Не получается. Тут надо сначала брать длину каждого элемента нового списка по максимальному значению (max) среди всех элементов первого списка, заполнять их нулями, а потом заменять в них элементы по индексам, взятым из значений первого массива.
Что-то мне кажется, что ТС все-таки перемудрил со скобками.

[notxleb]

dmshar, не перемудрил, матрица(тензор) даже еще упрощена. Можно допустить, что макс длина уже известна.

Answer 2

[dmshar]

С учетом всех уточнений, наиболее лаконичное, что мне удалось придумать:

import numpy as np
import itertools
ind=np.array([[[1,2,3]], [[3,2,4]]])
M=4

lt=np.zeros((len(ind),1,M+1),dtype=int)
for a in range(len(ind)):
    lt[a,0,list(itertools.chain(*ind[a]))]+=1

print(lt)

Результат:

[[[0 1 1 1 0]]
 [[0 0 1 1 1]]]

Если бы не странный размер исходного массива, о которой я писал выше:

ind.shape
Out[15]: (2, 1, 3)

все было бы на много проще.