Skip to main content
 首页 » 编程设计

python之使用 lambda 或列表理解来创建带循环的列表

2025年05月04日12落叶无声

对于以下代码,我如何使用 lambda 函数或使用 python 列表理解将其写成一行?

def f(): 
    lst=[] 
    for i in range(1, 101): 
        if i < 50: 
            lst.append('A') 
        else: 
            lst.append('B') 
    return lst 

请您参考如下方法:

您可以在列表理解中使用三元条件:

lst = ['A' if i < 50 else 'B' for i in range(1, 101)]