使用列表理解和双循环可根据条件附加值

Question

这是我的第一个代码示例。如何使用列表理解来完成此操作？

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)

Answer 1

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]