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

Question

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

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

我想将这些值保存到文件中。 我已经使用http://docs.scipy.org/doc/numpy/reference/generation/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']