web-dev-qa-db-ja.com

pickleファイルの読み方

データをいくつか作成し、次のように何度か保存しました。

with open('filename', 'a') as f:
        pickle.dump(data, f)

ファイルのサイズが増加するたびに、ファイルを開くと

with open('filename', 'rb') as f:
    x = pickle.load(f)

前回のデータのみを見ることができます。ファイルを正しく読み取るにはどうすればよいですか？

pythonpickle
39
Kenenbek Arzymatov

Pickleは一度に1つのオブジェクトをシリアル化し、1つのオブジェクトを読み戻します-漬けられたデータはファイルに順番に記録されます。

単純にpickle.loadを実行すると、ファイルにシリアル化された最初のオブジェクトを読み取る必要があります（作成した最後のオブジェクトではありません）。

最初のオブジェクトのシリアル化を解除した後、ファイルポインターは次のオブジェクトの先頭にあります-pickle.loadをもう一度呼び出すと、次のオブジェクトを読み取ります-ファイルの最後までそれを行います。

objects = []
with (open("myfile", "rb")) as openfile:
    while True:
        try:
            objects.append(pickle.load(openfile))
        except EOFError:
            break
55
jsbueno

以下は、pickleファイルの書き込みおよび読み取り方法の例です。ファイルにpickleデータを追加し続ける場合、必要なものが見つかるまで、またはファイルの最後に到達して例外が生成されるまで、ファイルから読み取りを続ける必要があることに注意してください。それが最後の関数が行うことです。

import os
import pickle


PICKLE_FILE = 'pickle.dat'


def main():
    # append data to the pickle file
    add_to_pickle(PICKLE_FILE, 123)
    add_to_pickle(PICKLE_FILE, 'Hello')
    add_to_pickle(PICKLE_FILE, None)
    add_to_pickle(PICKLE_FILE, b'World')
    add_to_pickle(PICKLE_FILE, 456.789)
    # load & show all stored objects
    for item in read_from_pickle(PICKLE_FILE):
        print(repr(item))
    os.remove(PICKLE_FILE)


def add_to_pickle(path, item):
    with open(path, 'ab') as file:
        pickle.dump(item, file, pickle.HIGHEST_PROTOCOL)


def read_from_pickle(path):
    with open(path, 'rb') as file:
        try:
            while True:
                yield pickle.load(file)
        except EOFError:
            pass


if __== '__main__':
    main()
6
Noctis Skytower