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

Question

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

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

Answer 1

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

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