Logo Questions Linux Laravel Mysql Ubuntu Git Menu
 

How can I get a list of all possible partitions of a vector in R?

Suppose I have an R vector of unique elements such as x <- c(1,2,3,4,5).

Is there a function to give me a list of all possible partitions of this vector x? I guess each partition would be a list of vectors, where each element in x belongs to one of the vectors. I want all possible partitions into any number of sets of any size.

(I think the number of such partitions is something like 2^n * n!, where n is the number of unique elements. I will probably not be using this function on vectors with more than 4 unique elements.)

like image 582
Ryan C. Thompson Avatar asked May 18 '12 23:05

Ryan C. Thompson


1 Answers

Here's a solution that will get you a complete list of partitions, each one of which is represented as a list of vectors. Since a list of lists is pretty ugly when printed to the screen, I've also shown you how to get a more nicely printed object.

library(partitions)

x <- c(2,4,6)       # Substitute the vector for which you want partitions 
parts <- listParts(length(x))
out <- rapply(parts, function(ii) x[ii], how="replace")

# This step is for cosmetic purposes only. It allows you to take advantage of
# the `print.equivalence` print method when printing the object to a console 
for(i in seq_along(out)) class(out[[i]]) <- c("list", "equivalence")
out
[[1]]
[1] (2,4,6)

[[2]]
[1] (2,6)(4)

[[3]]
[1] (2,4)(6)

[[4]]
[1] (4,6)(2)

[[5]]
[1] (2)(4)(6)

See also setparts() in the same package for a more compact way to represent the same set of partitions.

like image 98
Josh O'Brien Avatar answered Sep 26 '22 03:09

Josh O'Brien