python读取、写入csv文件

版权声明：本文为博主原创文章，未经博主允许不得转载。 https://blog.csdn.net/char_QwQ/article/details/83002407

总是记不住怎么读取csv文件，每次都是上网查，写个博客记录下来看看会不会记得更清楚。

个人比较喜欢用pandas的read_csv函数来读取csv文件

import pandas as pd
train_data = pd.read_csv('./data/train.csv')

读取后的数据是DataFrame格式：

可以使用values获取值：

train_data = train_data.values

然后就变成这样了：

写入csv文件：

data = pd.DateFrame(train_data)

data.to_csv("result.csv",index=False,sep=',')