Python 从 CSV 文件导入数据的终极指南161

CSV（逗号分隔值）文件是一种流行的存储文本数据的格式，它们通常用于电子表格和数据库。使用 Python 导入 CSV 文件非常简单，本文将逐步指导您完成整个过程。

导入必要的模块

第一步是导入必要的 Python 模块，我们将在该模块中使用 CSV 相关的功能。使用以下代码导入 csv 模块：```python
import csv
```

读取 CSV 文件

要读取 CSV 文件，请使用 () 函数。此函数返回一个包含文件内容的 reader 对象。将文件名作为参数传递给此函数，如下所示：```python
with open('', 'r') as f:
reader = (f)
```

遍历 CSV 数据

reader 对象是一个迭代器，其中每一行表示 CSV 文件中的一行。要遍历数据，请使用一个 for 循环：```python
for row in reader:
# 每一行都是一个列表，其中包含 CSV 文件中的数据
print(row)
```

处理 CSV 数据

遍历 CSV 数据后，您可以对其进行处理，例如提取特定列或执行聚合操作。使用 Python 中常见的列表操作来实现这一点：```python
# 获取第二列的数据
column2 = [row[1] for row in reader]
# 计算列中值的总和
total = sum(int(value) for value in column2)
```

写入 CSV 文件

您还可以使用 Python 将数据写入 CSV 文件。为此，请使用 () 函数，它返回一个 writer 对象。将文件名作为参数传递给此函数，如下所示：```python
with open('', 'w') as f:
writer = (f)
```

写入 CSV 数据

writer 对象提供了一个 writerow() 方法，该方法允许您将一行数据写入 CSV 文件。将要写入的数据作为列表传递给此方法，如下所示：```python
(['Name', 'Age', 'Occupation'])
(['John', 30, 'Engineer'])
```

定制 CSV 文件

可以通过调整 () 和 () 函数的可选参数来定制 CSV 文件。这些参数包括分隔符、换行符以及引号字符。```python
reader = (f, delimiter=';', quotechar='"')
writer = (f, delimiter=';', quotechar='"')
```

使用 Python 导入、处理和写入 CSV 文件非常简单。通过遵循本文中概述的步骤，您可以轻松地操作 CSV 数据并满足您的特定要求。通过使用 csv 模块的强大功能，您可以简化数据处理任务并从您的数据中提取有价值的见解。

2024-10-16

上一篇：用 Python 获取函数名

下一篇：Python 文件头：功能、类型和最佳实践