python怎么选取csv某几列

可以使用Python的CSV模块来读取CSV文件，并选择特定的列。下面是一个基本的示例代码，说明了如何选择CSV文件中的特定列：

import csv

def select_columns(csv_file, columns):
    with open(csv_file, 'r', newline='') as file:
        reader = csv.reader(file)
        selected_columns = []
        headers = next(reader)  # 读取并保存标题行

        # 确定选择的列的索引
        column_indices = [headers.index(column) for column in columns]

        for row in reader:
            selected_row = [row[index] for index in column_indices]
            selected_columns.append(selected_row)

    return selected_columns

# 例子使用
csv_file = 'data.csv'  # 替换为你的CSV文件路径
selected_columns = select_columns(csv_file, ['列1', '列3'])  # 替换为你想选择的列的名称

# 打印选择的列
for row in selected_columns:
    print(row)

请注意，上述代码假设CSV文件的第一行是标题行，其中包含列的名称。函数select_columns接受CSV文件路径和一个包含你想要选择的列名称的列表。它将返回一个包含选定列的列表。

在上面的代码中，我们使用了csv.reader来读取CSV文件的内容。首先，我们提取标题行，然后根据列名确定所选列的索引。接下来，我们遍历每一行并提取相应的列值，然后将它们添加到selected_columns列表中。

请确保将csv_file替换为你的CSV文件的实际路径，并在select_columns函数的调用中使用你想要选择的实际列名称。