初始化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 从迭代器构建数组。但是，由于该数组的 dtype 为 object，不幸的是，np.fromiter 在这种情况下不是一个选项。

Answer 2

其实很简单

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