熊猫不会从数据框中删除列

时间:2018-08-08 14:42:42

标签: python pandas

我试图用:删除列

prediction.drop('1',axis=1)

但是使用以下命令将数据帧保存到csv之后:

prediction.to_csv(path_or_buf='./path/file.csv')

我打开文件并看到以下内容:

,TARGET,1
0,1.0,0.0
1,1.0,0.0
2,1.0,0.0

大熊猫还认为第一列根本不存在(一个名为“”的列)。 熊猫对任何数据框都执行此操作。 如果我会尝试:

prediction.drop('',axis=1)

它将抛出:

KeyError: "[''] not found in axis"

这是生成数据的代码:

from keras.models import load_model
import pandas as pd
import numpy as np
model = load_model(filepath='./DRISK/model_mlp_a2.h5')
TEST = pd.DataFrame.from_csv('./DRISK/application_test.csv')
expanded = np.zeros((48744,4))
def encode_array(features):
    features = features.fillna(0)
    features = pd.get_dummies(features)
    features = np.array(features)
    features = np.append(features,expanded)
    features = np.reshape(features,(48744,251))
    return features
TEST_X = encode_array(TEST)
prediction = model.predict(x=TEST_X,verbose=1)
prediction = pd.DataFrame(prediction,columns=['TARGET','1'])
prediction.drop('',axis=1)
prediction.to_csv(path_or_buf='./DRISK/predictions.csv')

我广泛搜索了这个问题,但无济于事。 附言此外,它会忽略其他所有数据框中的第一列。

0 个答案:

没有答案