写csv文件_Python实例:对CSV文件的操作
时间: 2023-12-01 18:04:12 浏览: 34
好的,下面是一个简单的 Python 实例,演示如何对 CSV 文件进行读取、写入和操作:
首先,假设我们有一个 CSV 文件,名为 `example.csv`,内容如下:
```
Name, Age, City
John, 25, New York
Sarah, 32, Los Angeles
Tom, 19, Chicago
```
1. 读取 CSV 文件
我们可以使用 Python 的内置模块 `csv` 来读取 CSV 文件,示例代码如下:
```python
import csv
with open('example.csv') as csv_file:
csv_reader = csv.reader(csv_file)
for row in csv_reader:
print(row)
```
运行以上代码,输出如下:
```
['Name', ' Age', ' City']
['John', ' 25', ' New York']
['Sarah', ' 32', ' Los Angeles']
['Tom', ' 19', ' Chicago']
```
2. 写入 CSV 文件
我们可以使用 Python 的内置模块 `csv` 来写入 CSV 文件,示例代码如下:
```python
import csv
with open('example.csv', mode='w') as csv_file:
fieldnames = ['Name', 'Age', 'City']
writer = csv.DictWriter(csv_file, fieldnames=fieldnames)
writer.writeheader()
writer.writerow({'Name': 'Alice', 'Age': 28, 'City': 'San Francisco'})
writer.writerow({'Name': 'Bob', 'Age': 35, 'City': 'New York'})
```
运行以上代码后,会在当前目录下创建一个新的 `example.csv` 文件,并写入以下内容:
```
Name,Age,City
Alice,28,San Francisco
Bob,35,New York
```
3. 对 CSV 文件进行操作
我们可以使用 Python 的内置模块 `csv` 来对 CSV 文件进行操作,示例代码如下:
```python
import csv
with open('example.csv') as csv_file:
csv_reader = csv.DictReader(csv_file)
for row in csv_reader:
row['Age'] = int(row['Age']) + 1
with open('example_new.csv', mode='w') as new_file:
fieldnames = ['Name', 'Age', 'City']
writer = csv.DictWriter(new_file, fieldnames=fieldnames)
writer.writeheader()
for row in csv_reader:
writer.writerow(row)
```
运行以上代码后,会在当前目录下创建一个新的 `example_new.csv` 文件,并写入以下内容:
```
Name,Age,City
John,26,New York
Sarah,33,Los Angeles
Tom,20,Chicago
```
以上就是对 CSV 文件进行读取、写入和操作的 Python 实例。