**
csv文件的读入
**
1. csv.reader()函数
reader()函数会返回一个reader对象,这是一个可迭代对象,该对象里面的每一个元素都是一个列表,每一个列表都对应了csv文件中的一行。
import csv
#读入文件
path1 = './员工发展基金确认表.csv'
with open(path1,'r',encoding='utf-8',newline='') as f1:
csv_reader = csv.reader(f1)
print(type(csv_reader))
#<class '_csv.reader'>
print(csv_reader)
#<_csv.reader object at 0x0000021CD276FA70>
#csv_reader 是一个对象
for i in csv_reader:
print(i)
#读取的结果包括表头
#['\ufeff姓名', '工号', '员工发展基金', '员工签字']
#['邱大仁', '1001', '300', '']
- csv.DictReader()类
如果在==csv文件中存在映射关系(即csv文件的第一行为表格中的表头)==使用csv.DictReader()类
实例化DictReader类后,会得到一个DictReader对象,这是一个可迭代对象,我们可以使用循环来遍历它的每一个元素。但不同于reader()的是,该对象里面的每一个元素都是一个字典,每一个字典的值都可以对应csv文件中的一行,键对应csv文件的表头。
这个类还有个属性fieldnames,该属性可以将csv文件(表格)的表头(第一行)读取出来,返回值是列表。这个表头也就是字典的键(key)。
import csv
#读入文件
path1 = './员工发展基金确认表.csv'
with open(path1,'r',encoding='utf-8',newline='') as f1:
csv_reader = csv.DictReader(f1)
print(type(csv_reader))
#<class 'csv.DictReader'>
print(csv_reader)
#<csv.DictReader object at 0x0000021CD28075F8>
title = csv_reader.fieldnames
print(title)
#['\ufeff姓名', '工号', '员工发展基金', '员工签字']
for i in csv_reader:
print(i)
print(type(i))
#<class 'collections.OrderedDict'>
print(i) 的结果为:
【注意】
i 不包括表头那一行
OrderedDict([(’\ufeff姓名’, ‘邱大仁’), (‘工号’, ‘1001’), (‘员工发展基金’, ‘300’), (‘员工签字’, ‘’)])
OrderedDict([(’\ufeff姓名’, ‘徐小刚’), (‘工号’, ‘1005’), (‘员工发展基金’, ‘100’), (‘员工签字’, ‘’)])
等等
csv 文件的写入
1. csv.writer()
writer()函数会返回一个writer对象,可以调用该对象的方法将字符串文本写入csv文件。
(1)csv_writer.writerow()
参数row代表了你想要写入csv文件的内容,它必须是一个可迭代对象,这里推荐使用列表。
另外,如果你想通过这个方法写入多行数据。需要借助循环来实现。
import csv
member_list = [['邱大仁'], ['徐小刚', '陈知枫'], ['王晴', '廖雨']]
#第一种方法:writerow()+for循环
with open('writerow.csv','w',encoding='gbk',newline='') as f1:
csv_writer = csv.writer(f1)
for i in member_list:
csv_writer.writerow(i)
(2)csv_writer.writerows()
参数rows也是可迭代对象,可以理解为由多个row组成。
这个方法让我们可以在没有for循环的情况下达到上面题目所要求的效果
import csv
member_list = [['邱大仁'], ['徐小刚', '陈知枫'], ['王晴', '廖雨']]
#第一种方法:writerow()+for循环
with open('writerows.csv','w',encoding='gbk',newline='') as f1:
csv_writer = csv.writer(f1)
csv_writer.writerows(member_list)
2. csv.DictWriter()
DictWriter的操作类似于常规的writer(),但DictWriter会将字典写入(映射)到csv文件的行中。
即通过DictWriter写入的行,都必须是字典形式,这点和writer()还是有区别的。
所以,当我们需要将字典类型的数据写入csv文件时,使用DictWriter会是一种更好的选择。
有一点需要你特别注意,与DictReader不同,在上面的图片中,我特意标注出了一个参数fieldnames。
在实例化DictWriter时,必须要传入该参数。
我们需要将一个列表传给参数fieldnames,它的作用是设置csv文件的第一行数据(即表头数据)。当我们要将字典通过DictWriter对象写入csv文件时,通常会把字典的键(key)存入一个列表,再把这个列表传给参数fieldnames。
实例化DictWriter后,会得到一个DictWriter对象,我们需要通过该对象来调用两个方法来写入内容。
第一个方法:writeheader(),该方法可将表头(即字典中的键(key))写入csv的第一行。
也就是说,如果要写入表头,不仅要在DictWriter中传入参数fieldnames,还需要调用writeheader()方法。这点很重要,一定要记住。
第二个方法:writerow(row) ,与writer对象调用writerow(row)不同的是,这里的参数row必须是字典形式的数据。
for i in csv_reader:
#print(i)
#print(type(i))
#<class 'collections.OrderedDict'>
#提取员工姓名,用以命名每一个独立的csv文件
name = i['\ufeff姓名'] +'.csv'
#print(name)
file_path = path2 + name
print(file_path)
with open(file_path,'w',encoding='utf-8',newline='') as f2:
csv_writer = csv.DictWriter(f2,fieldnames=headers)
**csv_writer.writeheader()**
csv_writer.writerow(i)