how to take lists of size from a list like subsets of size
example L = ([1,2,3,4],[3,5,6,7],[7,8,9,10]) From the above list find sublists of size 2
([1,2,3,4],[3,5,6,7]) ([1,2,3,4],[7,8,9,10]) ([3,5,6,7],[7,8,9,10])
subsets can also be size 3,4,5 as specified
I'm not completely sure I understand your question, but what you're asking for sounds simply (at least almost simply) like cartesian product, which is implemented in Python in itertools.product. In the basic case, you could then just do:
(note the "list" around product, as product returns a generator)
The above can easily be generalised to having 3,4,...,n elements in each sublist (i.e. L^n) by modifying the last line to
However, this would also give those sublists where the same element might be present more than once in each list. If you don't want this, you can write a simple list-comprehension to handle the 2-case:
Manually, the above can also be extended to 3, 4, ... but not easily to any number n given as a parameter. Also, this has the drawback of creating the entire list in memory instead of being a light generator as the product returns. You would have to write a longer piece of code to get a generator with the above behaviour.
Note that this is all Python-stuff and not specific to Sage at all.
Asked: Sep 17 '10
Seen: 207 times
Last updated: Sep 17 '10
powered by ASKBOT version 0.7.22
Copyright Sage, 2010. Some rights reserved under creative commons license.