python - 在 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 在这种情况下不是一个选项。