Come faccio a salvare e ripristinare più variabili in Python?

Ho bisogno di salvare circa una dozzina di oggetti in un file e quindi ripristinarli in un secondo momento. Ho provato a usare un ciclo for con pickle e shelve ma non ha funzionato bene.

Modificare.
Tutti gli oggetti che stavo cercando di salvare erano nella stessa classe (avrei dovuto menzionarlo prima) e non mi rendevo conto che avrei potuto salvare l'intera classe in questo modo:

import pickle
def saveLoad(opt):
    global calc
    if opt == "save":
        f = file(filename, 'wb')
        pickle.dump(calc, f, 2)
        f.close
        print 'data saved'
    elif opt == "load":
        f = file(filename, 'rb')
        calc = pickle.load(f)
    else:
        print 'Invalid saveLoad option'

Se devi salvare più oggetti, puoi semplicemente metterli in un unico elenco, o tupla, ad esempio:

import pickle

# obj0, obj1, obj2 are created here...

# Saving the objects:
with open('objs.pkl', 'w') as f:  # Python 3: open(..., 'wb')
    pickle.dump([obj0, obj1, obj2], f)

# Getting back the objects:
with open('objs.pkl') as f:  # Python 3: open(..., 'rb')
    obj0, obj1, obj2 = pickle.load(f)

Se hai molti dati, puoi ridurre la dimensione del file passando protocol=-1a dump(); pickleutilizzerà quindi il miglior protocollo disponibile invece del protocollo storico predefinito (e più compatibile con le versioni precedenti). In questo caso, il file deve essere aperto in modalità binaria ( wbe rb, rispettivamente).

La modalità binaria dovrebbe essere usata anche con Python 3, poiché il suo protocollo predefinito produce dati binari (cioè non di testo) (modalità di scrittura 'wb'e modalità di lettura 'rb').

C'è una libreria incorporata chiamata pickle. Usando picklepuoi scaricare gli oggetti in un file e caricarli in seguito.

import pickle

f = open('store.pckl', 'wb')
pickle.dump(obj, f)
f.close()

f = open('store.pckl', 'rb')
obj = pickle.load(f)
f.close()

Dovresti guardare i moduli scaffale e sottaceto . Se hai bisogno di memorizzare molti dati, potrebbe essere meglio usare un database

Un altro approccio per salvare più variabili in un file pickle è:

import pickle

a = 3; b = [11,223,435];
pickle.dump([a,b], open("trial.p", "wb"))

c,d = pickle.load(open("trial.p","rb"))

print(c,d) ## To verify

È possibile utilizzare klepto, che fornisce la memorizzazione nella cache persistente alla memoria, al disco o al database.

dude@hilbert>$ python
Python 2.7.6 (default, Nov 12 2013, 13:26:39) 
[GCC 4.2.1 Compatible Apple Clang 4.1 ((tags/Apple/clang-421.11.66))] on darwin
Type "help", "copyright", "credits" or "license" for more information.
>>> from klepto.archives import file_archive
>>> db = file_archive('foo.txt')
>>> db['1'] = 1
>>> db['max'] = max
>>> squared = lambda x: x**2
>>> db['squared'] = squared
>>> def add(x,y):
...   return x+y
... 
>>> db['add'] = add
>>> class Foo(object):
...   y = 1
...   def bar(self, x):
...     return self.y + x
... 
>>> db['Foo'] = Foo
>>> f = Foo()
>>> db['f'] = f  
>>> db.dump()
>>> 

Quindi, dopo il riavvio dell'interprete ...

dude@hilbert>$ python
Python 2.7.6 (default, Nov 12 2013, 13:26:39) 
[GCC 4.2.1 Compatible Apple Clang 4.1 ((tags/Apple/clang-421.11.66))] on darwin
Type "help", "copyright", "credits" or "license" for more information.
>>> from klepto.archives import file_archive
>>> db = file_archive('foo.txt')
>>> db
file_archive('foo.txt', {}, cached=True)
>>> db.load()
>>> db
file_archive('foo.txt', {'1': 1, 'add': <function add at 0x10610a0c8>, 'f': <__main__.Foo object at 0x10510ced0>, 'max': <built-in function max>, 'Foo': <class '__main__.Foo'>, 'squared': <function <lambda> at 0x10610a1b8>}, cached=True)
>>> db['add'](2,3)
5
>>> db['squared'](3)
9
>>> db['f'].bar(4)
5
>>> 

Ottieni il codice qui: https://github.com/uqfoundation

Il seguente approccio sembra semplice e può essere utilizzato con variabili di dimensioni diverse:

import hickle as hkl
# write variables to filename [a,b,c can be of any size]
hkl.dump([a,b,c], filename)

# load variables from filename
a,b,c = hkl.load(filename)