有什么办法可以改变我的班级list()函数的工作方式吗?

我正在为我的学校作业写一个小班“AP”(“算术进步”)。我已经包含了一个使 AP 无限长的选项。

class AP:
    def __init__(self, first_term: float, co_diff: float, n: int = inf) -> None:
        self.a1 = first_term
        self.d = co_diff
        self.term = 0
        self.n = n

    def __repr__(self) -> str:
        return f'AP({self.a1}, {self.d}, {self.n})'

    def __str__(self) -> str:
        output_str = f"{', '.join([str(term) for term in list(self.terms(term=6))])}, ...."
        if self.n != inf:
            return output_str + f', {(self.a1 + ((self.n - 1) * self.d))}'
        return output_str

    def terms(self, term: int):
        return AP(self.a1, self.d, n=term)

    def __iter__(self):
        return self

    def __next__(self):
        self.term += 1
        if self.term < self.n:
            return (self.a1 + ((self.term - 1) * self.d))
        raise StopIteration

    def __eq__(self, o: object) -> bool:
        if isinstance(o, AP):
            return str(self) == str(o)

但是当有人想把它转换成一个列表,如果它是一个无限的列表,我想提出一个错误,如:

 ValueError: The list() function was called for an object of undefined length.

sum() 函数也是如此。有什么解决办法吗?

以上是有什么办法可以改变我的班级list()函数的工作方式吗?的全部内容。
THE END
分享
二维码
< <上一篇
下一篇>>