测试列类型是否为数字脚本

import pandas as pd

def is_column_numeric(dataframe, column_name):
    column = dataframe[column_name]
    return column.dtype == 'int64' or column.dtype == 'float64'

# 示例用法
data = {'Name': ['John', 'Emma', 'Mike'],
        'Age': [25, 30, 35],
        'Salary': [50000, 60000, 70000]}
df = pd.DataFrame(data)

column_name = 'Age'
is_numeric = is_column_numeric(df, column_name)
print(f"The column '{column_name}' is numeric: {is_numeric}")