Listet alle bekannten NumPy-Arrays im Python-Interpreter auf

Question

Ich möchte eine Sammlung aller numpy Arrays, die derzeit von meinem Python-Interpreter verwaltet werden. Frühere Antworten deuten darauf hin, dass gc.get_objects() dies für mich tun sollte. Allerdings habe ich nicht die erwarteten Ergebnisse erhalten:

In [1]: import numpy as np 

In [2]: import gc 

In [3]: x = np.ones(5) 

In [4]: any(item is x for item in gc.get_objects()) 
Out[4]: False 

Gibt es eine Möglichkeit alle bekannten NumPy Arrays (oder Objekte jeglicher Art wirklich) zu bekommen zur Zeit instanziiert?

Answer 1

Als Referenz von Jim Crist interne Kommunikation bei Continuum Analytics:

Dies erklärt, warum gc.get_objects nicht funktioniert: https://mail.scipy.org/pipermail/numpy-discussion/2014-September/071186.html

Numpy enthält ein Werkzeug für die Zuweisungen als Kontext-Manager Tracking. Dieses plus ein WeakSet könnte möglicherweise helfen, Ihr Problem zu lösen.

https://github.com/numpy/numpy/tree/master/tools/allocation_tracking

Answer 2

Ich bin mir nicht sicher, ob dies genau das Problem löst, das Sie zu lösen versuchen, aber was ist mit locals()? Das fängt an, zumindest zu bekommen, was vars vom Interpreter Ihrer aktuellen interaktiven Sitzung verfolgt werden? Diese

import numpy as np 
a = np.array([1, 2, 3]) 
b = np.array([2, 3, 4]) 
filter(lambda x : isinstance(x, np.ndarray), locals().values()) 

Answer 3

ist keine perfekte Antwort, aber nach this stackoverflow post können Sie locals() verwenden, um ein Wörterbuch aller lokal deklarierten Variablen zu erhalten.

Sie können dann alle bekannten NumPy Arrays erhalten mit einfachen dict Verständnis:

import numpy as np 
np_arrays = {k:v for k,v in locals().items() if isinstance(v, np.ndarray)} 

Sie dann über das Wörterbuch laufen kann aber Sie danach mögen.