python 使用csv模块读写csv格式文件的示例
importcsv
classHandleCsv:
'''
csv文件处理类
'''
def__init__(self,filename):
'''
构造器
:paramfilename:csv文件名
'''
self.filename=filename
defget_data(self):
'''
获取csv中所有数据
:return:嵌套字典的列表
'''
withopen(self.filename,mode='r',encoding='utf-8')asf:
cb=csv.reader(f)#实例化reader对象
header=next(cb)#获取表头，并将指针转向下一行
list_dict=[]
forrowincb:
list_dict.append(dict(zip(header,row)))
returnlist_dict
defget_one_row(self,row):
'''
获取单行数据
:paramrow:指定的行号
:return:对应行号的数据
'''
returnself.get_data()[row-1]
defwrite_csv(self,headers,values,data_type,mode='w'):
'''
写入数据到csv到文件中
:paramheaders:表头：列表类型
:paramvalues:表数据：1.嵌套元组的列表；2.嵌套字典的列表
:paramdata_type:传入的数据类型：1.'tuple'；2.'dict'
:parammode:写入方式，默认写入“w”
:return:
'''
withopen(file=self.filename,mode=mode,encoding='utf-8',newline='')asf:
ifdata_type=='tuple':
writer=csv.writer(f)#实例化writer对象
writer.writerow(headers)#写入表头
writer.writerows(values)#写入数据
elifdata_type=='dict':
writer=csv.DictWriter(f,headers)#实例化DictWriter对象
writer.writeheader()#写入表头
writer.writerows(values)#写入数据
else:
print("数据类型错误，请确认！")
以上就是python使用csv模块读写csv格式文件的示例的详细内容，更多关于python读写csv文件的资料请关注毛票票其它相关文章！