Это звучит как случай для заданного понимания. Таким образом, вы генерируете f (e) для тех элементов s, которые соответствуют предикату p (e). Общий синтаксис:
{ f(s) | e in set S & p(e) }
Так, например:
{ e*e | e in set {1,2,3,4,5,6} & e mod 2 = 0 } = {4, 16, 36}
Есть более сложные случаи, когда вы связываете более одного элемента из набора, но этого достаточно, чтобы удовлетворить ваши пример:)