Python, a popular choice for data analysis and processing, provides straightforward ways to handle CSV files – the most common data interchange format. Python 3's built-in csv module is designed to efficiently read and write CSV files. Let's dive into an example showcasing its functionalities.
Before we jump into the code, we'll need a CSV file to work with. For this example, let's consider a simple 'students.csv' file containing student names and their respective ages:
Reading CSV Files
The csv module's DictReader is a versatile tool to read CSV files. It converts each row in the CSV file into an OrderedDict, where column names become keys and row values become the values.
Here's how you can read 'students.csv' using Python 3:
Using DictReader
```python import csv with open('students.csv', 'r') as file: reader = csv.DictReader(file) for row in reader: print(f"Name: {row['Name']}, Age: {row['Age']}") ```
Retrieving Column Names
DictReader also allows you to access the column names. This could be helpful while processing the data:
```python with open('students.csv', 'r') as file: reader = csv.DictReader(file) fields = reader.fieldnames print(f"Fieldnames: {fields}") ```
Writing CSV Files
The csv module's DictWriter is handy for writing data to a CSV file. It takes column names as an argument, and each row is a dictionary where keys are column names, and values are the corresponding row data.
Let's write some data into a new CSV file:
Using DictWriter
```python import csv fields = ['Name', 'Age'] rows = [['John', 25], ['Sarah', 19]] with open('new_students.csv', 'w', newline='') as file: writer = csv.DictWriter(file, fieldnames=fields) writer.writeheader() for row in rows: writer.writerow(dict(zip(fields, row))) ```
In the world of data analysis, mastering CSV file handling is a fundamental skill. Python 3's csv module equips you with the right tools to tackle this task. Now that you've seen these examples, you're ready to dive deeper into more complex CSV data processing needs.