[英]select a sublist from a list of lists in python
我有一个列表A
的列表:
A = [ [0, 42, 2.6],
[1, 20, 5.6],
[2, 67, 3.5],
[3, 12, 3.2],
[4, 34, 1.1],
[5, 74, 4.7],
[6, 29, 2.9] ]
我想从中提取一个子列表B
,该子列表B
仅包含第二列小于30的那些列表(即B = [[1, 20, 5.6],[3, 12, 3.2],[6, 29, 2.9]]
)
当然,我可以将其转换为numpy
并使用np.where
。 但是我想知道是否只有通过使用列表才能做到这一点。 我尝试了列表理解,但是在这方面我不是很熟练:
B = [x for x in len(A) if x[1] <= 30]
但这确实不起作用。 有什么建议吗? 提前致谢。
尝试这个:
B = [x for x in A if x[1] <= 30]
您当前的代码正在尝试生成一个值,该值将在每次迭代时用作索引。 要实现该方法,请使用x
访问A
每一行:
B = [A[x] for x in range(len(A)) if A[x][1] <= 30]
但是请注意,使用列表推导或更实用的filter
要简单得多:
清单理解:
B = [i for i in A if i[1] <= 30]
filter
:
B = list(filter(lambda x:x[1] <= 30, A))
这是一种向量化方法:
import numpy as np
A = np.array([[0, 42, 2.6],
[1, 20, 5.6],
[2, 67, 3.5],
[3, 12, 3.2],
[4, 34, 1.1],
[5, 74, 4.7],
[6, 29, 2.9]])
A[A[:, 1] <= 30]
# array([[ 1. , 20. , 5.6],
# [ 3. , 12. , 3.2],
# [ 6. , 29. , 2.9]])
声明:本站的技术帖子网页,遵循CC BY-SA 4.0协议,如果您需要转载,请注明本站网址或者原文地址。任何问题请咨询:yoyou2525@163.com.