如何根据数据框的某些值将 csv 文件拆分为多个文件python

问题描述

所以你可以在 pandas 中使用 loc 函数。我假设您将 csv 作为 pandas df 阅读

现在你可以像这样编写一个 for 循环：

df = pd.read_csv("gips.csv")
features = df.Id.unique()
for f in features:
    df_split = df.loc[df.ID == f]
    df_split.to_csv(f"{f}.csv")

解决方法

我有一个数据如下

IID	A	B
GIP1	..	..
GIP1	..	..
GIP2	..	..
GIP2	..	..
GIP3	..	..

我想根据 ID 拆分 csv 文件。GIP1 将 csv 文件作为 GIP1.csv 和 GIP2 作为 GIP2.csv 等。

谁能帮我在python中做到这一点？