Python：从数组中提取索引值答案

【问题标题】：Python: Extract the index value from the arrayPython：从数组中提取索引值
【发布时间】：2023-03-31 18:44:02
【问题描述】：

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]

【问题讨论】：

嗨 Akhilesh Arkala，欢迎来到 StackOverflow！你能帮我理解你的问题吗？ “现在，我想打印 A3 相对于 A2 的索引” 你是什么意思？您的“预期输出”示例对我没有太大帮助。
嗨，@NerdOnTour 我想打印 A3 元素相对于 A2 的位置。如我的示例中所述 [15 1] 位于 A2 中的 (1,1) 位置。现在，我想将其打印为输出。
所以您不是在单独搜索数字 15 和 1，而是在 A2 中寻找 2 元素数组 [15, 1]？
是的，我正在 A2 中搜索 A3 的 2 个元素
啊，好的。感谢您的澄清。我认为，如果您可以在问题中尽可能清楚地说明这一点（通过编辑问题），这将对网站的其他用户有所帮助。

标签： python arrays numpy where-clause

【解决方案1】：

这（还）不是一个完整的答案，但可能已经解决了你的一些困惑。

为什么np.argwhere(A3 == A2).squeeze() 不工作？

== 运算符可以比较两个相同形状的数组。如果它们的形状不同，则（如果可能）广播它们以匹配形状。在这种情况下，A3 最初的形状为 (2,)，但广播为形状 (3,2) 以匹配 A2。那么A3 == A2就是一个bool数组，表示A2的元素是否等于A3的对应元素。
因此，会发生这种情况：

>>> import numpy as np
>>> 
>>> A2 = np.array([[0, 6], [15, 1], [1, 15]])
>>> A3 = np.array([15, 1])
>>> 
>>> A2.shape
(3, 2)
>>> A3.shape
(2,)
>>> 
>>> A2 == A3
array([[False, False],
       [ True,  True],
       [False, False]])

考虑到这一点，您可以了解argwhere 返回的内容以及原因：

>>> np.argwhere(A3==A2)
array([[1, 0],
       [1, 1]])

到目前为止，您的代码所做的并不是查找整个数组 [15, 1]，而是查找第一列中每次出现的 15 以及第二列中每次出现的 1。

【讨论】：

感谢您的解释和时间

【解决方案2】：

您正在尝试查找数组 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

【讨论】：

感谢您的代码@Houssem，但这给了我一个错误The truth value of an array with more than one element is ambiguous. Use a.any() or a.all()
你好@AkhileshArkala，试试新代码sn-p，如果在arr2中找到arr中的元素，它将返回True或False
谢谢@Houssem，但我也想打印索引
@AkhileshArkala 现在打印索引