读取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文件时去除空白行。
原文地址: https://www.cveoy.top/t/topic/bxJq 著作权归作者所有。请勿转载和采集!