在Python中获取数组中每行的第一个元素?

2024-04-20 11:41:25 发布

您现在位置:Python中文网/ 问答频道 /正文

假设我有一个元组数组,s,形式如下:

s = ((1, 23, 34),(2, 34, 44), (3, 444, 234))

我想返回另一个元组t,由每行的第一个元素组成:

t = (1, 2, 3)

哪种方法最有效?我当然可以遍历s,但是有没有更巧妙的方法呢?


Tags: 方法元素数组形式元组
3条回答

没有

t = tuple(x[0] for x in s)
import itertools
s = ((1, 23, 34),(2, 34, 44), (3, 444, 234))
print(next(itertools.izip(*s)))

itertools.izip返回迭代器。函数next返回迭代器中的下一个(在本例中是第一个)元素。

在Python 2.x中,zip返回一个元组。 izip使用更少的内存,因为迭代器直到需要时才生成其内容。

在Python 3中,zip返回迭代器。

伊格纳西奥给出的列表理解方法是最干净的。

仅仅是踢,你也可以做:

zip(*s)[0]

*ss展开为参数列表。所以它相当于

zip( (1, 23, 34),(2, 34, 44), (3, 444, 234))

并且zip返回n元组,其中每个元组包含每个列表中的nth项。

相关问题 更多 >