Python3 (notebook) 数据集导出为CSV文件或其他文件

以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)#打印出测试集预测的数据

Python3 (notebook) 数据集导出为CSV文件或其他文件_第1张图片
我们需要将其转换为DataFrame结构的表格数据

df = pd.DataFrame(predict_data)
print(df)

第一列为索引,第二列为数据
Python3 (notebook) 数据集导出为CSV文件或其他文件_第2张图片
然后利用***to_csv***将数据导出

df.to_csv('1.csv')#括号里面的为我们刚开始创建的CSV表格文件

打开文件
Python3 (notebook) 数据集导出为CSV文件或其他文件_第3张图片

你可能感兴趣的:(Python3 (notebook) 数据集导出为CSV文件或其他文件)