Předpokládejme, že mám index pole:
A = [1, 0, 3, 2, 0, 1]
pro další pole hodnot:
B = [21, 33, 10, 2]
Pak chci nové pole:
C = [B[A[0]], B[[A[1]], ..., B[A[[5]]] = [B[1], B[0], ..., B[1]] = [33, 21, ..., 21]
Jak se dělá to s numpy pole?
Předpokládejme, že mám index pole:
A = [1, 0, 3, 2, 0, 1]
pro další pole hodnot:
B = [21, 33, 10, 2]
Pak chci nové pole:
C = [B[A[0]], B[[A[1]], ..., B[A[[5]]] = [B[1], B[0], ..., B[1]] = [33, 21, ..., 21]
Jak se dělá to s numpy pole?
Můžete použít numpy indexování, když B
je numpy array:
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]