Python快速索引：使用2D索引数组获取2D矩阵的切片_Python_Arrays_Indexing_Profiling

Python快速索引：使用2D索引数组获取2D矩阵的切片

python arrays indexing

Python快速索引：使用2D索引数组获取2D矩阵的切片,python,arrays,indexing,profiling,Python,Arrays,Indexing,Profiling,通过逐行分析我的代码，我发现程序的瓶颈是以下命令： A[index_array] 其中A和索引数组是二维矩阵。结果是一个3D阵列。现在，在一个程序中，40%的时间都在使用昂贵的命令，如np.einsum和Softmax函数我是python编码新手，我找不到更高效、更快的方法来编写它。有人能给点建议吗

通过逐行分析我的代码，我发现程序的瓶颈是以下命令：

A[index_array]

其中A和索引数组是二维矩阵。结果是一个3D阵列。现在，在一个程序中，40%的时间都在使用昂贵的命令，如np.einsum和Softmax函数

我是python编码新手，我找不到更高效、更快的方法来编写它。有人能给点建议吗