python接口自动化的csv文件怎么创建和读取

python接口自动化的csv文件怎么创建和读取

import csv
import json
import chardet

#1.先定义（def）一个创建csv文件的方法，要使用csv模块，在把数据以data键值对的格式以with open方法，writer进去，添加文件路径之后，执行改方法
#2.可以使用插入chardet模块查看文件编码格式encoding（这一断属于学习可以不写）
#3.定义（def）一个读取csv文件的方法，以with open方法打开文件，使用csv.DictReader 读取文件，然后for循环遍历文件行，拿到数据之后使用添加（append）到新的data
#添加到data之后 使用json.dump方法转换为json格式，在执行定义的读取方法 就可以打印出来json格式的csv文件

#以下是三个步骤的代码示例

#第一段.创建一个csv文件 并输入数据
def create_csv_using_csv_module(file_path):
    data = [
        ['Name', 'Age', 'Tel'],
        ['Alis', 35, '136'],
        ['Bob', 30, '135'],
        ['Charlie', 35, '132']
    ]

    with open(file_path, mode='w', newline='', encoding='utf-8') as csv_file:
        writer = csv.writer(csv_file)
        writer.writerows(data)
# 示例文件路径
file_path = 'example.csv'
# 创建 CSV 文件
create_csv_using_csv_module(file_path)

#第二段.检测csv文件编码（这段可以不写，用来学习怎么查文件格式）
def detect_encoding(file_path):
    with open(file_path, 'rb') as file:
        rawdata = file.read()
        result = chardet.detect(rawdata)
    return result['encoding']
# 示例文件路径
file_path = 'example.csv'
# 检测文件编码
encoding = detect_encoding(file_path)
print(f"The encoding of the file is: {encoding}")


# 第三段.for循环遍历读取文件编码 并转换为json格式
def csv_to_json(csv_file_path):
    data = []
    with open(csv_file_path, mode='r', encoding='utf-8') as csv_file:
        csv_reader = csv.DictReader(csv_file)
        for row in csv_reader:
            data.append(row)
        print(data)
    json_data = json.dumps(data, indent=4)
    print(json_data)
#data是一个list 想要单独取第n行数据根据list索引 list[n-1]
    print(data[1])
# 示例文件路径
csv_file_path = 'example.csv'
# 执行转换并打印
csv_to_json(csv_file_path)

原文地址：https://blog.csdn.net/zhengxugang/article/details/145192718

免责声明：本站文章内容转载自网络资源，如侵犯了原著者的合法权益，可联系本站删除。更多内容请关注自学内容网（zxcms.com）！