Python中更容易控制流量

Question

在下面的代码中,我试图检查"shape_list"中是否已存在变量"new_shape".如果它不存在,我想添加它; 如果确实存在,我只想留下它.到目前为止,我只使用标志实现了这一点.我确信有一种方法可以在没有旗帜的情况下更有效地完成同样的事情.有什么建议？感谢您的帮助!

    flag = 0
    for shape in shape_list:
        if new_shape == shape:
            flag = 1
            break
    if flag == 0:
        shape_list.append(new_shape)

Answer 1

您可以使用

if new_shape not in shape_list:
    shape_list.append(new_shape)