我刚开始学习Python,我的问题可能解释得很差,因为我有MATLAB的背景。通常,在MATLAB中,如果我们有1000个15*15的数组,我们定义一个单元格或3D矩阵,其中每个元素都是大小为(15*15)的矩阵。
现在在Python中(使用numpy库),我有一个形状为(1000, 15, 15)的ndarray A和另一个形状为(500, 15, 15)的ndarry B。
我试图找到A中也属于B的元素。我特别要求返回一个向量,其中包含A中与B相同元素的索引。
通常在MATLAB中,我将它们重塑为2D数组(1000*225)和(500*225),并使用'ismember'函数传递'rows'参数来查找并返回相似行的索引。
在numpy(或任何其他库)中是否有类似的功能来完成同样的操作呢?我正在尝试避免for循环。
谢谢
this
。 - Divakar