使用 NumPy 数组作为另一个数组第二维的索引
根据数组提供的索引从 2D 数组中提取特定元素第二个数组，您可以利用 NumPy 的整数数组
考虑这个例子：
A = np.array([[0,1], [2,3], [4,5]]) B = np.array([[1], [0], [1]], dtype='int')
要获得以下所需的输出：
C = np.array([[1], [2], [5]])
您可以使用以下方法：
A[np.arange(A.shape[0]),B.ravel()]
怎么样有效：
或者，如果 B 是一维数组或列索引列表，则可以使用 .ravel() 跳过展平：
A[np.arange(A.shape[0]),B]
此方法提供了一种使用从另一个数组派生的索引从 2D 数组中提取元素的简单方法。
以上是如何使用另一个数组的索引从二维数组中提取元素？的详细内容。更多信息请关注PHP中文网其他相关文章！
