今天是鐵人賽的第二十三天,學習關於CSV(Comma-Separated Values)文件的處理。
-1CSV文件的讀寫:
Python的csv模組提供了簡單的方式來讀取和寫入CSV文件。使用 csv.reader() 讀取文件時,可以方便地將每一行轉換為列表,這對於逐行處理數據非常實用。
舉例:
import csv
import os
current_directory = os.getcwd()
file_path = os.path.join(current_directory, 'data.csv')
try:
with open(file_path, 'r') as csv_file:
csv_reader = csv.reader(csv_file)
for row in csv_reader:
print(row)
except FileNotFoundError:
print(f"文件 '{file_path}' 不存在. Please make sure the file exists and the path is correct.")
使用 csv.writer() 可以將數據寫入 CSV 文件。
import csv
data = [['Name', 'Age', 'City'], ['Alice', 30, 'New York'], ['Bob', 25, 'Los Angeles']]
with open('output.csv', 'w', newline='') as csv_file:
csv_writer = csv.writer(csv_file)
csv_writer.writerows(data) //將多行寫入文件
-2資料清理與處理:
在處理CSV數據時,如何清理和轉換數據。
舉例:
import csv
with open('output.csv', 'r') as csv_file:
csv_reader = csv.reader(csv_file)
cleaned_data = []
for row in csv_reader:
try:
age = int(row[1])
cleaned_data.append([row[0], age, row[2]])
except ValueError:
continue