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 = 4

r = map(lambda i:a in i, arr)

print(list(r))

—>[False, False, True]