使用 Python 从 csv 文件中删除特定列答案

【问题标题】：Delete specific column from csv file with Python使用 Python 从 csv 文件中删除特定列
【发布时间】：2019-11-28 13:24:54
【问题描述】：

我有一个函数可以选择我想要使用的列的索引（根据列标题中的文本自动选择），结果是一个类似于以下示例中的列表：

>>>print(usable_data_index)
[0, 2, 5, 7] #It may be like this
[0, 3, 4, 5, 6, 10] #Or like this one
[1, 2] #Or this
#It may vary.

然后我只想将索引在 usable_data_index 中的列写入新文件（这是我不知道该怎么做的）。我知道我可以这样写特定的列：

for row in csv_reader:
            csv_writer.writerow((row[0], row[1], row[3], row[4]))

但在这种情况下，我必须手动指定索引。

【问题讨论】：

【解决方案1】：

在你的 for 循环中添加一个名为 rows_to_add 的元组并将其传递给 csv writer：

rows_to_add=(row[index] for index in usable_data_index)
csv_writer.writerow(rows_to_add)

【讨论】：