使用pandas库读写文件比较方便,建议文件处理相关的使用该库。
#写入csv文件
import pandas as pd
from columns_name import list_array #list_array所在文件
def Txt2Csv(Txt_path, Csv_path):
'''
:param Txt_path: txt文件路径
:param Csv_path: csv文件路径
:return:
'''
with open(Txt_path, encoding='utf-8') as f:
contents = []
readlines = f.readlines() # readlines是一个列表
for i in readlines:
line = i.strip().split(",") # 去掉前后的换行符,之后按逗号分割开
contents.append(line) # contents二维列表
df = pd.DataFrame(contents)
# df.to_csv(Csv_path, header=False) # 不添加表头
# df.columns = ["Source", "Target", "Type", "Path"] # 添加表头
df.columns = list_array #list_array是一个数组 #添加表头另一种方式
df.to_csv(Csv_path, index=False,encoding="UTF_8_sig") #使用encoding在写入csv文件时指定编码方式,解决中文乱码问题
print("数据写入成功")