如何以任意顺序以python方式迭代两个列表

Question

我想迭代两个列表，以便我可以从一个列表中获取任意数量的值并保持我在另一个列表中的位置。

我使用索引来存储每个列表中的当前位置，然后使用 while 循环遍历它们，但这绝对不是 Python 的。

def alternate_iterate(a,b,cond=lambda x, y : x > y):
    pos_a = 0
    pos_b = 0
    retval = []

    while(True):

        if(pos_a == len(a) and pos_b == len(b)):
            break

        if(pos_a < len(a) and cond(a[pos_a],b[pos_b])):
            retval += [a[pos_a]]
            pos_a += 1
        elif(pos_b < len(b)):
            retval += [b[pos_b]]
            pos_b += 1

    return retval

#example usage
print(alternate_iterate(['abc','abcd','ab','abc','ab'],
                        ['xy','xyz','x','xyz'],
                        cond=lambda x,y: len(x) > len(y))

这应该打印['abc','abdc','xy','xyz','ab','abc','ab','x','xyz']，您没有完美的 1:1 交替顺序。元素的顺序和元素的类型应该只取决于 cond 的定义。

Answer 1

更 Pythonic 的方式通常是根本不使用索引，最好不要使用异常作为控制“预期”程序逻辑的手段。您还应该避免不必要的括号。

以下是使用迭代器的方法：

def merge(a, b, cond=lambda x, y : x < y):
    Done           = []
    iterA, iterB   = iter(a), iter(b)
    valueA, valueB = next(iterA, Done), next(iterB, Done)
    result         = []
    while not(valueB is Done and valueA is Done):
        if valueB is Done or valueA is not Done and cond(valueA, valueB):
            result.append(valueA)
            valueA = next(iterA, Done)
        else:
            result.append(valueB)
            valueB = next(iterB, Done)
    return result

这还有一个额外的好处，就是让函数可以将任何可迭代的数据作为参数有效地工作。

例如：

print(merge(range(5, 10), range(7, 15)))

# [5, 6, 7, 7, 8, 8, 9, 9, 10, 11, 12, 13, 14]

它还可以轻松地为惰性遍历创建函数的迭代器版本：

def iMerge(a, b, cond=lambda x, y : x < y):
    Done           = []
    iterA, iterB   = iter(a), iter(b)
    valueA, valueB = next(iterA, Done), next(iterB, Done)
    while not(valueB is Done and valueA is Done):
        if valueB is Done or valueA is not Done and cond(valueA, valueB):
            yield valueA
            valueA = next(iterA ,Done)
        else:
            yield valueB
            valueB = next(iterB, Done)

EDIT 将 None 更改为 Done 以使函数支持 None 作为输入列表中的合法值。

Answer 2

欢迎来到 Stackoverflow。总而言之，您似乎希望根据某个谓词的值从一个列表或另一个列表中获取一个值。您现有的逻辑似乎没有考虑到其中一个列表被用尽的可能性，此时我假设您希望从另一个列表中复制任何剩余的值。

您可以在列表上构建一个迭代器，而不是使用索引值来选择连续的列表元素，并使用next 函数来获取下一个值。

在这种情况下，您的逻辑最终会看起来像这样：

def alternate_iterate(a_lst, b_lst, cond=lambda x, y: x > y):
    a_iter = iter(a_lst)
    b_iter = iter(b_lst)
    a = next(a_iter)
    b = next(b_iter)
    ret = []
    while True:
        if cond(a, b):
            ret.append(a)
            try:
                a = next(a_iter)
            except StopIteration:
                ret.append(b)
                for x in b_iter:
                    ret.append(x)
                return ret
        else:
            ret.append(b)
            try:
                b = next(b_iter)
            except StopIteration:
                ret.append(a)
                for x in a_iter:
                    ret.append(x)
                return ret


print(alternate_iterate(['abc','abcd','ab','abc','ab'],
                        ['xy','xyz','x','xyz'],
                        cond=lambda x,y: len(x) > len(y)))

我得到的结果是

['abc', 'abcd', 'xy', 'xyz', 'ab', 'abc', 'ab', 'x', 'xyz']

这似乎是您所期望的。

与此类示例中的常见情况一样，您编写更多逻辑来处理罕见的极端情况（在这种情况下，一个列表或另一个列表已经用尽），而不是处理事情正常进行的“快乐路径” .

Answer 3

这个版本只使用迭代器来实现懒惰的功能（这是 Pythonic）：

a = ['abc','abcd','ab','abc','ab']
b = ['xy','xyz','x','xyz']

cond=lambda x,y: len(x) > len(y)

def alternate_iterate(a, b, cond):
    a, b = iter(a), iter(b)

    def _return_rest():
        def _f(val, it):
            yield val
            yield from it
        return _f

    v1, v2 = next(a, _return_rest), next(b, _return_rest)

    while True:
        if v1 is _return_rest:
            yield from v1()(v2, b)
            break

        if v2 is _return_rest:
            yield from v2()(v1, a)
            break

        if cond(v1, v2):
            yield v1
            v1 =  next(a, _return_rest)
        else:
            yield v2
            v2 = next(b, _return_rest)

print(list(alternate_iterate(a, b, cond)))

打印：

['abc', 'abcd', 'xy', 'xyz', 'ab', 'abc', 'ab', 'x', 'xyz']

Answer 4

将您的列表放入生成器中，然后您可以在每个列表上调用next 以获得下一个值。这个答案并不意味着是一个完整的解决方案，只是为了展示生成器如何使用非常简单的 Pythonic 代码以任何顺序生成值：

agen = iter(a)
bgen = iter(b)
print next(agen) # 'a'
print next(bgen) # 1
print next(bgen) # 2
print next(agen) # 'b'

等等。