为了账号安全,请及时绑定邮箱和手机立即绑定

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

/ 猿问

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

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]也可以

查看完整描述

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]


查看完整回答
反对 回复 2018-03-22

添加回答

回复

举报

0/150
提交
取消
意见反馈 帮助中心 APP下载
官方微信