如何从一个csv文件中提取某一列以写入另一个csv文件?

Question

如题所述

Answer 1

如何从一个CSV文件中提取某一列以写入另一个CSV文件？首先，需要明确你的CSV文件数据格式。以下是一个CSV文件示例，包含五个学生的学号和五个科目的成绩：

学号,科目1,科目2,科目3,科目4,科目5
001,90,85,92,88,94
002,88,91,89,93,86
003,93,95,90,92,96
004,87,89,91,90,95
005,91,94,89,93,92

要提取其中的某一列，例如提取“科目1”的成绩，并将其写入另一个CSV文件，可以按照以下步骤操作：

1. 新建一个CSV文件，用于存储提取的数据。

2. 使用Python的`csv`模块读取原始CSV文件。例如，要提取“科目1”的成绩，可以使用如下代码：

python
import csv

with open('原始CSV文件路径', 'r') as file:
reader = csv.reader(file)
data = [row for row in reader if row]

这行代码将读取原始CSV文件，并将数据存储在`data`列表中。这里使用了列表推导式来过滤掉空白行。

3. 提取所需的列（如“科目1”列），并创建一个新列表以存储提取的数据：

python
subject1_scores = [row[0] for row in data]

这行代码将提取所有学生的“科目1”成绩，并将它们存储在`subject1_scores`列表中。

4. 创建一个新CSV文件来存储提取的数据，并使用`csv`模块的`DictWriter`或`writer`来写入数据。以下代码示例展示了如何使用`csv.writer`写入提取的数据：

python
import csv

with open('新CSV文件路径', 'w', newline='') as file:
writer = csv.writer(file)
writer.writerow(['学号']) # 写入列标题
for score in subject1_scores:
writer.writerow([score]) # 写入数据

这段代码首先创建了一个新的CSV文件，并写入了列标题“学号”。然后，它遍历提取的数据，并将每一项写入CSV文件中。

通过上述步骤，你就可以从一个CSV文件中提取某一列数据，并将其写入另一个CSV文件。关键在于正确处理数据读取、提取和写入的逻辑。本示例代码基于Python编写，使用了`csv`模块来简化CSV文件的读写操作。