在python中从CSV文件中删除列

Question

我正在尝试从Python3中的csv文件中删除重复的列。我能够让代码无误地运行……但是，当我试图打印处理的行数时，我收到一个codio中的空白响应

对running的响应返回为1001codio@sigma-门户：~/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)