如果要判断pandas DataFrame中的列是否以特定字符串结尾,并获取满足条件的列名称,可以使用以下方法:
DataFrame.columns
属性获取所有列的名称列表。endswith()
方法检查列名称是否以特定字符串结尾。下面是一个示例代码:
import pandas as pd
def get_columns_ending_with(df, suffix):
columns_ending_with = [col for col in df.columns if col.endswith(suffix)]
return columns_ending_with
# 示例用法
df = pd.DataFrame({'A_suffix': [1, 2, 3], 'B_suffix': [4, 5, 6], 'C': [7, 8, 9]})
suffix = '_suffix'
result = get_columns_ending_with(df, suffix)
print(result)
输出结果为:
['A_suffix', 'B_suffix']
这个方法使用了列表推导式来筛选出满足条件的列名称,并将它们存储在columns_ending_with
列表中。你可以将这个方法应用到任何pandas DataFrame中,只需传入相应的DataFrame和特定字符串的后缀即可。
腾讯云相关产品和产品介绍链接地址:
请注意,以上链接仅供参考,具体产品选择应根据实际需求进行评估和决策。
领取专属 10元无门槛券
手把手带您无忧上云