可以直接将处理csv文件的方法应用于tsv。因为两种文件的唯一区别在于分隔符,csv是逗号,tsv是制表符。
说明:将以下代码中的delimiter,sep参数去掉,即可用于处理 csv 文件,逗号是默认分隔符。
写文件
使用 csv 库。
with open('file.tsv', 'w') as f:
tsv_w = csv.writer(f, delimiter='\t')
tsv_w.writerow(['id', 'name', 'score']) # 单行写入
tsv_w.writerows([[1, 'Frank', 99], [2, 'John', 70]]) # 多行写入
读文件
1使用 csv 库
import csv
with open('file.tsv') as f:
tsvreader = csv.reader(f, delimiter='\t')
for line in tsvreader:
print(line)
2使用 DataFrame
from pandas import DataFrame
df = DataFrame.from_csv('file.tsv', sep='\t')
print(df
3使用 pandas 库的 csv 读取函数
import pandas as pd
print(pd.read_csv('file.tsv', delimiter='\t'))
Python 写入tsv文件
newline用于避免多余的空行。如果没有指定newline=’’,则每行数据后面都会多一行空行。
with open(r'file.tsv', 'w', newline='') as f:
tsv_w = csv.writer(f, delimiter='\t')
tsv_w.writerow(['A1', 'A2', 'A3'])
tsv_w.writerows(np.array(data_frame).tolist()) # 多行写入