如何初始化列表对象的 Numpy 数组

Question

我正在尝试创建一个类似于

的 numpy 数组

array([[list([]), list([])],
       [list([]), list([])],
       [list([]), list([])]], dtype=object)

此数组的形状为 (3,2)。但是，每当我这样做时

np.array([[list(), list()], [list(), list()], [list(), list()]])

我最终得到

array([], shape=(3, 2, 0), dtype=float64)

我该如何解决这个问题？

Answer 1

您可以使用以下内容：

np.frompyfunc(list, 0, 1)(np.empty((3,2), dtype=object))

我们首先将 list 变成不带参数的 ufunc 和 returns 一个空列表，然后将其应用于对象类型的空 3x2 数组。

How to initialize Numpy array of list objects