使用python提取csv文档数据，输出到另一个csv文档

浮生说

已于 2023-07-02 21:36:48 修改

阅读量2k

点赞数

文章标签： python 开发语言

于 2023-07-02 21:21:24 首次发布

本文链接：https://blog.csdn.net/qq_45739340/article/details/131494647

版权

文章讲述了如何使用Python编写脚本从多个CSV文件中提取带宽和延时数据，进行单位转换和数据合并，最终将结果写入新CSV文件，以提高工作效率。

摘要由CSDN通过智能技术生成

背景：

有AB两部分测试数据，需要把这两部分的数据各提取出来，写到另一个文档里面，每一部分都有多个csv文档，手动操作不仅效率低且费时，并且有出现错误的概率，所以利用python来编写一个自动提取数据的脚本来提高工作效率。

目的：

把AB两部分的csv文档数据提取出来，按照格式（顺序排列）写入到另一个文档里面。

A部分
在这里插入图片描述
B部分

以下是AB两部分的csv待处理数据A文件和数据B文件，这里只截取了部分，实际上数据较多

在这里插入图片描述

我们要把AB表格里的Bandwidth（带宽）字段和Lantency（延时）数据，按下面表格的形式写进去，RandRead为A表格数据，RandWrite为B表格数据，说实话我自己看着都眼花。。更别说手动一个个填进去。。。那可真是不小的工作量！

在这里插入图片描述
思路：先提取A数据，再提取B数据，然后AB数据交叉相加，再把数据类型转换输出，得到的两列数据即为Q1+Q32

1、提取数据A和数据B代码

根据AB两个表格，我们只需要第五列和六列的数据，所以在遍历读取每一行的时候，把这两列的数据拿出来就可以了！这里有单位需要换算，我们需要先把数据类型转换成int，再做乘除法。

#!/usr/bin/env python
'''
*- coding: utf-8 -*-
@Project ：PythonProject
@File    : test_Demo1.py
@Author  : fushengshuo
@Date    : 2023/3/2 11:19
@Module  :
@Description : 
'''

import csv
from itertools import chain

# A数据文件地址
ran_read_datapath = "E:\\Desktop\\zlf\\sda\\A\\sda_dataA.csv"
# B数据文件地址
ran_write_datapath = "E:\\Desktop\\zlf\\sda\B\\sda_dateB.csv"
# 填写 '1job' or '4job'
Jobs = '1job'
# 提取数据输出的文件路径
Output_datapath = "E:\\Desktop\\zlf\\sda\\output\\sum_AB.csv"
# 打开A数据文件
with open(ran_read_datapath, mode="rt", encoding="utf-8-sig") as RdFile: