利用numpy.save函数保存Python中的数组数据

一、numpy.save函数的介绍

import numpy as np

# 创建一个数组
a = np.array([[1,2,3],[4,5,6]])

# 使用numpy.save函数保存数组到文件
np.save('a.npy', a)

# 加载保存的数组
b = np.load('a.npy')
print(b)

numpy.save函数是numpy库中提供的一个函数，用来将一个数组保存到文件中，在需要的时候，我们可以从文件中读取数组数据。该函数的调用方式非常简单。其中第一个参数表示保存到的文件名，第二个参数表示要保存的数组。

二、使用numpy.save保存不同形状的数组

import numpy as np

# 保存一维数组
a = np.array([1,2,3])
np.save('a.npy', a)

# 保存二维数组
b = np.array([[1,2,3],[4,5,6]])
np.save('b.npy', b)

# 保存三维数组
c = np.array([[[1,2],[3,4]],[[5,6],[7,8]]])
np.save('c.npy', c)

# 加载保存的数组
a_load = np.load('a.npy')
b_load = np.load('b.npy')
c_load = np.load('c.npy')

print(a_load)
print(b_load)
print(c_load)

numpy.save函数可以保存不同形状的数组，比如一维数组、二维数组、三维数组等等。不同形状的数组可以通过给numpy.save传递不同的数组参数来保存。

三、使用numpy.save保存不同类型的数组

import numpy as np

# 保存整型数组
a = np.array([1,2,3])
np.save('a.npy', a)

# 保存浮点型数组
b = np.array([[1.0,2.0,3.0],[4.0,5.0,6.0]])
np.save('b.npy', b)

# 保存复数数组
c = np.array([[1+2j,3+4j],[5+6j,7+8j]])
np.save('c.npy', c)

# 加载保存的数组
a_load = np.load('a.npy')
b_load = np.load('b.npy')
c_load = np.load('c.npy')

print(a_load)
print(b_load)
print(c_load)

numpy.save函数同样可以保存不同类型的数组。整型、浮点型、复数等类型的数组都可以被保存到文件中。在加载这些数组时需要注意它们的数据类型。

四、使用numpy.savez保存多个数组

import numpy as np

# 保存多个数组
a = np.array([1, 2, 3])
b = np.array([4, 5, 6])
np.savez('ab.npz', a=a, b=b)

# 加载保存的多个数组
data = np.load('ab.npz')
print(data['a'])
print(data['b'])

numpy.savez函数可以保存多个数组到同一个文件中，保存时需要将每个数组命名，加载时也需要指定相应的数组名进行获取。