使用指定行数将大型数据框写入多个csv文件

要将大型数据框写入多个csv文件，可以使用如下的解决方法：

import pandas as pd
import math

def write_large_dataframe_to_csv(df, num_rows_per_file, file_prefix):
    num_rows = len(df)
    num_files = math.ceil(num_rows / num_rows_per_file)
    
    for i in range(num_files):
        start_idx = i * num_rows_per_file
        end_idx = min((i + 1) * num_rows_per_file, num_rows)
        
        file_name = f"{file_prefix}_{i}.csv"
        df_subset = df[start_idx:end_idx]
        
        df_subset.to_csv(file_name, index=False)
        
# 示例用法
df = pd.DataFrame({'A': range(1000), 'B': range(1000)})
write_large_dataframe_to_csv(df, num_rows_per_file=200, file_prefix='data')

在上面的代码示例中，write_large_dataframe_to_csv 函数接受三个参数：df 是要写入csv文件的大型数据框，num_rows_per_file 是每个csv文件的行数，file_prefix 是生成的文件名的前缀。

函数首先计算出需要多少个csv文件，然后使用一个循环来逐个写入每个文件。在每次迭代中，根据给定的行数范围切割数据框，并使用to_csv方法将切割后的数据框写入到csv文件中。

在示例用法中，我们创建了一个包含1000行的数据框，并将其切割为每个文件200行的子集。生成的文件名将以data_0.csv、data_1.csv等形式命名。