Question

我在Python中有一个列表，如下所示：

l = [[1,2,inf,inf,87],[3,4,3,5,3],[73,5,3,3,nan],[8,3,3,2,2],[3,3,2,2,inf]]

如何删除包含inf或nan？

的列表

感谢。

Answer 1

[subl for subl in l if all(math.isfinite(x) for x in subl)]

（给定Python＆gt; = 3.2;在早期的Pythons中，替换为numpy.isfinite，或显式检查值）

Answer 2

您可以检查简单的条件，即

import math
l = [[1,2,math.inf,math.inf,87],[3,4,3,5,3],[73,5,3,3,math.nan],[8,3,3,2,2],[3,3,2,2,math.inf]]

[ i for i in l if (math.inf not in i) and (math.nan not in i) ]

[[3, 4, 3, 5, 3], [8, 3, 3, 2, 2]]

Answer 3

@Amadan回答的一个变种：

list(filter(lambda x: all(map(math.isfinite, x)), l))

更通用的版本将使用numpy：

import numpy as np
list(filter(lambda x: all(map(np.isfinite, x)), l))

Answer 4

inf为math.inf，nan为np.nan：

import math
lst = [[1,2,math.inf,math.inf,87],[3,4,3,5,3],[73,5,3,3,np.nan],[8,3,3,2,2],[3,3,2,2,math.inf]]
for l1 in lst:
    if math.inf not in l1 and np.nan not in l1:
        print(l1)

在某些项目中包含nan或inf的列表列表

4 个答案: