Python CSV 文件处理:删除指定列并生成新文件

本文将介绍如何使用 Python 代码读取 CSV 文件,删除其中一列,并生成新的 CSV 文件。

实现思路:

  1. 读取原始 CSV 文件,将每一行数据存储为一个列表。
  2. 遍历每个列表,删除指定列的数据。
  3. 将处理后的数据写入新的 CSV 文件中。

代码实现:

import csv

# 原始 CSV 文件路径,替换为自己的文件路径
input_file = 'input.csv'

# 新的 CSV 文件路径,替换为自己想保存的路径
output_file = 'output.csv'

# 要删除的列的索引,索引从 0 开始
column_index_to_delete = 1

# 读取原始 CSV 文件
with open(input_file, 'r', newline='') as input_csv_file:
    reader = csv.reader(input_csv_file)
    rows = [row for row in reader]

# 遍历每个列表,删除指定列的数据
for row in rows:
    del row[column_index_to_delete]

# 将处理后的数据写入新的 CSV 文件中
with open(output_file, 'w', newline='') as output_csv_file:
    writer = csv.writer(output_csv_file)
    writer.writerows(rows)

说明:

  1. csv.reader() 函数可以读取 CSV 文件中的每一行数据,并将其转化为一个列表,其中每个元素表示一个单元格的内容。
  2. csv.writer() 函数可以将列表中的数据写入到 CSV 文件中。
  3. del 关键字可以删除列表中指定索引的元素。
  4. newline='' 参数可以在写入 CSV 文件时去除空白行。

示例:

假设 input.csv 文件内容如下:

Name,Age,City
John,25,New York
Jane,30,London
David,28,Paris

执行代码后,output.csv 文件内容将变为:

Name,City
John,New York
Jane,London
David,Paris

总结:

通过上述代码,您可以轻松地读取 CSV 文件、删除指定列并生成新的 CSV 文件。根据您的实际需求修改代码中的文件路径和要删除的列索引,即可实现 CSV 文件的列删除操作。

Python CSV 文件处理:删除指定列并生成新文件

原文地址: https://www.cveoy.top/t/topic/m1Iq 著作权归作者所有。请勿转载和采集!

免费AI点我,无需注册和登录