python - 列出当前在 Python 解释器中所有已知的 NumPy 数组

Question

我想要一个由我的 Python 解释器当前管理的所有 numpy 数组的集合。以前的答案表明 gc.get_objects() 应该为我做这件事。但是我没有得到预期的结果:

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

有没有办法让所有已知的 NumPy 数组(或任何类型的对象)当前实例化？

Answer 1

这不是一个完美的答案，但根据 this stackoverflow post你可以使用 locals()获取所有本地声明变量的字典。

然后您可以使用简单的字典理解获取所有已知的 NumPy 数组:

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

然后您可以随心所欲地遍历字典。