Numpy: Przekształcenie tablicy indeksów w tablicy elementów

0

Pytanie

Załóżmy, że mam tablicę indeksów:

A = [1, 0, 3, 2, 0, 1]

dla innej tablicy wartości:

B = [21, 33, 10, 2]

Wtedy potrzebuję nową tablicę:

C = [B[A[0]], B[[A[1]], ..., B[A[[5]]] = [B[1], B[0], ..., B[1]] = [33, 21, ..., 21]

Jak to zrobić z tablicami numpy?

arrays indexing numpy python
2021-11-24 01:35:59
1

Najlepsza odpowiedź

1

Można zastosować indeksacji numpy, gdy B jest tablicą numpy:

import numpy as np

A = [1, 0, 3, 2, 0, 1]
B = np.array([21, 33, 10, 2])
output = B[A]
print(output) # [33 21  2 10 21 33]
2021-11-24 06:33:08

W innych językach

Ta strona jest w innych językach

Русский
..................................................................................................................
Italiano
..................................................................................................................
Română
..................................................................................................................
한국어
..................................................................................................................
हिन्दी
..................................................................................................................
Français
..................................................................................................................
Türk
..................................................................................................................
Česk
..................................................................................................................
Português
..................................................................................................................
ไทย
..................................................................................................................
中文
..................................................................................................................
Español
..................................................................................................................
Slovenský
..................................................................................................................