需求背景
最近为公司开发了一套邮件日报程序，邮件一般就是表格，图片，然后就是附件。附件一般都是默认写到txt文件里，但是pm希望邮件里的附件能直接用excel这种软件打开，最开始想保存为excel，但是一想excel的文件体积会多出好多倍，csv文件默认也是使用excel打开的，但是根本还是文本文件，体积小，保存也方便，于是最终决定使用csv模块来保存文件。
python写csv文件
python提供了内置模块读写csv文件，这里我只用到了写，读这里就不做介绍了，也不难,主要是解决乱码问题。
def save2csv(file_name=none, header=none, data=none):
“””
保存成csv格式文件,方便excel直接打开
:param file_name: 保存的文件名
:param header: 表头,每一列的名字
:param data: 具体填充数据
:return:
“””
if file_name is none or isinstance(file_name, basestring) is false:
raise exception(‘保存csv文件名不能为空,并且必须为字符串类型’)
if file_name.endswith(‘.csv’) is false:
file_name += ‘.csv’
file_obj = open(file_name, ‘wb’)
file_obj.write(codecs.bom_utf8) # 防止乱码
writer = csv.writer(file_obj)
if data is none or isinstance(data, (tuple, list)) is false:
raise exception(‘保存csv文件失败,数据为空或者不是数据类型’)
if header is not none and isinstance(header, (tuple, list)) is true:
writer.writerow(header)
for row in data:
writer.writerow(row)
注意:有三句话就是为了防止乱码的
file_obj = open(file_name, ‘wb’)
file_obj.write(codecs.bom_utf8) # 防止乱码
writer = csv.writer(file_obj)
在文件头部写入codecs.bom_utf8就能防止乱码了,文件都是utf-8编码格式的
感谢阅读，希望能帮助到大家，谢谢大家对本站的支持！