import numpy as np
A = np.array([[0.25, 0.25], [0.25, 0.75], [0.75, 0.25], [0.75, 0.75], [0.8, 0.9]])
B = np.array([0, 1, 2, 4, 0, 3, 3])
C = A[B, :]
print(C)
aW1wb3J0IG51bXB5IGFzIG5wCgpBID0gbnAuYXJyYXkoW1swLjI1LCAwLjI1XSwgWzAuMjUsIDAuNzVdLCBbMC43NSwgMC4yNV0sIFswLjc1LCAwLjc1XSwgWzAuOCwgMC45XV0pIApCID0gbnAuYXJyYXkoWzAsIDEsIDIsIDQsIDAsIDMsIDNdKQpDID0gQVtCLCA6XQpwcmludChDKQ==