Python中如何从可迭代对象返回的元组提取指定值赋值给变量

问题场景 #

我有一个可迭代对象，每次迭代返回一个元组，但我只关心其中某一个值。比如下面这个Triples类：

class Triples:
    def __init__(self):
        self.x = 0
    def __iter__(self):
        return self
    def __next__(self):
        self.x += 1
        if self.x < 5:
            return (self.x*2-1, self.x*2, self.x*2*-1)
        else:
            raise StopIteration

直接迭代它会输出完整元组：

(1, 2, -2)
(3, 4, -4)
(5, 6, -6)
(7, 8, -8)

但我只想得到每个元组的第二个元素：

2
4
6
8

我试过几种方法：

• 通过索引取值：

  for x in Triples():
      print(x[1])
  

• 先赋值再使用：

  for x in Triples():
      y = x[1]
      print(y)
  

• 用占位符变量：

  for (_,x,*_) in Triples():
      print(x)
  

但我希望能更明确地告知Python忽略其他值，而不是把它们存到不用的变量里（加del(_)既麻烦又没必要）。另外，一些错误尝试：

• 直接下标访问迭代器会报错：

  for x in Triples()[1]:
      print(x)
  

  错误：TypeError: 'Triples' object is not subscriptable
• 转成列表再取下标会得到错误结果：

  for x in list(Triples())[1]:
      print(x)
  

  输出的是第二个元组的所有元素：

  3
  4
  -4
  

想问有没有更优的实现方式，还是只能用已经试过的方法？

解答 #

推荐方案：使用operator.itemgetter配合惰性迭代 #

可以用operator.itemgetter来提取指定位置的元素，结合map或者生成器表达式，既能明确意图，又不会产生多余变量，且保持惰性迭代（不会提前生成所有元组）：

方法1：生成器表达式

from operator import itemgetter

for x in (itemgetter(1)(t) for t in Triples()):
    print(x)

方法2：map函数

from operator import itemgetter

for x in map(itemgetter(1), Triples()):
    print(x)

这两种方式都直接明确了“提取元组第二个元素”的逻辑，代码可读性高，效率和直接用索引相当。

自定义生成器封装逻辑 #

如果不想引入operator模块，也可以写一个简单的生成器函数来封装提取逻辑，后续修改提取位置时只需调整函数即可：

def extract_second(iterable):
    for item in iterable:
        yield item[1]

for x in extract_second(Triples()):
    print(x)

关于占位符_的补充 #

在Python中，_作为“不需要的变量”占位符是广泛接受的惯例，比如for _, x, _ in Triples()这种写法本身简洁且符合规范，不需要额外加del(_)——Python的垃圾回收会自动处理未被引用的变量，不会有性能负担，这种写法的可读性也足够清晰。

内容的提问来源于stack exchange，提问作者wfaulk