你可以使用zip.
zip从传递给它的iterables返回相同索引上的项.
>>> from math import isnan
>>> list1 = [[1, 2, 'nan'], [3, 7, 8], [1, 1, 1], [10, -1,'nan']]
>>> list2 = [1, 2, 3, 4]
>>> out = [(x,y) for x,y in zip(list1,list2)
if not any(isnan(float(z)) for z in x)]
>>> out
[([3, 7, 8], 2), ([1, 1, 1], 3)]
现在解压缩以获得所需的输出:
>>> list1_clean, list2_clean = map(list, zip(*out))
>>> list1_clean
[[3, 7, 8], [1, 1, 1]]
>>> list2_clean
[2, 3]
拉链帮助:
>>> print zip.__doc__
zip(seq1 [, seq2 [...]]) -> [(seq1[0], seq2[0] ...), (...)]
Return a list of tuples, where each tuple contains the i-th element
from each of the argument sequences. The returned list is truncated
in length to the length of the shortest argument sequence.
如果你想要一个内存有效的解决方案,你可以使用itertools.izip,因为它返回一个迭代器.