在numpy矩阵中初始化对象

Question

我想要一个充满唯一对象的numpy矩阵。 目前，我正在创建列表列表，然后将其转换为numpy数组（请参见下面的代码，在解决方法下）。 我使用它是因为我想使用切片来访问矩阵中的元素

我想知道是否有更好的方法来创建这样的矩阵。

import random
import numpy as np

class RandomCell(object):
    def __init__(self):
        self.value = random.randint(0, 10)
    def __repr__(self):
        return str(self.value)

# workaround
temp_matrix = [[RandomCell() for row in range(3)] for col in range(3)]
workaround_matrix = np.array(temp_matrix)

编辑：我想创建对象矩阵而不是生成随机数矩阵

Answer 1

从列表列表构建数组的方法很好。 另一种选择是

arr = np.array([RandomCell() for item in range(9)]).reshape(3,3)

通常，为了节省内存，您可以使用np.fromiter从迭代器构建数组。 但是，由于此数组具有np.fromiter object ，不幸的是，在这种情况下， np.fromiter不是一个选项。

Answer 2

其实很简单

import numpy as np
np.random.randint(0, 10, (3,3))