usando uma matriz numpy como índices da 2ª dim de outra matriz? [duplicado

Question

Jun 18, 2016, 03:54 PM

Esta pergunta já tem uma resposta aqui:

Por exemplo, tenho duas matrizes numpy,

A = np.array(
  [[0,1], 
   [2,3], 
   [4,5]])
B = np.array(
  [[1],
   [0],
   [1]], dtype='int')

e eu quero extrair um elemento de cada linha deA, e esse elemento é indexado porB, então quero os seguintes resultados:

C = np.array(
  [[1],
   [2],
   [5]])

Eu tenteiA[:, B.ravel()], mas transmitiráB, não é o que eu quero. Também olhou paranp.take, parece não ser a solução certa para o meu problem

No entanto, eu poderia usarnp.choose por transposiçãoA,

np.choose(B.ravel(), A.T)

mas qualquer outra solução melhor?

questionAnswers(2)

Configuração SSL no Plug-in do Maven Tomcat

Comportamento Scanset em scanf em C

Notificação por push quando o aplicativo está em segundo plano

A expressão do Automapper deve ser resolvida para o membro de nível superior

Escrevendo um visualizador de área de transferência em c #