Python – 从第 j 列值中提取第 i 列值

有时候，在数据处理过程中，我们需要从一列数据中提取另一列数据。比如，我们有一列数据包含了名字和姓氏，但是我们需要分别得到名字和姓氏。在这种情况下，我们可以通过 Python 对列表和字符串的操作来完成。本文将介绍如何从第 j 列值中提取第 i 列值。

1. 读取 CSV 文件

首先，我们需要从 CSV 文件中读取数据。Python 提供了 csv 模块用于读取和写入 CSV 文件。

以下是一个示例 CSV 文件：

name,age,gender
John Smith,25,Male
Jane Doe,30,Female

我们可以使用以下代码将数据从 CSV 文件中读取到列表中：

import csv

data = []
with open('example.csv', 'r') as csvfile:
    reader = csv.reader(csvfile)
    for row in reader:
        data.append(row)

现在，我们已经将 CSV 文件的每一行数据存储到了一个列表中。

2. 提取第 i 列值

下一步，我们需要从第 j 列值中提取第 i 列值。

假设我们要提取第二列的值（也就是年龄），那么我们可以使用以下代码：

ages = [row[1] for row in data[1:]]

这行代码使用列表推导式提取第二列的值，并将结果存储到一个名为 ages 的列表中。注意，我们使用了 data[1:] 而不是 data，这是因为第一行是标题，不包含实际的数据。

3. 结论

使用以上代码，我们可以从 CSV 文件中提取指定列的值。这是一个基础的数据处理操作，非常实用。