如何提取Dataframe中包含指定字符串的列名？

提取包含指定字符串的列名

在数据框中，通过特定名称访问列可能至关重要。此问题解决了需要识别名称包含特定字符串的列的情况，即使它不完全匹配。提供的示例是在“spike-2”、“hey spike”和“spiked-in”等列名称中搜索“spike”。

解决方案：

要实现此目的，一种简单的方法涉及迭代 DataFrame 的列：

<code class="python">for col in df.columns:
    if 'spike' in col:
        # Do something with the column name</code>

在此解决方案中，检查每个列名称以检查它是否包含目标字符串。如果找到匹配项，则可以将列名称存储在变量中以供进一步使用。

另一种选择是利用列表理解和过滤来创建仅包含匹配列的新数据框：

<code class="python">spike_cols = [col for col in df.columns if 'spike' in col]
df2 = df.filter(regex='spike')</code>

第一行生成包含“spike”的列名称列表，而第二行过滤数据帧以仅包含这些列。

通过利用这些技术，您可以有效地识别和访问列他们的名字包含特定的字符串，扩大你的分析能力。

以上是如何提取Dataframe中包含指定字符串的列名？的详细内容。更多信息请关注PHP中文网其他相关文章！