python中常见关于Excel表格读写操作------csv操作

公众号写的更加详细，可以关注看看哟！【软件测试小助手】

一、csv

1、读取csv文件中内容【可以自己手动创建一个csv文件，逗号分隔】

import scv
import codecs
# 创建一个csv文件，并填入内容
def load_csv():
  with open('a.csv','a') as file:
     f=csv.reader(file)
     print(list(f))   # 直接打印f，是一个object对象
  file.close()  # 关闭文件

2、写入到scv文件中

def load_csv():
    header = {'title', 'content'} # 表头数据
    data1 = ['test1', 'test2']  # 单行数据
    data2 = [['test1', 'one'], ['test2', 'two']] # list数据
  with open('b.csv', 'w', newline='') as file:
      writer = csv.writer(file)
      writer.writerow(header)
      writer.writerow(data1)  # 写入单行
      writer.writerows(data2) # 写入多行

3、读取文件DictReader方法【读取结果为字典类型】

# 打印出来的数据是字典类型，表格的表头为键、每一行的值为值，值有几行就打印几个字典
def read_csv():
    with open('b.csv', 'r') as file:
        f = csv.DictReader(file)
        for i in f:
            print(i['title'], i['content'])

4、写入文件DictWriter方法【写入内容为字典类型】

def load_csv():
    datas = {'title': '1', 'content': '测试test1'}
    f = codecs.open('b.csv', 'w', 'gbk')
    writer = csv.DictWriter(f, fieldnames=['title', 'content'])
    writer.writeheader()
    writer.writerow(datas)  # 写入一行
    writer.writerows(datas)  # 写入多行