python - 如何根据索引访问队列元素

Question

我有一个队列

from collections import deque
queue = deque([(1, 2), (5, 7), (3, 9)])

并给出索引。

我的功能如下:

def iterQue(que, index):
    data = []
    for d in que:
        data.append(d[index])
    return data

输出:

索引=0，得到[1,5,3]

索引=1，得到[2,7,9]

有没有更好的方法来修改我的函数？

Answer 1

您可以执行以下操作:

from collections import deque
queue = deque([(1, 2), (5, 7), (3, 9)])

li = [[] for _ in range(2)]

for item in queue:
    for i in range(2):
        li[i].append(item[i])

print(li)

for idx, item in enumerate(li):
    print('index = {}, get {}'.format(idx, item))

或者按照托比亚斯下面评论中的建议(感谢@tobias_k)

from collections import deque
queue = deque([(1, 2), (5, 7), (3, 9)])

li = [ list(item) for item in zip(*queue)]
print(li)

for idx, item in enumerate(li):
    print('index = {}, get {}'.format(idx, item))

两种情况下的输出如下所示

[[1, 5, 3], [2, 7, 9]]

index = 0, get [1, 5, 3]
index = 1, get [2, 7, 9]