Logo Questions Linux Laravel Mysql Ubuntu Git Menu
 

pythonic way to iterate over a set/remnant set

Tags:

python

what is a clean way of doing the following:

def foo(aSet) :
    for a in aSet:
        for remaining in aSet - {a} :
            doSomething(a,remaining)

I am thinking there must be some way of writing that as just one for loop?

like image 798
MK. Avatar asked Feb 16 '26 22:02

MK.


2 Answers

aSet = {1,2,3}
[[i,j] for i in aSet for j in aSet if j != i]
#=> [[1, 2], [1, 3], [2, 1], [2, 3], [3, 1], [3, 2]]

In your case you need a generator

(doSomething(i, j) for i in aSet for j in aSet if j != i)
like image 73
fl00r Avatar answered Feb 18 '26 16:02

fl00r


>>> from itertools import permutations
>>> aSet = {1, 2, 3}
>>> list(permutations(aSet, 2))
[(1, 2), (1, 3), (2, 1), (2, 3), (3, 1), (3, 2)]

>>> for a, remaining in permutations(aSet, 2):
...     print(a, remaining, end=', ')
... 
1 2, 1 3, 2 1, 2 3, 3 1, 3 2,
like image 20
GingerPlusPlus Avatar answered Feb 18 '26 17:02

GingerPlusPlus