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]]>>>总结
以上是内存溢出为你收集整理的python – 如何根据另一个列表的值过滤列表全部内容,希望文章能够帮你解决python – 如何根据另一个列表的值过滤列表所遇到的程序开发问题。
如果觉得内存溢出网站内容还不错,欢迎将内存溢出网站推荐给程序员好友。
欢迎分享,转载请注明来源:内存溢出
评论列表(0条)