Pandas 判断哪行有无穷值并输出

Pandas 判断哪行有无穷值并输出

Pandas 是一个强大的数据分析工具，可以轻松处理数据，进行数据清洗和分析。有时候，我们需要判断数据中是否存在无穷值，并输出含有无穷值的行。本文将介绍如何使用 Pandas 来判断哪行有无穷值并输出。

示例数据

首先，让我们创建一个包含无穷值的示例数据。我们使用 Pandas 的 DataFrame 来创建一个包含无穷值的数据集。

import pandas as pd

# 创建示例数据
data = {
    'A': [1, 2, float('inf'), 4],
    'B': [5, float('-inf'), 7, 8]
}

df = pd.DataFrame(data)
print(df)

运行以上代码，我们将得到如下输出：

     A    B
0  1.0  5.0
1  2.0 -inf
2  inf  7.0
3  4.0  8.0

在示例数据中，我们故意添加了无穷值，其中包括正无穷大（inf）和负无穷大（-inf）。

判断含有无穷值的行并输出

接下来，我们将使用 Pandas 来判断哪行含有无穷值，并输出这些行。我们可以使用 isnull() 方法来判断数据是否为空（包括无穷值），然后使用 any(axis=1) 方法来判断哪些行有无穷值。

# 判断哪行含有无穷值并输出
inf_rows = df[df.isnull().any(axis=1)]
print(inf_rows)

运行以上代码，我们将得到如下输出：

      A    B
1   2.0 -inf
2   inf  7.0

从以上输出中，我们可以看到第 1 行和第 2 行含有无穷值，分别为负无穷和正无穷。

总结

在本文中，我们介绍了如何使用 Pandas 来判断哪行含有无穷值，并输出这些行。通过简单的代码，我们可以轻松地处理数据中的无穷值，更好地进行数据分析和清洗。