列表理解从元组列表中提取多个字段

Question

我有一个元组列表

servers = [('server1', 80 , 1, 2), ('server2', 443, 3, 4)]

我想创建一个只有前两个字段的新列表,如:

 [('server1', 80), ('server2', 443)]

但我看不出如何为多个元素制作列表理解.

hosts = [x[0] for x in servers]  # this works to give me ['server1', server2']

hostswithports = [x[0], x[1] for x in servers] # this does not work

我更喜欢学习使用循环的pythonic方式 - 我做错了什么？

Answer 1

您可以使用扩展的可迭代解包.

>>> servers = [('server1', 80 , 1, 2), ('server2', 443, 3, 4)]
>>> [(server, port) for server, port, *_ in servers]
[('server1', 80), ('server2', 443)]

使用_作为一次性占位符名称是一种常见的约定.

Answer 2

你在做什么几乎是正确的.您试图将列表中的每个元组转换为新元组.但你忘了实际宣布元组.这就是括号正在做的事情:

hosts = [(x[0], x[1]) for x in servers]

Answer 3

使用基本切片,如果您的任何列表元素没有预期数量的子元素,则具有不失败的好处.

[el[:2] for el in servers]

[('server1', 80), ('server2', 443)]

Answer 4

你可以使用itemgetter:

from operator import itemgetter


servers = [('server1', 80 , 1, 2), ('server2', 443, 3, 4)]

result = list(map(itemgetter(0, 1), servers))

print(result)

产量

[('server1', 80), ('server2', 443)]

更具可读性的替代方案如下:

from operator import itemgetter

get_server_and_port = itemgetter(0, 1)
servers = [('server1', 80, 1, 2), ('server2', 443, 3, 4)]
result = [get_server_and_port(e) for e in servers]

print(result)  # [('server1', 80), ('server2', 443)]