在一个文件中保存许多 numpy 变量

Question

我有一个np.loadtxt()从不同文件夹收集数据的脚本。我希望能够将它们全部保存在当前文件夹中，而不必savetxt()为每个文件夹做一个。目的是有一个可移植的文件来在不同的计算机上执行脚本。

Answer 1

您可以save像这样使用 NumPy函数：

import numpy as np

# Create sample data

a = np.array([1.0, 2.0, 3.0])

b = np.array([4.0, 5.0])

# Save to file

np.save('myfile.npy', [a, b])

# Load back in

a2, b2 = np.load('myfile.npy')

请注意，虽然文档没有明确说明这一点，但您确实可以在单个文件中存储多个不同大小的数组，如上所述（实际上这是通过将每个数组存储为 dtype 的一维数组的元素来完成的object，但不要介意执行）。