Estou lutando para selecionar as colunas específicas por linha de uma matriz NumPy.
Suponha que eu tenha a seguinte matriz que eu chamaria de X
:
[1, 2, 3]
[4, 5, 6]
[7, 8, 9]
Eu também tenho um list
índice de coluna para cada linha que eu chamaria de Y
:
[1, 0, 2]
Eu preciso obter os valores:
[2]
[4]
[9]
Em vez de um list
com índices Y
, também posso produzir uma matriz com a mesma forma em X
que cada coluna é um bool
/ int
no intervalo de 0-1 valor, indicando se esta é a coluna necessária.
[0, 1, 0]
[1, 0, 0]
[0, 0, 1]
Sei que isso pode ser feito iterando a matriz e selecionando os valores de coluna de que preciso. No entanto, isso será executado com frequência em grandes arrays de dados e é por isso que deve ser executado o mais rápido possível.
Eu queria saber se existe uma solução melhor?