如何在 Python3 上制作三个列表中的两个列表

Question

目标是：将 3 个相同长度的列表变成 2 个列表，长度相差不超过 1，来自不同源列表的元素必须交替，并且不能违反它们的顺序。例如：如果“A”在源列表中排在“B”之前，那么“B”在最终列表中不能排在“A”之前。 所以，我决定写一个函数：

def list_3_2(list1,list2,list3):
    #split all lists in one list
    a=sum((list(i) for i in zip(list1,list2,list3)),[])
    #I want to separate list "a" to two new lists: l1 and l2 
    l1=[]
    l2=[]
    #//////
    return(l1,l2)

list_3_2(['1','2','3'],['4','5','6'],['7','8','9'])

然后我面临分离的问题。我发现了一些类似的问题，但它们的列表结构不同，例如main_list=['3 5', '1 2', '1 7']。统一后我得到另一个结果：a=['1', '4', '7', '2', '5', '8', '3', '6', '9']。我怎么能把这个列表分开？

Answer 1

嘿，试试吧，希望就是你想要的

def list_3_2(list1,list2,list3):
        #split all lists in one list
        #I want to separate list "a" to two new lists: l1 and l2 
        a=[]
        [a.extend(j) for j in (list(list(i) for i in zip(list1,list2,list3)))]
        #//////
        div=int(len(a)/2)
        l1=a[0:div]
        l2=a[div:]
        return  l1,l2
    
print(list_3_2(['1','2','3'],['4','5','6'],['7','8','9']))

Answer 2

好吧，根据您分享的最终确定信息，我相信您非常接近解决方案，您只需要拆分最终数组即可。您对此有何看法？

def list_3_2(list1,list2,list3):
    a=sum((list(i) for i in zip(list1,list2,list3)),[])
    return a[:len(a)//2], a[len(a)//2:]
    
print(list_3_2(['1','2','3'],['4','5','6'],['7','8','9']))

基本上，将列表分成两部分，其中一个最多1 元素大于另一个，对应于在向下舍入的中点的索引处拆分任意列表。如果原始数组（输入链）具有奇数长度，则这些部分将具有不同的长度。

数组的中点索引为len(array)//2——len(array)是全长，//2 是“除以四舍五入”的运算。

最后，为了使用这个中点索引分割数组，我们使用了 python 中的splicing 机制。语法如下：

a[:m] = a 的所有元素，从索引 0（= 开头）到索引 m-1。

a[m:] = 从索引m 到索引len(a) - 1 的a 的所有元素（= 结尾）。

Answer 3

也许我误解了这个问题。

from itertools import chain


def list_3_2(l1, l2, l3):
    x = list(chain.from_iterable(zip(l1, l2, l3)))
    m = len(x) // 2
    return (x[:m], x[m:])

压缩列表，将生成的元组连接成一个列表，然后将其分成两半。