根据条件从列表列表中删除元素

Question

我有一个列表列表（来自 dataframe 的某个列，其中 col 的每个元素都是一个列表），我将其转换为形状数组（2819、11）。

我正在尝试查找不包含 for 循环的代码，以消除包含 np.nan 的列表

Answer 1

您可以将np.isnan()与any() ) 结合使用来仅索引您想要的行：

l = np.array([
    [1, 2, 3],
    [1, np.nan, 4],
    [5, 6, 7],
    [np.nan, np.nan, 7]
])


l[~(np.isnan(l).any(axis=1))]

#array([[1., 2., 3.],
#       [5., 6., 7.]])

这使用~来反转isnan()中的 boolean 值，因此您可以获得错误的行。