python—csv写入文件

1147-柳同学

发表文章数:589

热门标签

,
首页 » Python » 正文

def write_result(c, c_type):
    file_name = 'Titanic.test.csv'
    x, passenger_id = loaddata(file_name, False)

    if type == 3:
        x = xgb.DMatrix(x)
    y = c.predict(x)
    y[y > 0.5] = 1
    y[~(y > 0.5)] = 0
	
	# 以写入方式打开文件夹
    predictions_file = open("Prediction_%d.csv" % c_type, "w")
    
    open_file_object = csv.writer(predictions_file)
    # csvwriter.writerows(rows)
    # 将参数 row 写入 writer 的文件对象,并根据当前设置的变种进行格式化。本方法的返回值就是底层文件对象 write 方法的返回值
    open_file_object.writerow(["PassengerId", "Survived"])
    open_file_object.writerows(zip(passenger_id, y))
    predictions_file.close()

生成文件如图:
python—csv写入文件

未经允许不得转载:作者:1147-柳同学, 转载或复制请以 超链接形式 并注明出处 拜师资源博客
原文地址:《python—csv写入文件》 发布于2021-01-25

分享到:
赞(0) 打赏

评论 抢沙发

评论前必须登录!

  注册



长按图片转发给朋友

觉得文章有用就打赏一下文章作者

支付宝扫一扫打赏

微信扫一扫打赏

Vieu3.3主题
专业打造轻量级个人企业风格博客主题!专注于前端开发,全站响应式布局自适应模板。

登录

忘记密码 ?

您也可以使用第三方帐号快捷登录

Q Q 登 录
微 博 登 录