피클 기능

Question

피클 파일에 데이터를 저장하는 기능을 빌드하려고합니다.

with open('x.pickle', 'wb') as f: 
    pickle.dump(x, f) 

을하지만 실제로는 그것에서 함수 그래서 다시하고 다시 작성할 필요가 없습니다 만들고 싶어 : 나는 이런 식으로 피클을 사용하여 작동하는지 알고있다. 다음은 내가 시도한 것입니다 :

def pickle_dump(x): 
    with open('%s.pickle'%x, 'wb') as f: 
     pickle.dump(x, f) 
    return 

'with'를 사용하고 파일이 존재하지 않으면 파일을 만듭니다. 나는 함수 안에서도 작동 할 것이라고 생각했지만, 대신 File Not Found Error를 리턴했다.

도움을 주시면 감사하겠습니다.

Answer 1

문자열 경로 이름을 실제 객체와 혼동하는 것처럼 보입니다. 그들은 서로 다른 두 가지입니다.

def pickle_dump(path, x): 
    """Pickle object `x` to `path`.""" 
    with open('%s.pickle' % path, 'wb') as f: 
     pickle.dump(x, f) 

예 :

import pandas as pd 
df = DataFrame() # the actual object you want to pickle 
pickle_dump(path='files/filename', x=df) 

이 역에서 같은 일을하려면 :

을 :

지금

def pickle_load(path): 
    with open('%s.pickle' %path, 'rb') as f: 
     return pickle.load(f) 

당신이 함수의 반환 값에 변수를 할당 할이 실행

result = pickle_load('mypath/filename') 

commen에서 t, 함수 본문에 변수 x을 명명했습니다. 그러나 로컬 변수임을 깨달으십시오. 함수 밖에서는 액세스 할 수 없습니다. 그것은 return 문이 필요한 경우입니다.

Answer 2

개체와 파일 이름을 모두 전달하십시오.

import cPickle as pickle # Python 2 

def pickle_dump(obj, filename): 
    if filename.split('.')[-1] != 'p': 
     filename += '.p' # Add pickle file extension `.p` if needed. 
    with open(filename, 'wb') as f: 
     pickle.dump(obj, f, protocol=pickle.HIGHEST_PROTOCOL)