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

实际上非常简单

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

Answer 2

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

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

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

在numpy矩阵中初始化对象

2 个答案: