一尘不染

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

javascript

我有一个数据如下

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

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

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


阅读 64

收藏
2022-09-30

共1个答案

一尘不染

所以你可以在 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")

2022-09-30