從2D數組中提取元素，使用索引從另一個數組

Problem:

Consider two NumPy arrays:

A = np.array([[0, 1], [2, 3], [4, 5]]) B = np.array([[1], [0], [1]]) # Index arrayThe goal is to extract one element from each row of A, where the specific element is specified by the index in the corresponding row of B. The expected output should be:

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

A = np.array([[0, 1], [2, 3], [4, 5]])
B = np.array([[1], [0], [1]])  # Index array

在這種替代方法中，您將A型A型A匹配B的形狀，然後使用NP.Choose使用NP.Choose來選擇基於Flatteed Barteed b的blattened Baray。 Itable解開包裝（python> = 3.6）：

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

[np.stack（[range（a.shape [0]），b.ravel（）]，axis = 1）]在這種情況下，它將創建一個帶有行索引和B索引的2D數組，可用於從a中選擇所需的元素。 最合適的解決方案取決於任務的特定要求和約束，例如效率，可讀性，以及與Numpy的較舊版本的合適性。