如何从 numpy.ndarray 中删除 nan 值

Question

我有一些 numpy.ndarray 变量。它们包括 nan 值，我想从中删除每个 nan 值。数组包含 int、float、str 等值。这些数组的一个例子：

['A' 'B' 'C' 'D' nan 'E' 'F']

另一个：

[nan  1.]

并且可能存在数组包含 float、str 和 nan 值的情况。在这种情况下，我怎样才能只删除 nan 值？

我使用了以下代码：

x[:, ~np.isnan(x).any(axis=0)]

并得到以下错误：

ufunc 'isnan' not supported for the input types, and the inputs could not be safely coerced to any supported types according to the casting rule ''safe''

Answer 1

这可能是因为np.isnan() 未能处理集合中可能的元素类型中的字符串类型。您可以尝试使用 panda's isnull() 删除 NaN 值。

import pandas as pa
import numpy as np

a = ['A', np.nan, np.nan, 1.67, 8]
a = [x for x in a if not pa.isnull(x)]
print(a)

Answer 2

pandas 中有一个名为pandas.DataFrame.dropna 的函数，它会删除所有具有nan 值的列。如果您没有创建 DataFrame，请创建，然后执行 df.dropna()。