如何在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)