Python | Numpy | 读取文件和存储数据

使用Numpy读取文件

1、纯数字的数组

arr = np.loadtxt('data.txt',delimiter=',')
print(arr)

[[1. 2. 3.]
 [4. 5. 6.]
 [7. 8. 9.]]

2、含有字符串的数组

arr = np.loadtxt('data.txt',str,delimiter=',')
print(arr)

[['name' 'score']
 ['Liyuan' '90']
 ['Chenya' '85']]

多文件一起存储，读取

arr1 = np.array([1, 2, 3]) #假设有三个ndarray
arr2 = np.eye(5)
arr3 = np.zeros((3, 4))
np.savez('arr.npz', x0=arr1, x1=arr2, x2=arr3)  #将其保存
f = np.load('arr.npz')  #加载
print(f['x1'])    #读取

[[1. 0. 0. 0. 0.]
 [0. 1. 0. 0. 0.]
 [0. 0. 1. 0. 0.]
 [0. 0. 0. 1. 0.]
 [0. 0. 0. 0. 1.]]