根据列从一个csv中提取不同的csv文件

Question

您好，我是 python 初学者。 我想根据列中的值从一个大 csv 中提取不同的 csv 文件。 我的 csv 看起来像这样：

IDMember    Risk    Area
1             2       30
2             1       20
3             2       10

我有数千个具有相同 IDMember 的值，我想为每个这个 ID 提取一个 csv，该 ID 在 csv 文件名中。 我该怎么做？

我正在尝试这样的事情，但我认为我错过了一些东西：

import os
import pandas

df = r'MyCSVPath'
outFolder = r'MyOutputFolder'

col = 'MemberID'

for col in df.groupby('MemberID'):
    df.to_csv(f'{MemberID}.csv')

Answer 1

假设我们有一个重复 id 的数据框。

   IDMember  Risk  Area
0         1     2    30
1         2     1    20
2         3     2    10
3         4     2    15
4         1     2    45
5         2     2    40
6         4     3    50
7         1     3    30

脚本：

import os
import pandas as pd

df = pd.read_csv('csvFile.csv') #reading the csv file

df1 = df['IDMember'].drop_duplicates(keep='first', inplace=False) #creating dataframe showing unique ids

for elem in df1:  #iterating through each id
  df2 = df.loc[df['IDMember'] == elem]
  print(df2)
  df2.to_csv('IDmember'+str(elem)+'.csv')  #writing to separate csv by each id