sub list like subsets

i like this post (click again to cancel)
0
i dont like this post (click again to cancel)

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

asked Sep 17 '10

sriram gravatar image sriram
11 2 2 6

updated Sep 17 '10

i like this answer (click again to cancel)
0
i dont like this answer (click again to cancel)

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:

sage: L = [ [1,2,3,4],[3,5,6,7],[7,8,9,10] ]  # note that this is a list, while you did a tuple
sage: from itertools import product
sage: list(product(L, L))
[([1, 2, 3, 4], [1, 2, 3, 4]), ([1, 2, 3, 4], [3, 5, 6, 7]), ...

(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

list(product(repeat(L, n)))

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:

sage: [ [a,b] for a in L for b in L if a != b ]
[[[1, 2, 3, 4], [3, 5, 6, 7]], [[1, 2, 3, 4], [7, 8, 9, 10]], [[3, 5, 6, 7], [1, 2, 3, 4]], [[3, 5, 6, 7], [7, 8, 9, 10]], [[7, 8, 9, 10], [1, 2, 3, 4]], [[7, 8, 9, 10], [3, 5, 6, 7]]]

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.

Cheers, Johan

link

posted Sep 17 '10

jsrn gravatar image jsrn
114 1 8

updated Sep 17 '10

find all sublists of set of a specified size say 3,4,5... from a given given list of sets sriram (Sep 17 '10)
You could also try, e.g., 'combinations([[1,2], [3,4,5], [6], [7,8,9]], 3)'. (See 'combinations?' for caveats.) There's also 'combinations_iterator'. Mitesh Patel (Sep 17 '10)

Your answer

Please start posting your answer anonymously - your answer will be saved within the current session and published after you log in or create a new account. Please try to give a substantial answer, for discussions, please use comments and please do remember to vote (after you log in)!
[hide preview]

Question tools

Tags:

Stats:

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.