你可以使用pandas的DataFrame的columns属性来获取列名列表,然后使用isin方法和dtype属性来筛选出值为整数或浮点数的列名,并将其添加到新的列表中。以下是一个示例代码:
import pandas as pd
# 创建示例DataFrame
data = {'col1': [1, 2, 3],
'col2': [4.1, 5.2, 6.3],
'col3': ['a', 'b', 'c']}
df = pd.DataFrame(data)
# 遍历DataFrame的列名列表
int_float_cols = []
for col in df.columns:
# 检查列的数据类型是否为整数或浮点数
if df[col].dtype == 'int64' or df[col].dtype == 'float64':
int_float_cols.append(col)
print(int_float_cols)
输出结果将是['col1', 'col2']
,因为'col1'和'col2'列的值为整数和浮点数类型。