Pandas 如何查找包含某个值的列 - Python

在 Pandas 中，我们可以使用 DataFrame 对象的 isin 方法获取包含指定值的行或者使用 any 方法来查找包含指定值的列。

使用 isin 方法获取包含指定值的行

以下是一个示例 DataFrame：

import pandas as pd
import numpy as np

data = {'A': [1, 2, 3, 4, 5],
        'B': ['Apple', 'Banana', 'Grape', 'Orange', 'Apple'],
        'C': ['Red', 'Yellow', 'Green', 'Orange', 'Red']}

df = pd.DataFrame(data)

print(df)

输出：

   A       B       C
0  1   Apple     Red
1  2  Banana  Yellow
2  3   Grape   Green
3  4  Orange  Orange
4  5   Apple     Red

现在我们想获取包含字符串 'Apple' 的行：

mask = df.isin(['Apple']).any(axis=1)

print(df[mask])

输出：

   A      B    C
0  1  Apple  Red
4  5  Apple  Red

使用 any 方法获取包含指定值的列

现在假设我们想查找包含字符串 'Apple' 的列：

mask = df.isin(['Apple']).any()

print(df.loc[:, mask])

输出：

        B
0   Apple
1  Banana
2   Grape
3  Orange
4   Apple

在 Pandas 中，使用 isin 方法或者 any 方法可以方便地查找包含指定值的行或列。