gpt4 book ai didi

python - pandas 按多个值过滤列值

转载 作者:行者123 更新时间:2023-12-01 09:03:37 33 4
gpt4 key购买 nike

我有多个列的 df,例如 MLB、NBA、NHL、NFL、TESTNBA,我想返回一个列表,其中列中包含字符串 MLB 或 NBA。就像下面这样:

df_check = ['MLB', 'NBA', 'TESTNBA']

value_cols = [col for col in df.columns if df_check in col]

以上失败并出现以下错误:

类型错误:“in”需要字符串作为左操作数,而不是列表

有什么方法可以按多个值过滤列吗?

最佳答案

您可以使用pandas.DataFrame.filter :

设置

df = pd.DataFrame(columns=['MLB', 'NBA', 'NHL', 'NFL', 'TESTNBA'])    
df_check = ['MLB', 'NBA']
<小时/>
df.filter(regex='|'.join(df_check)).columns

Index(['MLB', 'NBA', 'TESTNBA'], dtype='object')

关于python - pandas 按多个值过滤列值,我们在Stack Overflow上找到一个类似的问题: https://stackoverflow.com/questions/52260609/

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