从 Pandas 数据框列标题中提取信息 [重复]

Posted 2023-03-11

【中文标题】从 Pandas 数据框列标题中提取信息 [重复]

【英文标题】：Extracting information from Pandas dataframe column headers [duplicate]

【发布时间】：2020-12-09 04:07:02

【问题描述】：

我有一个带有列标题的 pandas 数据框，其中包含信息。我想遍历列标题并在每个标题上使用逻辑运算来提取具有我拥有的相关信息的列。

我的 df.columns 命令给出了这样的结果：

['(param1:x)-(param2:y)-(param3:z1)',
'(param1:x)-(param2:y)-(param3:z2)',
'(param1:x)-(param2:y)-(param3:z3)']

我只想选择包含 (param3:z1) 和 (param3:z3) 的列。

这可能吗？

【问题讨论】：

试试df.columns.str.contains('param3:z1|param3:z3')

【参考方案1】：

你可以使用filter:

df = df.filter(regex='z1|z3')

【讨论】：

嗨 NYC Coder，这在我的示例数据上已成功。明天我将在我的完整数据集上尝试它，看看它是否有效。非常感谢。