将列放在DataFrame中，然后写入输出

Question

尝试使用熊猫在DataFrame中删除列。该代码无法正常工作，我真的不知道为什么。简单快捷。我想念什么？

import pandas

df = pandas.read_csv('data.csv')
for col in df.columns.values:
    if input('Delete ' + col + '? yes/no') == 'yes':
        df.drop([col], axis=1)

df.to_csv('out.csv')

Answer 1

像这样工作。感谢@julian。

import pandas

df = pandas.read_csv('data.csv')
for col in df.columns.values:
    if input('Delete ' + col + '? yes/no') == 'yes':
        df = df.drop([col], axis=1)

df.to_csv('out.csv')

Answer 2

您可以对“拖放”使用“就地”选项，如下所示：

df = pandas.read_csv('data.csv')
for col in df.columns.values:
   if input('Delete ' + col + '? yes/no') == 'yes':
      df.drop([col], axis = 1, inplace = TRUE)

df.to_csv('out.csv')