gpt4 book ai didi

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

转载 作者:IT老高 更新时间:2023-10-28 21:08:29 25 4
gpt4 key购买 nike

我有一个带有列名的数据框,我想找到一个包含某个字符串但不完全匹配的数据框。我在 'spike-2''hey spike''spiked- 等列名中搜索 'spike' in' ('spike' 部分始终是连续的)。

我希望列名作为字符串或变量返回,因此我稍后使用 df['name']df[name] 访问该列像平常一样。我试图找到方法来做到这一点,但无济于事。有什么建议吗?

最佳答案

只需遍历 DataFrame.columns,现在这是一个示例,您最终会得到一个匹配的列名列表:

import pandas as pd

data = {'spike-2': [1,2,3], 'hey spke': [4,5,6], 'spiked-in': [7,8,9], 'no': [10,11,12]}
df = pd.DataFrame(data)

spike_cols = [col for col in df.columns if 'spike' in col]
print(list(df.columns))
print(spike_cols)

输出:

['hey spke', 'no', 'spike-2', 'spiked-in']
['spike-2', 'spiked-in']

解释:

  1. df.columns 返回列名列表
  2. [col for col in df.columns if 'spike' in col] 使用变量 col 迭代列表 df.columns如果 col 包含 'spike',则将其添加到结果列表中。此语法为 list comprehension .

如果您只想要包含匹配列的结果数据集,您可以这样做:

df2 = df.filter(regex='spike')
print(df2)

输出:

   spike-2  spiked-in
0 1 7
1 2 8
2 3 9

关于python - 查找名称包含特定字符串的列,我们在Stack Overflow上找到一个类似的问题: https://stackoverflow.com/questions/21285380/

25 4 0
Copyright 2021 - 2024 cfsdn All Rights Reserved 蜀ICP备2022000587号
广告合作:1813099741@qq.com 6ren.com