python提取csv文件几列并保存为csv

要提取csv文件的几列并保存为csv，可以使用以下代码：

import csv

# 要提取的列的索引
columns = [0, 3, 5]

# 读取原始csv文件
with open('原始文件.csv', 'r') as f:
    reader = csv.reader(f)
    header = next(reader)  # 获取表头
    
    # 获取要提取的列的表头
    header_new = [header[i] for i in columns]
    
    # 逐行读取数据并提取要保留的列
    rows_new = []
    for row in reader:
        row_new = [row[i] for i in columns]
        rows_new.append(row_new)
        
# 保存提取后的csv文件
with open('提取后的文件.csv', 'w', newline='') as f:
    writer = csv.writer(f)
    writer.writerow(header_new)
    writer.writerows(rows_new)

在这个代码中，首先定义了要提取的列的索引，然后使用csv模块读取原始csv文件，获取表头和数据。接着，根据要提取的列的索引，获取要保留的列的表头和数据，并保存到一个新的列表中。最后，使用csv模块将新的表头和数据写入到新的csv文件中。

相关问题

python提取csv文件几列

可以使用Python的csv模块来读取csv文件，并根据需要提取特定的列。

以下是一个示例代码，假设我们有一个名为data.csv的csv文件，其中包含三列数据，分别是id、name和age。我们想要提取name和age这两列数据。

import csv

with open('data.csv', 'r') as csvfile:
    reader = csv.reader(csvfile)
    for row in reader:
        name = row[1]
        age = row[2]
        print(name, age)

在这个示例中，我们使用了Python的内置函数open来打开csv文件，并使用csv.reader来读取文件内容。然后，我们使用一个for循环遍历每一行数据，从每一行中提取第2列和第3列数据（即name和age），并打印出来。

需要注意的是，csv文件中的每一行数据都是一个列表，列表的第一个元素表示第1列数据，第二个元素表示第2列数据，以此类推。因此，我们可以通过列表索引来获取特定的列数据。

python 提取csv文件某几列

在 Python 中，可以使用 csv 模块来读取 CSV 文件并提取其中的某几列。

下面是一个简单的示例，展示了如何提取 CSV 文件中的第一列和第三列：

import csv

# 打开 CSV 文件
with open('data.csv', 'r') as f:
    # 创建 CSV 阅读器
    reader = csv.reader(f)
    # 读取每行数据
    for row in reader:
        # 提取第一列和第三列数据
        col1 = row[0]
        col3 = row[2]
        # 处理数据
        ...

注意：上面的代码假设 CSV 文件中的每行数据都是用逗号分隔的。如果 CSV 文件使用其他字符来分隔数据，那么可以使用 csv.reader() 的 delimiter 参数来指定分隔符。

例如，如果使用分号来分隔数据，可以这样创建 CSV 阅读器：

reader = csv.reader(f, delimiter=';')

如果想要更高级的 CSV 处理功能，可以使用 pandas 库。 Pandas 可以轻松地读取 CSV 文件，并提供了丰富的数据分析功能。