如何YAML保存numpy数组

Question

嗨,有没有办法如何保存numpy 2D阵列,然后从文件安全加载它？我有这个数组:

     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拯救？