python – 如何根据另一个列表的值过滤列表
发布时间:2020-12-20 13:01:10 所属栏目:Python 来源:网络整理
导读:例如,我有两个相同长度的列表: a = [[1,2],[2,[3,3],[4,[5,6]]b = [1,2,3,1] 我希望有一个这样的功能 func(a,b,1) = [[1,6]]func(a,2) = [[2,3]]func(b,2) = [2,2] 该函数的作用是返回一个元素的列表,其列表b中相同索引的相应元素等于第三个参数. 在Matlab
例如,我有两个相同长度的列表:
a = [[1,2],[2,[3,3],[4,[5,6]] b = [1,2,3,1] 我希望有一个这样的功能 func(a,b,1) = [[1,6]] func(a,2) = [[2,3]] func(b,2) = [2,2] 该函数的作用是返回一个元素的列表,其列表b中相同索引的相应元素等于第三个参数. 在Matlab中,我将做一些像(b == 1),a(b == 2),b(b == 2)一样简单的事情.在Python中实现这一目标的最有效方法是什么? 解决方法
如果你想要非常类似Matlab的功能,你可以使用numpy:
>>> import numpy >>> a = [[1,6]] >>> b = [1,1] >>> a = numpy.array(a) >>> b = numpy.array(b) >>> a[b==1] array([[1,6]]) >>> a[b==2] array([[2,3]]) >>> b[b==2] array([2,2]) 如果做不到这一点,我可能只是使用列表理解: >>> [i for i,j in zip(a,b) if j == 1] [[1,6]] >>> [i for i,b) if j == 2] [[2,3]] >>> [i for i,j in zip(b,b) if j == 2] [2,2] 将它包装在一个函数中是微不足道的: >>> def func(a,x): ... return [i for i,b) if j == x] ... >>> func(a,2) [[2,3]] >>> (编辑:李大同) 【声明】本站内容均来自网络,其相关言论仅代表作者个人观点,不代表本站立场。若无意侵犯到您的权利,请及时与联系站长删除相关内容! |