Python 二维数组查询元素所在子数组索引

Question

List也好，np.ndarray也可以，有没有方法做下面的运算呢
下面是一个想法，不知道py是否有比较效率的方法(可以避开for,while循环的那种)
Input:a=4
          arr=[[1,2,3],[5,6],[4,7,8]]
          If a in arr
Output: True 2
能返回[False,False,True]也可以

Answer 1

# 可以用Python中的map()+lambda表达式来解决：arr = [[1,2,3], [5,6], [4,7,8]]a = 4r = map(lambda i:a in i, arr)print(list(r))—>[False, False, True]