20201128-2 【自动化办公】读写csv文件
Exercise 1 import csv # 设置员工发展基金确认表路径 source_path = './员工发展基金确认表.csv' # 设置存放拆分结果文件的文件夹路径 result_path = './员工发展基金文件夹_demo/' # 打开员工发展基金确认表 with open(source_path, 'r', encoding='utf-8', newline='') as source_file: # 将文件对象转换为DictReader对象 source_csv = csv.DictReader(source_file) # 将csv对象的表头读取出来 headers = source_csv.fieldnames # 循环处理确认表中除表头外的每一行数据 for csv_row in source_csv: # 根据获取的员工名字拼接新文件名 file_name = csv_row['姓名'] + '.csv' # 拼接新文件路径 file_path = result_path + file_name # 创建新文件并添加数据 with open(file_path, 'w', encoding='utf-8', newline='') as target_file: # 将文件对象转换为DictWriter对象 target_writer = csv.DictWriter(target_file, fieldnames=headers) # 写入表头 target_writer.writeheader() # 写入数据 target_writer.writerow(csv_row)
Exercise 2 功能块代码展示 首先来看第一个功能块——读取csv文件内容。 第3、4行，设置了【员工发展基金确认表.csv】的文件路径。 第8至13行，打开了【员工发展基金确认表.csv】文件，再读取该csv文件中的内容。 然后是第二个功能块——将内容写入新的csv文件。 第5、6行，设置存放拆分结果文件的文件夹路径。 第17至20行，使用字符串拼接，得到了要写入内容的新csv文件的路径。 第22至29行，创建并打开新csv文件，并将内容写入其中。 import csv # 设置员工发展基金确认表路径 source_path = './员工发展基金确认表.csv' # 设置存放拆分结果文件的文件夹路径 result_path = './员工发展基金文件夹_demo/' # 打开员工发展基金确认表文件 with open(source_path, 'r', encoding='utf-8', newline='') as source_file: # 将文件对象转换为DictReader对象 source_csv = csv.DictReader(source_file) # 将csv文件的表头读取出来 headers = source_csv.fieldnames # 循环处理确认表中除表头外的每一行数据 for csv_row in source_csv: # 根据获取的员工名字拼接新文件名 file_name = csv_row['姓名'] + '.csv' # 拼接新文件路径 file_path = result_path + file_name # 创建新文件并添加数据 with open(file_path, 'w', encoding='utf-8', newline='') as target_file: # 将文件对象转换为DictWriter对象 target_writer = csv.DictWriter(target_file, fieldnames=headers) # 写入表头 target_writer.writeheader() # 写入数据 target_writer.writerow(csv_row)
Exercise 3 读取csv文件内容功能块 # 导入csv模块 import csv # 设置员工发展基金确认表路径 source_path = './员工发展基金确认表.csv' # 打开员工发展基金确认表 with open(source_path, 'r', encoding='utf-8', newline='') as source_file: # 将文件对象转换成DictReader对象 source_csv = csv.DictReader(source_file) # 将csv文件的表头读取出来 headers = source_csv.fieldnames # 循环打印除表头以外的所有数据 for csv_row in source_csv: # 打印数据 print(csv_row)
现有列表：
member_list = [['邱大仁'], ['徐小刚', '陈知枫'], ['王晴', '廖雨']]
请你使用 for循环 和 writerow(row) 将列表内容分三行写入csv文件 'writerow_demo.csv' 中。
# 导入csv模块
import csv
member_list = [
['邱大仁'],
['徐小刚', '陈知枫'],
['王晴', '廖雨']
]
# 打开并创建'./writerow_demo.csv'文件，注意参数的设置，获取文件对象
with open('./writerow_demo.csv', 'w', encoding='utf-8', newline='') as demo_file:
# 将文件对象转换为writer对象
csv_demo = csv.writer(demo_file)
# 循环遍历列表中的元素
for member in member_list:
# 将列表中的元素写入csv文件中
csv_demo.writerow(member)
Exercise 4 内容写入新的csv文件功能块 现有字典： info_dict = {'姓名': '徐小刚', '工号': '1005', '员工发展基金': '100', '员工签字': ''} 将其写入文件'徐小刚信息.csv'中。 提示：最好提前把字典中的键(key)取出来存入列表中，然后在实例化DictWriter时，将其传给参数fieldnames。 设置完参数还不够，要想写入表头，还得通过DictWriter对象调用方法writeheader()。 # 导入csv模块 import csv info_dict = {'姓名': '徐小刚', '工号': '1005', '员工发展基金': '100', '员工签字': ''} # 设置表头 headers = ['姓名', '工号', '员工发展基金', '员工签字'] # 创建并打开'./徐小刚信息.csv' with open('./徐小刚信息.csv', 'w', encoding='utf-8', newline='') as target_file: # 将文件对象转换为DictWriter对象 target_writer = csv.DictWriter(target_file, fieldnames=headers) # 写入表头 target_writer.writeheader() # 将字典写入csv文件 target_writer.writerow(info_dict)
Exercise 5 回顾一下代码所需的关键知识点。 ① csv是Python的内置模块，使用前需要导入这个模块。 ② 最好在一开始就设置好要打开的csv文件的路径和存放拆分结果的文件夹的路径，因为这样可以简化代码，让代码看起来更加清晰易读。 ③ 读取csv文件的时候使用with open() as ...（注意设置新学的参数newline），然后使用DictReader将文件对象转换为DictReader对象，最后再通过DictReader对象的属性fieldnames获取表头。 ④ 通过for循环来遍历DictReader中的每一个元素，从而获取csv文件中除表头外的每一行内容。 ⑤ 使用字典的取值和字符串拼接得到完整的文件路径，然后再打开这个文件。使用DictWriter将文件对象转换为DictWriter对象，必须要将之前得到的表头传给参数fieldnames。 ⑥ 将表头写入csv文件时，还需要使用方法writeheader()，写入对应的每一行内容可以使用方法writerow()。 重复上面的⑤和⑥，直至遍历完整个【员工发展基金确认表.csv】。 # 导入csv模块 import csv # 设置员工发展基金确认表的路径 source_path = './员工发展基金确认表.csv' # 设置存放拆分结果文件的文件夹（员工发展基金文件夹）的路径。 result_path = './员工发展基金文件夹/' # 打开员工发展基金确认表 with open(source_path, 'r', encoding='utf-8', newline='') as source_file: # 将文件对象转换为DictReader对象 source_csv = csv.DictReader(source_file) # 将csv文件的表头读取出来 headers = source_csv.fieldnames # 循环处理确认表中除表头外的每一行数据 for csv_row in source_csv: # 根据获取的员工名字拼接出新文件名 file_name = csv_row['姓名'] + '.csv' # 拼接新文件路径 file_path = result_path + file_name # 创建新文件并添加数据 with open(file_path, 'w', encoding='utf-8', newline='') as target_file: # 将文件对象转换为DictWriter对象 target_writer = csv.DictWriter(target_file, fieldnames=headers) # 写入表头 target_writer.writeheader() # 写入数据 target_writer.writerow(csv_row)
