Numpy数据存取_np.save

Numpy数据存取

• numpy提供了便捷的内部文件存取，将数据存为np专用的npy(二进制格式)或npz(压缩打包格式)格式
• npy格式以二进制存储数据的，在二进制文件第一行以文本形式保存了数据的元信息（维度，数据类型），可以用二进制工具查看查看内容
• npz文件以压缩打包文件存储，可以用压缩软件解压

import numpy as np
 
a = np.array([['张三','李四','王五','赵六'],['11','12','13','14','15']])
b = a = np.arange(24).reshape((2,3,4))
 
np.save('x.npy',a) # 存为.npy文件
np.savez("y.npz", ar0 = a, ar1 = b) # 多个数组存入一个.npz压缩包
 
c = np.load('x.npy') # .npy文件读入数组
c
 
d = np.load("y.npz") # .npz压缩包读入
d["ar0"] # 单独输出数组

---

CSV文件存取

Comma-Separated Value,逗号分隔值文件

一种数据存储格式

• 广泛支持
• 类表格
• 存储一维或二维数据
• 文本格式

CSV文件：

• 显示：表格状态
• 源文件：换行和逗号分隔行列的格式化文本

---

Numpy存储CSV文件

将ndarray数组写入CSV文件中

np.savetxt(frame,array,fmt='%.18e',delimiter=None)

• frame 存储文件、字符串或产生器的名字，可以是.gz或.bz2的压缩文件，对大型数据有用，压缩后存储或读取，节省存储资源
• array 存入文件的数组
• fmt 写入文件中每个元素的字符串格式，例如
  • %s (ASCII字符)
  • %d （整数）
  • %.2f（2位小数的浮点数）
  • %.18e（科学计数法，常用）
    • np各类型元素存储到CSV中都是字符串，字符串显示的格式，默认%.18e，科学计数法，保留18位小数的浮点数形式存储数据，需要根据情况修改
• delimiter 分隔字符串，默认是任何空格，需要改为 逗号

np的savetxt函数并不是专为生成csv文件用的，它可以生成任何带特定分隔字符的文本文件，但csv文件使用广泛，所以我们一般用此函数只生成csv

import numpy as np
 
a = np.arange(100).reshape(5,20)
np.savetxt('a.csv',a,fmt='%d',delimiter=',') #整数
np.savetxt('a.csv',a,fmt='%.1f',delimiter=',') #一位小数的浮点数
 
b = np.array([['a','b','c','d'],['11','12','13','14']])
np.savetxt('b.csv',b,fmt='%s',delimiter=',') #ASCII字符，不能存储非ASCII字符串

csv文件只能存储一维、二维数据，不能存储多维数据

import numpy as np
 
a = np.arange(24).reshape((2,12))
# a = np.arange(24).reshape((2,3,4)) # 存储多维数据出错
a
np.savetxt('y.csv',a,fmt='%d',delimiter=',')

---

Numpy读取CSV文件

将CSV文件数据读入ndarray数组

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

• frame 文件、字符串或产生器，可以是.gz或bz2压缩文件
• dtype 数据类型，可选，csv的字符串以什么数据类型读入数组中，默认np.float 浮点数
• delimiter 分隔字符串，默认是任何空格，改为 逗号
• skiprows 跳过前x行，一般跳过第一行表头
• usecols 读取指定的列，索引，元组类型
• unpack 如果True,读入属性将分别写入不同数组变量，False 读入数据只写入一个数组变量，默认False

import numpy as np
 
b = np.loadtxt('a.csv', delimiter=',') # 默认浮点型
b = np.loadtxt('a.csv', dtype=np.int, delimiter=',') #数据为整型
 
#b = np.loadtxt('a.txt', dtype=np.str, delimiter=',') #数据为字符串，输出默认带 b，要去掉用下面方式输出：
b = np.loadtxt('a.txt', dtype=bytes, delimiter=',').astype(str)
b = np.loadtxt('a.txt', dtype=bytes, delimiter=',',skiprows=1,usecols=(2,3)).astype(str) #跳过第一行，读入第3、4列
 
b