Python“所有其他元素”成语

Question

我觉得我花了很多时间用 Python 编写代码，但没有足够的时间创建 Pythonic 代码。最近我遇到了一个有趣的小问题，我认为可能有一个简单、惯用的解决方案。解释一下原文，我需要收集列表中的每个连续对。例如，给定列表 [1,2,3,4,5,6]，我想计算 [(1,2),(3,4),(5, 6)]。

当时我想出了一个快速解决方案，看起来像是翻译后的 Java。重新审视这个问题，我能做的最好的事情就是

l = [1,2,3,4,5,6]
[(l[2*x],l[2*x+1]) for x in range(len(l)/2)]

在长度不均匀的情况下丢弃最后一个数字的副作用。

我是否缺少一种更惯用的方法，或者这是我能得到的最好的方法？

Answer 1

这会做得更整齐一些：（

>>> data = [1,2,3,4,5,6]
>>> zip(data[0::2], data[1::2])
[(1, 2), (3, 4), (5, 6)]

但如果您不熟悉范围的“跨步”功能，它的可读性可能会降低）。

与您的代码一样，它会丢弃具有奇数个值的最后一个值。

Answer 2

经常被引用的一个是：

zip(*[iter(l)] * 2)

---

我更喜欢这个更具可读性的 iter 解决方案版本：

it = iter(l)
list(zip(it, it))
# [(1, 2), (3, 4), (5, 6)]

Answer 3

使用 range() 的步进功能怎么样：

[(l[n],l[n+1]) for n in range(0,len(l),2)]

Answer 4

我通常将 itertools 文档中的 grouper 配方复制到我的为此的代码。

def grouper(n, iterable, fillvalue=None):
    "grouper(3, 'ABCDEFG', 'x') --> ABC DEF Gxx"
    args = [iter(iterable)] * n
    return izip_longest(fillvalue=fillvalue, *args)

Answer 5

试试这个

def pairs(l, n):
    return zip(*[l[i::n] for i in range(n)])

所以，

pairs([1, 2, 3, 4], 2)
给出

[(1, 2), (3, 4)]

Answer 6

正确的做法可能不是计算列表，而是编写迭代器->迭代器函数。这是更通用的——它适用于每个可迭代对象，如果您想将其“冻结”到列表中，可以使用“list()”函数。

def groupElements(iterable, n):
    # For your case, you can hardcode n=2, but I wanted the general case here.
    # Also, you do not specify what to do if the 
    # length of the list is not divisible by 2
    # I chose here to drop such elements
    source = iter(iterable)
    while True:
        l = []
        for i in range(n):
            l.append(source.next())
        yield tuple(l)

我很惊讶 itertools 模块还没有这样的功能——也许是未来的修订版。在那之前，请随意使用上面的版本:)

Answer 7

toolz 是一个构建良好的库，其中有许多在 itertools 中被忽略的函数式编程细节。 partition 解决了这个问题（可以选择填充奇数长度列表的最后一个条目）

>>> list(toolz.partition(2, [1,2,3,4,5,6]))
[(1, 2), (3, 4), (5, 6)]

Answer 8

如果您不想丢失列表中元素的数量，请尝试以下操作：

>>> l = [1, 2, 3, 4, 5]
>>> [(l[i],  l[i+1] if i+1 < len(l) else None)  for i in range(0, len(l), 2)]
[(1, 2), (3, 4), (5, None)]