python - 如何 YAML 保存 numpy 数组

Question

您好，有没有一种方法可以保存 numpy 二维数组，然后从文件中安全地加载它？我有这个数组:

     matrix = np.random.randint(2, size=(self.row,self.col))

但是我无法保存它

Answer 1

最明显的方法是安装 pyyaml 模块(Google App Engine 使用)，并将矩阵保存为列表。

import numpy as np
import yaml
matrix = np.random.randint(2, size=(10,7))
with open('stack21658676.yaml', 'w') as f:
    yaml.dump(matrix.tolist(), f)

with open('stack21658676.yaml') as f:
    loaded = yaml.load(f)
loaded = np.array(loaded)
print loaded

虽然我不得不问你为什么想要 YAML 而不是“本地”numpy 保存？