10 examples of 'if else in list comprehension' in Python

231
def is_comprehension(x):
232
    """Returns true if this is a comprehension type."""
233
    return isinstance(x, (ObjectComprehension, SetComprehension, ArrayComprehension))

466
def p_if_stmt(p):
467
    'if_stmt : IF test COLON suite'
468
    p[0] = ast.If([(p[2], p[4])], None)

45
def create_comprehension(for_node: ast.For) -> ast.comprehension:
46
    return ast.comprehension(target=for_node.target, iter=for_node.iter, ifs=[])

1042
@utils.check_messages("using-constant-test", "missing-parentheses-for-call-in-test")
1043
def visit_comprehension(self, node):
1044
    if node.ifs:
1045
        for if_test in node.ifs:
1046
            self._check_using_constant_test(node, if_test)

1276
def visit_ComprehensionNode(self, node):
1277
    if node.expr_scope:
1278
        self.env_stack.append(self.env)
1279
        self.env = node.expr_scope
1280
    # Skip append node here
1281
    self._visit(node.loop)
1282
    if node.expr_scope:
1283
        self.env = self.env_stack.pop()
1284
    return node

2845
def p_comp_for2(p):
2846
    ''' comp_for : FOR exprlist IN or_test comp_iter '''
2847
    target = p[2]
2848
    set_context(target, Store, p)
2849
    gens = []
2850
    gens.append(ast.comprehension(target=target, iter=p[4], ifs=[]))
2851
    for item in p[5]:
2852
        if isinstance(item, ast.comprehension):
2853
            gens.append(item)
2854
        else:
2855
            gens[-1].ifs.append(item)
2856
    p[0] = gens

929
@backtrackable
930
@memoize
931
def parse_comprehension_iter(self):
932
    """
933
    rule =
934
        | comprehension_for
935
        | comprehension_where
936
    """
937
    result = self.parse_comprehension_for()
938
    if result is None:
939
        result = self.parse_comprehension_where()
940
941
    return result

984
def p_if_stmt_4(p):
985
    '''if_stmt : IF test COLON suite if_stmt_star ELSE COLON suite'''
986
    #             1    2     3     4            5    6     7     8
987
    last = p[5]
988
    while len(last.orelse) > 0:
989
        last = last.orelse[0]
990
    last.orelse.extend(p[8])
991
    p[0] = ast.If(p[2], p[4], [p[5]], rule=inspect.currentframe().f_code.co_name, **p[1][1])

2844
def p_comp_for(self, p):
2845
    """comp_for : FOR exprlist IN or_test comp_iter_opt"""
2846
    targs, it, p5 = p[2], p[4], p[5]
2847
    if len(targs) == 1:
2848
        targ = targs[0]
2849
    else:
2850
        targ = ensure_has_elts(targs)
2851
    store_ctx(targ)
2852
    comp = ast.comprehension(target=targ, iter=it, ifs=[])
2853
    comps = [comp]
2854
    p0 = {"comps": comps}
2855
    if p5 is not None:
2856
        comps += p5.get("comps", [])
2857
        comp.ifs += p5.get("if", [])
2858
    p[0] = p0

691
def handle_gen_for (self, node):
692
    return self._handle_comp_for(node)