Python CSV 文件处理:删除指定列并生成新文件
Python CSV 文件处理:删除指定列并生成新文件
本文将介绍如何使用 Python 代码读取 CSV 文件,删除其中一列,并生成新的 CSV 文件。
实现思路:
- 读取原始 CSV 文件,将每一行数据存储为一个列表。
- 遍历每个列表,删除指定列的数据。
- 将处理后的数据写入新的 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)
说明:
csv.reader()函数可以读取 CSV 文件中的每一行数据,并将其转化为一个列表,其中每个元素表示一个单元格的内容。csv.writer()函数可以将列表中的数据写入到 CSV 文件中。del关键字可以删除列表中指定索引的元素。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 文件的列删除操作。
原文地址: https://www.cveoy.top/t/topic/m1Iq 著作权归作者所有。请勿转载和采集!