Python(3.4)列表推导 - 如何控制执行流程

时间:2014-11-28 03:11:51

标签: python list python-3.x list-comprehension control-flow

这是一个反映我目前正在处理的代码的示例:

def find_match():

    num=[[2, 3, 5, 7, 8, 10, 12], [12, 1, 3, 5, 6, 8, 10], [11, 12, 2, 4, 5, 7, 9]]
    name= ['Type One','Type Two','Type Three']

    match = [[[name[ri], name[ci][5:], list(sorted(set (rv) & set(cv)))]
    for ci,cv in enumerate (num) if rv != cv] for ri,rv in enumerate(num)]

    return match
>>> match=find_match()
[[[3, 5, 8, 10, 12], [2, 5, 7, 12]], [[3, 5, 8, 10, 12], [5, 12]], [[2, 5, 7, 12], [5, 12]]]

再一次,这只是一个例子,所以我知道该函数绝不是一般化的,实际函数采用一个嵌套列表作为arg,而arg又通过另一个函数创建。

这可以找到每个子列表与所有其他子列表相比的匹配

我现在要做的是在新的匹配对列表中为每个匹配添加一个名称,这是我想要的输出

[['Type One', ['Two', [3, 5, 8, 10, 12]], ['Three', [2, 5, 7, 12]]], ['Type Two', ['One', [3, 5,   
8, 10, 12]], ['Three', [5, 12]]], ['Type Three', ['One', [2, 5, 7, 12]], ['Two', [5, 12]]]]

你看,生成字符串名称,例如'One',并打印模式,将发生在嵌套for循环中的同一行执行中,其中一组匹配的标题,例如'Type One '将在外循环中执行。

这是我使用嵌套for循环所需的输出:

matches=[]
match=[]

for ri,rv in enumerate(num):
    match.append(name[ri])
    for ci,cv in enumerate(num):
        if rv!= cv:
            match.append([name[ci][5:],sorted((list(set(cv) & set (rv))))])

     matches.append(match)
     match=[]

return matches

这允许我通过索引访问列表的一部分,例如:

>>> match=find_match()
>>> match[0]
['Type One', ['Two', [3, 5, 8, 10, 12]], 'Type One', ['Three', [2, 5, 7, 12]]]
>>> match[0][0]
'Type One'
>>> match[0][1]
['Two', [3, 5, 8, 10, 12]]
>>> match[0][2]
'Type One'

使用LC,它只返回一个值列表,但是它的行为类似于嵌套的for循环,其中所有语句都在最终循环中执行:

for ri,rv in enumerate(num):
    for ci,cv in enumerate(num):
        if rv!= cv:
            match.append(name[ri])
            match.append([name[ci][5:],sorted((list(set(cv) & set (rv))))]) 

这与此LC相同:

match = [[[name[ri], name[ci][5:], list(sorted(set (rv) & set(cv)))]
for ci,cv in enumerate (num) if rv != cv] for ri,rv in enumerate(num)]

这两种产生:

[[['Type One', 'Two', [3, 5, 8, 10, 12]], ['Type One', 'Three', [2, 5, 7, 12]]], [['Type Two',     
'One', [3, 5, 8, 10, 12]], ['Type Two', 'Three', [5, 12]]], [['Type Three', 'One', [2, 5, 7,    
12]], ['Type Three', 'Two', [5, 12]]]]

所以,我想知道是否有办法控制LC内的执行流程,以便语句在各自的范围内,而不是一次执行所有语句。如果有令人难以置信的东西,它将使嵌套循环的创建变得如此简单,代码更少,格式化它几乎毫不费力。

感谢您花时间阅读我的问题,任何帮助都会非常非常感谢

1 个答案:

答案 0 :(得分:0)

num = [[2, 3, 5, 7, 8, 10, 12], [12, 1, 3, 5, 6, 8, 10], [11, 12, 2, 4, 5, 7, 9]]
name = ['Type One','Type Two','Type Three']

[[name[ci]] 
 + [[name[ri][5:]] 
    + [sorted(set(cv) & set (rv))]   
    for ri,rv in enumerate(num) if rv!=cv] 
 for ci,cv in enumerate(num)]

产量

[['Type One', ['Two', [3, 5, 8, 10, 12]], ['Three', [2, 5, 7, 12]]],
 ['Type Two', ['One', [3, 5, 8, 10, 12]], ['Three', [5, 12]]],
 ['Type Three', ['One', [2, 5, 7, 12]], ['Two', [5, 12]]]]