使用 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']

Save vectors to file in Python with NumPy