您描述的基本上是一个固定长度的双端队列(通常称为双端队列)。
你可以使用 Python 的内置双端队列:
from collections import deque
d = deque(maxlen=2)
for i in range(10):
d.append(i)
print(d)
# deque([0], maxlen=2)
# deque([0, 1], maxlen=2)
# deque([1, 2], maxlen=2)
# deque([2, 3], maxlen=2)
# deque([3, 4], maxlen=2)
# deque([4, 5], maxlen=2)
# deque([5, 6], maxlen=2)
# deque([6, 7], maxlen=2)
# deque([7, 8], maxlen=2)
# deque([8, 9], maxlen=2)
您也可以使用appendleft 代替append:
for i in range(10):
d.appendleft(i)
print(d)
# deque([0], maxlen=2)
# deque([1, 0], maxlen=2)
# deque([2, 1], maxlen=2)
# deque([3, 2], maxlen=2)
# deque([4, 3], maxlen=2)
# deque([5, 4], maxlen=2)
# deque([6, 5], maxlen=2)
# deque([7, 6], maxlen=2)
# deque([8, 7], maxlen=2)
# deque([9, 8], maxlen=2)
或者,如果你想要它,你可以继承列表并自己实现append。请注意略有不同的结果:
class MyList(list):
def __init__(self, max_len, *args, **kwargs):
super().__init__(*args, **kwargs)
self.max_len = max_len
def append(self, obj):
if len(self) < self.max_len:
super().append(obj)
else:
self.insert(0, obj) # inserting to the left
self.pop() # deleting the last element on the right
li = MyList(2)
for i in range(10):
li.append(i)
print(li)
# [0]
# [0, 1]
# [2, 0]
# [3, 2]
# [4, 3]
# [5, 4]
# [6, 5]
# [7, 6]
# [8, 7]
# [9, 8]