удаление столбца из файла CSV в Python

Question

Я пытаюсь удалить повторяющийся столбец из файла csv в Python3. Я могу заставить код работать без ошибок... однако, когда я пытаюсь распечатать количество обработанных строк, я получаю пустой ответ в codio.

Ответ на бег возвращается как 1001codio@sigma-portal:~/workspace$

Может ли кто-нибудь указать, как я могу это исправить? В идеале я бы хотел, чтобы он напечатал, сколько столбцов было удалено.

import csv
import re
data = []

import csv

input_file = 'customerdata.csv'
output_file = 'outputcustomerdata.csv'
cols_to_remove = [11] # Column indexes to be removed (starts at 0)

cols_to_remove = sorted(cols_to_remove, reverse=True) # Reverse so we remove from the end first
row_count = 0 # Current amount of rows processed

with open(input_file, "r") as source:
   reader = csv.reader(source)
   with open(output_file, "wt") as result:
       writer = csv.writer(result)
       for row in reader:
           row_count += 1
           print('\r{0}'.format(row_count), end='') # Print rows processed
           for col_index in cols_to_remove:
               del row[col_index]
           writer.writerow(row)

Answer 1

Попробуй панд

import pandas as pd
df = pd.read_csv('customerdata.csv')
df = df.drop('column_name', axis=1)

# save your file
df.to_csv('outputcustomerdata.csv', encoding='utf-8')

print(df)