使用R或Python根据多行值将csv分割为较小的文件

data <- read_csv("input.csv")

import pandas as pd

data = pd.read_csv("input.csv")

split_data <- function(data, n) {
  num_files <- ceiling(nrow(data) / n)
  
  for (i in 1:num_files) {
    start <- (i - 1) * n + 1
    end <- min(i * n, nrow(data))
    
    split <- data[start:end, ]
    
    filename <- paste0("output", i, ".csv")
    write_csv(split, filename)
  }
}

split_data(data, 5)

def split_data(data, n):
    num_files = math.ceil(len(data) / n)
    
    for i in range(num_files):
        start = i * n
        end = min((i + 1) * n, len(data))
        
        split = data[start:end]
        
        filename = f"output{i+1}.csv"
        split.to_csv(filename, index=False)

split_data(data, 5)