使用NumPy将向量保存到Python文件中

Question

我有一个带有数值的变量，一个带字符串值的变量，以及两个用NumPy定义的向量

a = 10
b = "text string"
positions = np.array([])
forces = np.array([])

我想将这些值保存到文件中。我用http://docs.scipy.org/doc/numpy/reference/generated/numpy.savetxt.html用

保存了两个向量

np.savetxt('test.out', (positions,forces))

但我还需要存储a和b的值。

这怎么可能？

Answer 1

我个人建议使用numpy.savez和numpy.load。例如：

numpy.savez('test.npz', a=a, b=b, positions=positions, forces=forces)

您可以像这样重新加载：

data = numpy.load('test.npz')
a = data['a']