python excel xlsx csv txt 文件读写操作

读操作
csv tsv 格式

# read csv or tsv
import numpy as np

category = np.loadtxt(Gaze_file, delimiter='\t', skiprows=1, usecols=[3], dtype=str)
# csv delimiter=','
# csv delimiter='\t'

# read excel
import csv
with open (filename) as f:
    reader = csv.reader(f)
    for row in reader:
        c1, c2 = row 
        column1.append(c1)
        column2.append(c2)
	return column1, column2

.xls .xlsx格式

# read excel
import openpyxl

def readdexcel(filename,sheetname):
    wb=openpyxl.load_workbook(filename)
    sheet= wb[sheetname]
    ws=wb.active
    Num_row=ws.max_row
    Num_col=ws.max_column
    data=np.zeros((Num_row,Num_col), dtype=float)
    for ii in range (Num_col):
        for jj in range(Num_row):
            data[jj][ii]=sheet.cell(row=jj+1, column=ii+1).value
    return data

txt格式
直接读取成字符串格式

with open("test.txt", "r") as f:
    data = f.readlines()
    print(data)

读取成列表格式，以‘\n’为分隔

with open(txt_file) as fin:
    test_list = [row.strip() for row in list(fin)]

写操作
txt格式
注意第二个：“a”表示不覆盖txt原有内容，持续写入。而"w"表示覆盖原有内容写入

with open('DReyeVE_sal_check.txt', "a") as f:
    f.write(content+'\n')
with open('DReyeVE_sal_check.txt', "w") as f:
    f.write(content+'\n')

csv tsv 格式

#csv format
import csv

with open ('Annotation.csv', mode='w', newline='') as f:
    writer = csv.writer(f)
    for ii in range(len(Video_Annotations)):
        writer.writerow(Video_Annotations[ii])#必须是列表格式
    print('writen into Annotation.csv')

.xls .xlsx格式

import openpyxl
def writetoxlsx(file_path, data):
    outwb = openpyxl.Workbook()  # 打开一个将写的文件
    outws = outwb.create_sheet(index=0)  # 在将写的文件创建sheet
    i = 1  # 注意：'cell'函数中行列起始值为1
    for line in data:
        for x in range(0,len(line)):
            outws.cell(column = x+1 , row = i , value = float(line[x]))
        i += 1
    outwb.save(file_path)  # 保存结果

参考博文：
https://blog.youkuaiyun.com/weixin_43269020/article/details/87553600