文章目录
csv介绍案例csv介绍
对于任何带有单字符分隔符的文件,可以使用python中的csv模块。
csv.Dialect的属性
对于多字符分隔文件,将无法使用csv模块,需要使用字符串的split方法或者正则表达式中的re.split进行行拆分或者其他清理工作。
完整内容查看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] = 1y[~(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()
生成文件如图: