admin管理员组

文章数量:829195

csv文件的读与写

csv文件的读取与写入

1、csv文件的读取

import pandas as pdcsv_path = './dataA1.csv'
df = pd.read_csv(csv_path)# 可以通过csv文件的表头读取出某一列,表头见下图
print('item_id列前5行内容:')
print(df['item_id'][0:5])# 或者通过转化为array数组,通过下表索引读取
data = np.array(df)
print('读取前5行:')
print(data[0:5, :])
print(data.shape)


运行效果:

2、读取CSV文件表头、读取无表头csv文件

pd.read_csv默认不读取csv文件的第一行。
对于无表头csv文件可能丢失第一行数据,处理方法:header = None

csv_path = './A/dataA1.csv'
df = np.array(pd.read_csv(csv_path, header=None))
print(df[0])

运行效果:

3、csv文件的写入,以及设置表头

data = np.array([1,3,4,5],[2,3,4,5],[3,5,2,4])
head = 'line1,line2,line3'  # 设置csv文件表头
csv_save_path = './csv_save.csv'
np.savetxt(csv_save_path, data, fmt="%s", delimiter=',', header=head)

运行结果:

4、存入的csv文件表头含有#号(如上图),表示csv文件表头文件的注释符号是#

去掉#号处理方式:comment = ‘’

data = np.array([[1,3,4,5],[2,3,4,5],[3,5,2,4]])
head = 'line1,line2,line3,line4'  # 设置csv文件表头
csv_save_path = './csv_save.csv'
np.savetxt(csv_save_path, data, fmt="%s", delimiter=',', header=head,comments='')

运行效果

5、其他参数

np.savetxt()中的默认fmt参数是'%.18e',它适用于数字(整数/浮点数)。
如果想保存字符串,则需要将fmt参数更改为'%s'

本文标签: csv文件的读与写