import pandas as pd

def copy_to_files(dataframe, column, file_prefix, range_start, range_end): # 根据值范围切分数据 data_ranges = pd.cut(dataframe[column], bins=[range_start, range_end], include_lowest=True)

# 遍历每个数据范围
for data_range in data_ranges.unique():
    # 创建新的文件名
    file_name = f"{file_prefix}_{data_range}.csv"
    
    # 根据数据范围筛选数据
    filtered_data = dataframe[data_ranges == data_range]
    
    # 将数据保存为CSV文件
    filtered_data.to_csv(file_name, index=False)

示例数据

data = {'Name': ['Tom', 'Nick', 'John', 'Lucas', 'Emily'], 'Age': [25, 30, 35, 40, 45], 'Salary': [50000, 60000, 70000, 80000, 90000]} df = pd.DataFrame(data)

按值范围将数据复制到多个文件中

copy_to_files(df, 'Salary', 'output', 60000, 80000)