Python：从数组中提取索引值 [英] Python: Extract the index value from the array

问题描述

Python： 我有一个具有3行和2列的阵列A2和具有1行和2列的A3。现在，我想打印A3中两个元素相对于A2的索引

示例

A2 = np.array([[0 6]
               [15 1]
                [1 15]]) 

A3 = [15 1] 

现在，我要编写一个代码来打印A3的两个元素相对于A2的位置。

预期输出

(1，0)

请用python帮助我

我尝试使用

np.argwhere(A3 == A2).squeeze()

但这将输出为[1，0]，[1，1]

推荐答案

您正在尝试查找数组%2的元素是否在数组%1中？

def search_two_elements_of_array_in_array(arr,arr2):
    """
    :return: index if two elements of arr are in arr2 numpy array and print index
    """
    for i in range(len(arr)):
        if arr[i] in arr2:
            print(i)
            return i
    return -1

这篇关于Python：从数组中提取索引值的文章就介绍到这了，希望我们推荐的答案对大家有所帮助，也希望大家多多支持IT屋！