有什么办法可以改变我的班级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() 函数也是如此。有什么解决办法吗?