Python中数据的CSV文件存取

Python中数据的CSV文件存取

csv（逗号分隔值文件格式）用来存储批量数据

---

一.数据存储

np.savetxt(frame,  array,  fmt,  delimiter=None)

• frame : 文件、字符串或产生器,可以是.gz或.bz2的压缩文件
• array : 存入文件的数组
• fmt : 写入文件的格式,例如:%d %.2f %.18e
• delimiter : 分割字符串,默认是任何空格

import numpy as np

data = np.random.randint(1,20,(7,8))
print(data)
np.savetxt('first_csv_save.csv',data,fmt='%d',delimiter=' ')

二.数据读取

np.loadtxt(frame,  dtype=np.float,  delimiter=None, unpack=False)

• frame : 文件、字符串或产生器,可以是.gz或.bz2的压缩文件
• dtype : 数据类型,可选
• delimiter : 分割字符串,默认是任何空格
• unpack : 如果True,读入属性将分别写入不同变量

import numpy as np

load_data = np.loadtxt('first_csv_save.csv',dtype=np.int,delimiter=None,unpack=False)
print(load_data)

三.多维数据存储

a.tofile(frame,  sep='',  format='%s')

• frame : 文件、字符串
• sep : 数据分割字符串,如果是空串,写入文件为二进制
• format : 写入数据的格式

import numpy as np

data = np.random.randint(1,20,(2,3,4))
print(data)
data.tofile('3D_csv_save.csv',sep=" ",format="%d")

四.多维数据读取

np.fromfile(frame,  dtype=float,  count=‐1,  sep='')

• frame : 文件、字符串
• dtype : 读取的数据类型
• count : 读入元素个数,‐1表示读入整个文件
• sep : 数据分割字符串,如果是空串,写入文件为二进制

import numpy as np

load_data = np.fromfile('3D_csv_save.csv',dtype=int,count=-1,sep=" ")
print(load_data)