python - 类方法返回列表/数组

Question

从初始化为列表的类调用类方法时，我试图返回一个 numpy 数组。

import numpy as np

class test_object:

    def __init__(self, val=1):
        self.value = val

    @staticmethod
    def generate(number=5):

        # create list of test_objects
        obj = [test_object(val=i) for i in range(number)]

        return obj

    def powers(self, pow=2):
        return self.value**pow


numbers = test_object.generate(number=10)

array = np.zeros(len(numbers))

for i in range(len(numbers)):
    array[i] = numbers[i].powers()

print(array)

是否有可能修改类/类方法'powers'，它会像这样工作:

array = numbers.powers()

数组初始化和for循环应该在类方法中完成。

Answer 1

这是一个解决方案:

import numpy as np

class test_object():
    def __init__(self, val=1):
        self.value = val

    class test_object_list(list):
        def powers(self):
            return np.asarray([i.powers() for i in self], dtype=float)

    @classmethod
    def generate(cls, number=5):
        # create list of test_objects
        obj = cls.test_object_list([cls(val=i) for i in range(number)])
        return obj

    def powers(self, pow=2):
        return self.value**pow


numbers = test_object.generate(number=10)
array = numbers.powers()
print(array)

请注意，我已将 @staticmethod 替换为 @classmethod，因为它更合适。