python 读写csv指定

您好！要在Python中读写CSV文件，可以使用内置的csv模块。以下是一个示例代码，演示如何读取和写入CSV文件的特定列：

import csv

def read_csv(filename, columns):
    data = []
    with open(filename, 'r') as file:
        reader = csv.DictReader(file)
        for row in reader:
            selected_columns = {col: row[col] for col in columns}
            data.append(selected_columns)
    return data

def write_csv(filename, data, columns):
    with open(filename, 'w', newline='') as file:
        writer = csv.DictWriter(file, fieldnames=columns)
        writer.writeheader()
        writer.writerows(data)

# Example usage
filename = 'data.csv'
columns_to_read = ['column1', 'column3']
columns_to_write = ['column1', 'column2']

# 读取特定列
data = read_csv(filename, columns_to_read)
print(data)

# 写入特定列的数据
write_csv('output.csv', data, columns_to_write)

在上面的示例中，`read_csv`函数接受文件名和要读取的列的列表作为参数，并返回一个包含选定列数据的列表。`write_csv`函数接受文件名、要写入的数据和要写入的列的列表作为参数，并将数据写入CSV文件。

请注意，示例中的代码假设CSV文件的第一行包含列标题。如果您的CSV文件没有列标题，您可以将`write_csv`函数中的`writer.writeheader()`行注释掉。

希望这可以帮助到您！如果有任何其他问题，请随时提问。