带有 NumPy 和对象引用的 Python 3

Question

我需要创建一个大型矩阵(数组)结构(3 轴)，每个元素应存储对 Python 对象(myclass 实例)的引用。是否可以使用 numpy 来创建这样的数组。我应该使用哪种数据类型来存储 Python 引用？numpy 的优点是支持不同层次的切片。备选方案是创建嵌套(nested)列表，但这是一个麻烦的解决方案。

Answer 1

使用dtype=object在数组中存储Python对象。

import numpy as np

class Foo: pass

arr=np.empty(2,dtype=object)
arr[:]=[Foo(),Foo()]
print(arr)
# [<__main__.Foo instance at 0xb7827d0c>
#  <__main__.Foo instance at 0xb748b5ac>]

print(arr.dtype)
# object