Pandas 查找列名称包含特定字符串的列

阅读更多：Pandas 教程

引言

在Pandas中查找包含特定字符串的列名称是一个常见的需求。本文将介绍两种方法来解决这个问题。

方法一：使用str.contains函数

Pandas的str.contains函数是一种简便的方法，它返回一个布尔值，指示该字符串是否包含指定的子字符串。

我们可以使用df.columns.str.contains函数来查找包含特定字符串的列，如下所示：

import pandas as pd

df = pd.DataFrame({'A_1': [1, 2, 3], 'A_2': [4, 5, 6], 'B_1': [7, 8, 9]})

specific_string = 'A'

result = df.loc[:, df.columns.str.contains(specific_string)]

print(result)

输出：

   A_1  A_2
0    1    4
1    2    5
2    3    6

在本例中，我们首先创建一个Dataframe，它包含三个列：A_1、A_2、B_1。我们希望查找包含’A’的列名称。

我们使用df.columns.str.contains('A')查找列名称中包含’A’的列，并将其赋值给结果变量result。然后，我们将结果输出到控制台。

此方法可以通过使用正则表达式来进一步指定要查找的列名的模式。例如，如果要查找以大写字母开头的列名，可以使用以下代码：

result = df.loc[:, df.columns.str.contains('^[A-Z]')]
print(result)

方法二：使用filter函数

另一种方法是使用filter函数。该函数采用一个函数作为其参数，该函数应返回True或False，以指示列是否应包含在结果中。

我们可以使用这个函数来查找包含特定字符串的列，如下所示：

import pandas as pd

df = pd.DataFrame({'A_1': [1, 2, 3], 'A_2': [4, 5, 6], 'B_1': [7, 8, 9]})

specific_string = 'A'

result = df.filter(regex=specific_string)

print(result)

输出：

   A_1  A_2
0    1    4
1    2    5
2    3    6

在本例中，我们首先创建一个Dataframe，它包含三个列：A_1、A_2、B_1。然后，我们使用df.filter(regex='A')函数查找所有包含’A’的列，并将结果赋值给result变量。最后，我们将结果输出到控制台。

这个方法的好处是可以使用任何简单的函数作为过滤器，因此，它比str.contains更加灵活。

总结

在本文中，我们介绍了两种方法来查找包含特定字符串的列。使用这些方法可以快速、简便地找到需要的数据。使用str.contains可以让你对列名使用复杂的正则表达式，而使用filter函数则可以更加灵活地定制查找逻辑。