如何在Python中保存自制数据集以便以后可以使用它？

Question

我从PNG图像创建了一个数组数组。如何在Python中保存此数据集，以便以后或在其他Python脚本中访问它而无需重新扫描所有图像？

Answer 1

您可以使用python的pickle库将数据转储到文件中。

import pickle

dataset = [1,2,3,4]

with open('my_dataset.pickle', 'wb') as output:
    pickle.dump(dataset, output)

然后你可以把它装回另一个脚本。

import pickle

with open('my_dataset.pickle', 'rb') as data:
    dataset = pickle.load(data)