Início / coding / Perguntas / 79337016
Accepted
Kuraga
Asked: 2025-01-08 02:37:00 +0800 CST

Projetar um índice de outro array

  • 772

Eu tenho um array ade tamanho (M, N, K). E eu tenho um array bde tamanho (M, N)com valores inteiros de [0, K-1].

Como faço para obter a matriz... cde tamanho (M, N), onde c[i, j] == a[i, j, b[i, j]] da maneira mais simples?

Qual parte do guia de indexação é?

numpy

2 respostas

  1. Para extrair dados dessa maneira, você passa índices correspondentes como matrizes bidimensionais em cada uma das 3 dimensões, onde as 2 primeiras devem ser o resultado de meshgridmatrizes (observação indexign='ij'):

    a[*np.meshgrid(range(M), range(N), indexing='ij'), b]

    Veja Indexação avançada no NumPy e numpy.meshgrid para mais detalhes.

    Exemplo:

    import numpy as np

M, N, K = 3, 4, 5
a = np.arange(M*N*K).reshape(M, N, K)
b = np.arange(M*N).reshape(M, N) % K
first, second = np.meshgrid(range(M), range(N), indexing='ij')
c = a[first, second, b]

assert all(a[i, j, b[i,j]] == c[i,j] for i in range(M) for j in range(N))
    • 2
  2. Best Answer

    Você pode usar indexação avançada :

    c = a[np.arange(M)[:, None], np.arange(N), b]

    Saída:

    array([[ 0,  6, 12, 18],
       [24, 25, 31, 37],
       [43, 49, 50, 56]])

    Tomando o exemplo de @Vitalizzare:

    # input
M, N, K = 3, 4, 5
a = np.arange(M*N*K).reshape(M, N, K)
b = np.arange(M*N).reshape(M, N) % K

# output
array([[ 0,  6, 12, 18],
       [24, 25, 31, 37],
       [43, 49, 50, 56]])
    • 2

relate perguntas