尝试使用熊猫在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')
答案 0 :(得分:0)
像这样工作。感谢@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')
答案 1 :(得分:0)
您可以对“拖放”使用“就地”选项,如下所示:
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')