Используйте понимание списка с двойным циклом для добавления значений в зависимости от условий - PullRequest
0 голосов
/ 05 апреля 2019

Вот мой первый пример кода. Как я могу сделать это, используя понимание списка?

for c in c_list:
    c_id_list.add(c.id) 

Могу ли я использовать списки для достижения этой цели?

for c in c_list:  
    c.p_list = []
    for p in p_list:
        if p.c_id == c.id:
            c.p_list.append(p)

1 Ответ

2 голосов
/ 05 апреля 2019

1.

c_id_list = [c.id for c in c_list]

2.

for c in c_list:  
    c.p_list = [p for p in p_list if p.c_id == c.id]
...