记录博客 ZH-BLOG

Python csv 的读取与写入

时间:2018-07-23 15:08:56分类:python 基础

csv 的读取

import csv,os
# 小文件直接转为list读取,在内存中
# csv_file=open(os.path.join('E:\\','path','abc.csv'))
# csv_reader=csv.reader(csv_file)
# print(type(csv_reader))
# csv_data=list(csv_reader)
# print(csv_data)
# csv_file.close()

# 对于大型的 CSV 文件,避免将整个文件一次性装入内存
csv_file=open(os.path.join('E:\\','path','abc.csv'))
csv_reader=csv.reader(csv_file)
for row in csv_reader:
    print('Row:%s:  %s'%(csv_reader.line_num,str(row)))
csv_file.close()

将 exel 中内容写入到 csv

import os,csv,openpyxl
wb=openpyxl.load_workbook(os.path.join('E:\\','path','abc.xlsx'))
sheet=wb.active
csv_file=open(os.path.join('E:\\','path','abc.csv'),'w',newline='')
csv_writer=csv.writer(csv_file)
for row in sheet.iter_rows():
    row_val=[]
    for cell in row:
        row_val.append(cell.value)
    csv_writer.writerow(row_val)
wb.close()
csv_file.close()