Pandas 检查字符串是否在数据框中

在本文中，我们将介绍如何使用Pandas库来检查一个字符串是否在数据框中。Pandas是Python中一个数据分析的库，它提供了一些很有用且易于使用的数据结构和数据分析工具。

数据框与字符串

在Pandas中，数据框是一个二维表格，它由行和列组成。每一列可以是不同的数据类型，例如整数、浮点数和字符串等。下面是一个简单的示例，其中包含一些员工的名称、年龄和薪水：

import pandas as pd

data = {'Name': ['Tom', 'Jack', 'Steve', 'Ricky', 'Vin', 'David'],
        'Age': [28, 34, 29, 42, 25, 39],
        'Salary': [15000, 25000, 20000, 45000, 12000, 18000]}

df = pd.DataFrame(data)

print(df)

输出结果如下：

    Name  Age  Salary
0    Tom   28   15000
1   Jack   34   25000
2  Steve   29   20000
3  Ricky   42   45000
4    Vin   25   12000
5  David   39   18000

我们可以检查某一列中是否包含某个字符串。例如，我们可以检查“Name”列中是否包含“Tom”：

if 'Tom' in df['Name'].values:
    print('Tom is in the dataframe')
else:
    print('Tom is not in the dataframe')

输出结果为：

Tom is in the dataframe

同样，我们也可以检查一个字符串是否在整个数据框中。例如，我们可以检查数据框中是否包含字符串“Tom”：

if 'Tom' in df.values:
    print('Tom is in the dataframe')
else:
    print('Tom is not in the dataframe')

输出结果为：

Tom is in the dataframe

使用str.contains()方法

Pandas提供了一个非常有用的方法str.contains()，它可以检查一个字符串是否在数据框中。例如，我们可以使用以下代码来检查“Name”列中是否包含字符串“Tom”：

if df['Name'].str.contains('Tom').any():
    print('Tom is in the dataframe')
else:
    print('Tom is not in the dataframe')

输出结果为：

Tom is in the dataframe

同样，我们也可以检查一个字符串是否在整个数据框中，例如：

if df.astype(str).apply(lambda x: x.str.contains('Tom')).any().any():
    print('Tom is in the dataframe')
else:
    print('Tom is not in the dataframe')

输出结果为：