Python：如何将csv文件某一列中所有相同元素的行分别另外以要求的命名格式保存

Posted 2021-08-19 fxalll

假设我们需要根据设备ID重新提取对应的数据，保存在csv文件中，文件名分别为"task1-1A.csv",“task1-1B.csv”,“task1-1C.csv”…

代码如下：

import pandas as pd
import numpy as np

data = pd.read_csv("../data/data1.csv",encoding="gbk")
data = np.array(data)
# print(data)
total = []
i = 0
for col in data:
    # print(col[1])
    if(col[1] not in total):
        i = i + 1
        total.append(col[1])
# print(i,total)
# print(data[0])
name = ['订单号','设备ID','应付金额','实际金额','商品','支付时间','地点','状态','提现']
for num01 in range(0,i):
    k = 1
    res = []
    for num02 in data:
        if(num02[1] == total[num01]):
            # print(num02)
            res.append(num02)
        k = k + 1
    test = pd.DataFrame(columns=name,data=res)
    nameUsed = ['A','B','C','D','E','F','G'] #全自动化，如果有需要请按照顺序继续往下写
    test.to_csv('../data/task1-1{}.csv'.format(nameUsed[num01]),encoding='gbk')
    print('task1-1{}.csv'.format(nameUsed[num01])+'已完成！')
print('全部完成！')

运行后，打开task1-1A.csv看一看：

成功，其余展示略。