From e4568cb834d20659e6a8d54151cfbbc70b6c833a Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Nam=20L=C3=AA=20Quang?= Date: Fri, 7 Jan 2022 10:39:49 +0700 Subject: [PATCH] Update 6-read-and-write-csv-data.py Changed the key from 'favorite_color' to 'favorite color' and vice versa as required in exercises 3 and 4. --- .../6-read-and-write-csv-data.py | 23 +++++++++++++++---- 1 file changed, 18 insertions(+), 5 deletions(-) diff --git a/ch12-file-input-and-output/6-read-and-write-csv-data.py b/ch12-file-input-and-output/6-read-and-write-csv-data.py index 233b1d4..babcc18 100644 --- a/ch12-file-input-and-output/6-read-and-write-csv-data.py +++ b/ch12-file-input-and-output/6-read-and-write-csv-data.py @@ -40,18 +40,31 @@ file_path = Path.home() / "favorite_colors.csv" -with file_path.open(mode="w", encoding="utf-8") as file: - writer = csv.DictWriter(file, fieldnames=["name", "favorite_color"]) +def change_key(color, old_key, new_key): + value = color[old_key] + del color[old_key] + color[new_key] = value + return color + +favorite_colors =[change_key(color, 'favorite_color', 'favorite color') + for color in favorite_colors] + +with file_path.open('w', encoding = 'utf-8') as file: + writer = csv.DictWriter(file, favorite_colors[0].keys()) writer.writeheader() writer.writerows(favorite_colors) - # Exercise 4 favorite_colors = [] -with file_path.open(mode="r", encoding="utf-8") as file: - reader = csv.DictReader(file) +with file_path.open('r', encoding = 'utf-8') as file: + header = file.readline() + header = header.replace('\n', '') + header = header.split(',') + + reader = csv.DictReader(file, fieldnames = header) for row in reader: + row = change_key(row, 'favorite color', 'favorite_color') favorite_colors.append(row) print(favorite_colors)