在Python中查找2多维数组中的数组索引

Question

我有一个数组

v = (x,y,z)

还有两个多维数组

l = (a,b,c),(d,e,f)

和

r = (g,h,i),(l,m,n),(x,y,z).

我想知道v 的索引是在第一个还是第二个多维数组中。我试过numpy.where(v==l)[0][0] 但它返回：

对于大小为 0 的轴 0，索引 0 超出范围。

只有在我知道必须搜索索引的矩阵之前才有效，但我不知道。谢谢

如果我想知道索引和包含它的数组呢？

Answer 1

定义一个函数，该函数接受要搜索的项目和要搜索的数组列表，并使用循环在每个数组中查找该项目。使用异常处理捕获IndexError。

>>> import numpy as np
>>> v = np.array([[1, 2, 3]])
>>> r = np.array([[1, 2, 3], [0, 9, 8], [2, 4, 4]])
>>> l = np.array([[4, 5, 6], [7, 8, 9]])
def get_index(seq, *arrays):
    for array in arrays:
        try:
            return np.where(array==seq)[0][0]
        except IndexError:
            pass
...         
>>> get_index(v, r, l)
0
>>> get_index(np.array([7, 8, 9]), r, l)
1

如果在任何数组中都找不到该项目，您将得到 None 作为输出。

更新：

如果您也想要名称，则在字典中传递数组：

 def get_index(seq, **arrays):
    for name, array in arrays.items():
        try:
            return name, np.where(array==seq)[0][0]
        except IndexError:
            pass
...         
>>> get_index(v, **dict(r=r, l=l))
('r', 0)
>>> get_index(np.array([7, 8, 9]), **dict(r=r, l=l))
('l', 1)

Answer 2

def containsArray(v):
    for index, arr in enumerate(l):
        if v == arr:
            ret index;

    for index, arr in enumerate(r):
        if v == arr:
            ret index;