Python按条件拆分CSV教程：按列值分成多个文件

用pandas按某一列的值把一个大CSV拆成多个小文件，比如按城市、按日期、按类别拆分，自动命名、自动建文件夹，附完整代码。

场景痛点

一份全国订单表，要按城市分别发给各区域负责人；一份全年日志，要按月份拆成12个文件归档。手动筛选、另存为、改名，几十次重复劳动还容易漏。用Python按列groupby一下，自动拆成多个文件并命名好。

用到的库

pip install pandas

完整代码

# Python按条件拆分CSV
import os
import pandas as pd

def split_by_column(input_path, output_dir, column):
"""
按某一列的值拆分，每个值一个CSV文件。
比如按"城市"列拆分，输出 北京.csv、上海.csv...
"""
os.makedirs(output_dir, exist_ok=True)
df = pd.read_csv(input_path, encoding="utf-8-sig")

# 检查列是否存在
if column not in df.columns:
raise ValueError(f"列 {column} 不存在，可用列：{list(df.columns)}")

# 按列分组
grouped = df.groupby(column)
total_files = 0
for key, group in grouped:
# 文件名清洗：去掉非法字符
safe_key = str(key).replace("/", "_").replace("\\", "_").replace(":", "_")
out_path = os.path.join(output_dir, f"{safe_key}.csv")
group.to_csv(out_path, index=False, encoding="utf-8-sig")
print(f"  {safe_key}.csv：{len(group)} 行")
total_files += 1

print(f"拆分完成：{total_files} 个文件，输出到 {output_dir}/")

def split_by_condition(input_path, output_dir, condition_fn, prefix):
"""
按自定义条件拆分，满足条件的进一个文件，不满足的进另一个。
比如"大于1万金额"vs"小于等于1万"。
"""
os.makedirs(output_dir, exist_ok=True)
df = pd.read_csv(input_path, encoding="utf-8-sig")

mask = condition_fn(df)
df_true = df[mask]
df_false = df[~mask]

df_true.to_csv(os.path.join(output_dir, f"{prefix}_true.csv"),
index=False, encoding="utf-8-sig")
df_false.to_csv(os.path.join(output_dir, f"{prefix}_false.csv"),
index=False, encoding="utf-8-sig")
print(f"条件拆分：满足 {len(df_true)} 行，不满足 {len(df_false)} 行")

def main():
input_file = "orders.csv"
out_dir = "split_output"

# 1. 按城市列拆分
split_by_column(input_file, out_dir, column="城市")

# 2. 按条件拆分：金额大于10000的单独一个文件
split_by_condition(
input_file,
out_dir,
condition_fn=lambda df: df["金额"] > 10000,
prefix="high_value",
)

if __name__ == "__main__":
main()