python获取含1的所有列名

假设有一个DataFrame df，可以使用以下代码获取所有包含1的列名：

col_names = [col for col in df.columns if '1' in col]

这将生成一个列表，其中包含所有包含1的列名。如果您只想获取包含数字1的列名，可以使用正则表达式：

import re

col_names = [col for col in df.columns if re.search(r'\d1\d*', col)]

这将返回包含数字1的列名，例如'col_1', 'col_10', 'col_21'等。