以Titanic数据集的某两列数据为例,将其导出并打开
首先我们要创建一个CSV的表格文件,放在绝对路径里,我们最终要将数据导入到这个文件里面
from sklearn import model_selection
from sklearn import linear_model
from sklearn import metrics
predictor = ttnk.columns[1:]
X = ttnk[predictor]
y = ttnk.Survived
X_train,X_test,y_train,y_test = model_selection.train_test_split(X,y,test_size = 0.25,random_state = 1234)
model = linear_model.LogisticRegression()
model.fit(X_train,y_train)
predict_data = model.predict(X_test)
print(preict_data)#打印出测试集预测的数据
我们需要将其转换为DataFrame结构的表格数据
df = pd.DataFrame(predict_data)
print(df)
第一列为索引,第二列为数据
然后利用***to_csv***将数据导出
df.to_csv('1.csv')#括号里面的为我们刚开始创建的CSV表格文件
打开文件