快捷导航

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]
回复

使用道具 举报

可能感兴趣的问答

发新帖
  • 微信访问
  • 手机APP