, используя массив Numpy в качестве индексов для 2 -го измерения другого массива

для извлечения специфических элементов из 2D -массива, основанного на индексах, предоставленных вторым массивом, вы можете использовать numpy raster. Пример:

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

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

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

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

A[np.arange(A.shape[0]),B.ravel()]

объединенный индексный массив [np.arange (a.shape [0]), b.ravel ()] указывает индексы строки и столбца в A для извлечения. .ravel ():

a [np.arange (a.shape [0]), b]
] Этот метод обеспечивает простой подход для извлечения элементов из 2D массива, используя индексы, полученные из другого массива.
• ]]