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

Posted 方舟aark

tags:

篇首语:本文由小常识网(cha138.com)小编为大家整理,主要介绍了Python:如何将csv文件某一列中所有相同元素的行分别另外以要求的命名格式保存相关的知识,希望对你有一定的参考价值。


假设我们需要根据设备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看一看:

成功,其余展示略。

以上是关于Python:如何将csv文件某一列中所有相同元素的行分别另外以要求的命名格式保存的主要内容,如果未能解决你的问题,请参考以下文章