查找名称包含固定列中特定值的列

Question

我有一个带有列名的数据框，我想找到包含某个值的数据框。我正在像“...._segment”这样的列名中搜索“segment”列中的值。 我希望将列名作为字符串或变量返回，因此稍后我可以正常使用 df['name'] 或 df[name] 访问该列。 enter image description here

Answer 1

我不知道你是想获取包含你想要的字符串的列名，还是至少有一个值包含你想要的字符串的列的列名。

如果数据框是：

In [1]: import pandas as pd 
   ...: df = pd.DataFrame({'a_1': ['b_1', 'b_2'], 'b_1': ['a_1', 'a_2']})                                                                             
In [2]: df                                                                                                                                            
Out[2]: 
   a_1  b_1
0  b_1  a_1
1  b_2  a_2

对于第一种情况，如果要查找与a_*匹配的所有列名：

In [3]: import re                                                                                                                                     
In [4]: columns = [col for col in df.columns if isinstance(col, str) and re.match('a_.*', col)]                                                       
In [5]: columns                                                                                                                                       
Out[5]: ['a_1']

对于第二种情况，如果要查找至少有一个值与a_.* 匹配的所有列：

In [6]: columns = [col for col, ser in df.iteritems() if ser.str.match('a_.*').any()]                                                                 
In [7]: columns                                                                                                                                       
Out[7]: ['b_1']

其中：

df.iteritems：返回（列名，列值（系列））对的迭代器。

Series.any：如果系列中的任何值为True，则返回True。