将numpy数组列表存储到本地文件中?

时间:2017-03-12 15:48:31

标签: python arrays list numpy

我目前正在研究神经网络,预处理步骤需要一段时间才能完成,特别是当我想调整网络结构的参数时。我的输入数据由numpy数组列表组成,存储在4个不同的变量中。可以将这些numpys列表存储到本地文件中,这样我就可以加载它们并开始训练而不是加载原始文件并开始预处理。

1 个答案:

答案 0 :(得分:1)

您可以使用numpy.save(或numpy.savez一次保存多个数组)来保存numpy数组,然后使用numpy.load重新加载它们。

e.g。保存数组:

import numpy
test_array_1 = numpy.ones([5, 6]) #creating a test array
test_array_2 = numpy.zeros([6, 7]) #creating a test array
numpy.savez("testfile.npz", test_array_1=test_array_1, test_array_2=test_array_2) #saving my two test arrays

e.g。加载数组

import numpy
data = numpy.load("testfile.npz") #loading the two arrays into a variable called 'data'
print(data["test_array_1"]) #using one array directly
test_array_2 = data["test_array_2"] #assigning one to a varibale