Question

我目前正在将一些Scala代码移植到Python中，我想知道什么是最类似于Scala partition的pythonic方法？特别是，在Scala代码中我有一种情况，我根据是否从我传入的某个过滤谓词返回true或false来分区项目列表：

val (inGroup,outGroup) = items.partition(filter)

在Python中做这样的事情的最佳方法是什么？

Answer 1

使用过滤器（需要两次迭代）：

>>> items = [1,2,3,4,5]
>>> inGroup = filter(is_even, items) # list(filter(is_even, items)) in Python 3.x
>>> outGroup = filter(lambda n: not is_even(n), items)
>>> inGroup
[2, 4]
>>> outGroup

简单循环：

def partition(item, filter_):
    inGroup, outGroup = [], []
    for n in items:
        if filter_(n):
            inGroup.append(n)
        else:
            outGroup.append(n)
    return inGroup, outGroup

示例：

>>> items = [1,2,3,4,5]
>>> inGroup, outGroup = partition(items, is_even)
>>> inGroup
[2, 4]
>>> outGroup
[1, 3, 5]

Answer 2

Scala的

val (inGroup,outGroup) = items.partition(filter)

Python - 使用列表理解

inGroup = [e for e in items if _filter(e)]
outGroup = [e for e in items if not _filter(e)]

Answer 3

此版本是懒惰的，并且不会将谓词两次应用于同一元素：

def partition(it, pred):
    buf = [[], []]
    it = iter(it)

    def get(t):
        while True:
            while buf[t]:
                yield buf[t].pop(0)
            x = next(it)
            if t == bool(pred(x)):
                yield x
            else:
                buf[not t].append(x)

    return get(True), get(False)

示例：

even = lambda x: x % 2 == 0

e, o = partition([1,1,1,2,2,2,1,2], even)
print list(e)
print list(o)

Answer 4

Scala具有丰富的列表处理API，而Python就是这样。

您应该阅读文档itertools。你可能会发现分区的消息。

from itertools import ifilterfalse, ifilter, islice, tee, count

def partition(pred, iterable):
    '''
    >>> is_even = lambda i: i % 2 == 0
    >>> even, no_even = partition(is_even, xrange(11))
    >>> list(even)
    [0, 2, 4, 6, 8, 10]
    >>> list(no_even)
    [1, 3, 5, 7, 9]

    # Lazy evaluation
    >>> infi_list = count(0)
    >>> ingroup, outgroup = partition(is_even, infi_list)
    >>> list(islice(ingroup, 5))
    [0, 2, 4, 6, 8]
    >>> list(islice(outgroup, 5))
    [1, 3, 5, 7, 9]
    '''
    t1, t2 = tee(iterable)
    return ifilter(pred, t1), ifilterfalse(pred, t2)

python相当于scala分区

4 个答案: