This is a fairly long question. It's about a problem that's easy to
specify in terms of sets, but that I found hard to solve in R by using
them, because of the strange design of R data structures. In explaining
it, I'm going to touch on the reshape2 library, dcast, sets, and the
non-orthogonality of R.
My problem stems from some drug-trial data that I've been analysing for
the Oxford Pain Research Unit. Here's an example. Imagine a data frame
representing patients in a trial of pain-relief drugs. The trial lasts for
ten days. Each patient's pain is measured once a day, and the values are
recorded in a data frame, one row per patient per day. Like this:
ID Day Pain
1 1 10
1 2 9
1 4 7
1 7 2
2 2 8
2 3 7
3 1 10
3 3 6
3 4 6
3 8 2
Unfortunately, many patients have measurements missing. Thus, in the
example above, patient 1 was only observed on days 1, 2, 4, and 7, rather
than on the full ten days. But a patient's measurements are only useful to
us if that patient has a certain minimum set of days, so I need to check
for patients who lack those days. Let's assume that these days are numbers
1, 4, and 9.
Such a question is trivial to state in terms of sets. Let D(i) denote the
set of days on which patient i was measured: then I want to find out which
patients p, or how many patients p, have a D(p) that contains the set
{1,4,9}.
The obvious way to solve this is to write a function that tells me whether
one set is a superset of another. Then flatten my data frame so that it
looks like this:
ID Days
1 {1,2,4,7}
2 {2,3}
3 {1,3,4,8}
And finally, filter it by some R translation of
flattened[ includes( flattened$Days, {1,4,9} ), ]
I started with the built-in functions that operate on sets represented as
vectors. These are described in
https://stat.ethz.ch/R-manual/R-devel/library/base/html/sets.html ,
"Set Operations". For example:
> union( c(1,2,3), c(2,4,6) )
[1] 1 2 3 4 6
> intersect( c(1,2,3), c(2,4,6) )
[1] 2
So I first wrote a set-inclusion function:
# True if vector a is a superset of vector b.
#
includes <- function( a, b )
{
return( setequal( union( a, b ), a ) )
}
Here are some sample calls:
> includes( c(1), c() )
[1] TRUE
> includes( c(1), c(1) )
[1] TRUE
> includes( c(1), c(1,2) )
[1] FALSE
> includes( c(2,1), c(1,2) )
[1] TRUE
> includes( c(2,1,3), c(1,2) )
[1] TRUE
> includes( c(2,1,3), c(4,1,2) )
[1] FALSE
I then made myself a variable holding my sample data frame:
df <- data.frame( ID = c( 1, 1, 1, 1, 2, 2, 3, 3, 3, 3 )
, Day = c( 1, 2, 4, 7, 2, 3, 1, 3, 4, 8 )
)
And I tried flattening it, using dcast and an aggregator function as
described in (amongst many other places)
http://seananderson.ca/2013/10/19/reshape.html , "An Introduction to
reshape2" by Sean C. Anderson.
The idea behind this is that (for my data) dcast will call the aggregator
function once per patient ID, passing it all the Day values for the
patient. The aggregator must combine them in some way, and dcast puts its
results into a new column. For example, here's an aggregator that merely
sums its arguments:
aggregator_making_sum <- function( ... )
{
return( sum( ... ) )
}
If I call it, I get this:
> dcast( df, ID~. , fun.aggregate=aggregator_making_sum )
Using Day as value column: use value.var to override.
ID .
1 1 14
2 2 5
3 3 16
And here's an aggregator that converts the argument list to a string:
aggregator_making_string <- function( ... )
{
return( toString( ... ) )
}
Calling it gives this:
> dcast( df, ID~. , fun.aggregate=aggregator_making_string )
Using Day as value column: use value.var to override.
ID .
1 1 1, 2, 4, 7
2 2 2, 3
3 3 1, 3, 4, 8
In both of these, the three dots denote all arguments to the aggregator,
as explained in Burns Statistics's
http://www.burns-stat.com/the-three-dots-construct-in-r/ . My first
aggregator sums them; my second converts them to a string. Both uses of
dcast generate a data frame with a column named "." , which contains the
aggregates. In the second data frame, that may not be so clear: the first
column of numbers is row numbers; the second column of numbers are the
IDs; and the remaining columns form the strings, belonging to "." .
But what I want is neither a sum nor a string but a set. Specifically, a
set that's compatible with the R set operations I called in my 'includes'
function. Since these sets are vectors, my aggregator should just pack its
arguments into a vector:
aggregator_making_set <- function( ... )
{
return( c( ... ) )
}
But when I tried it, I got an error:
> dcast( df, ID~. , fun.aggregate=aggregator_making_set )
Using Day as value column: use value.var to override.
Error in vapply(indices, fun, .default) : values must be length 0,
but FUN(X[[1]]) result is length 4
It's not an informative error message, because it expects me to know how
dcast is coded. And I'm surprised that values need to be length 0: length
1 would seem more appropriate. But perhaps it's trying to say that 'c'
doesn't work on three-dots argument lists. Let's test that hypothesis:
test_c_on_three_dots <- function( ... )
{
return( c( ... ) )
}
> test_c_on_three_dots( 1 )
[1] 1
> test_c_on_three_dots( 1, 2 )
[1] 1 2
> test_c_on_three_dots( 1, 2, 3 )
[1] 1 2 3
So 'c' does indeed work on three-dots argument lists. The error must have
been caused by something else. Let's try making a set and putting it into
a data frame directly:
> df <- data.frame( col1=c(1,2), col2=c(3,4) )
> df
col1 col2
1 1 3
2 2 4
> set <- union( c(5,6), c(6,7) )
> set
[1] 5 6 7
> df[ 1, ]$col1 <- set
Error in `$<-.data.frame`(`*tmp*`, "col1", value = c(5, 6, 7)) :
replacement has 3 rows, data has 1
So that's the problem. Already in 1968, there was a language named Algol68
which had arrays and, in order to make things easy for its programmers,
allowed you to create arrays of every data type the language provided. You
could have arrays of Booleans, arrays of integers, arrays of records,
arrays of discriminated unions, arrays of procedures, arrays of I/O
formats, arrays of pointers, and arrays of arrays. The idea was
"orthogonality" (see for example
http://stackoverflow.com/questions/1527393/what-is-orthogonality ): that
the programmer does not have to think about unexpected interactions
between the concept of array and the concept of the element type, because
there are none. If you have a data type, you can make arrays of that type.
Pop-2 (1970), Snobol4 (1966), and Lisp (1958) were similarly generous. But
R (1993) isn't. It wants to make life hard by forcing me to use different
kinds of container for different kinds of element. And by providing a nice
implementation of sets and then not letting me store them.
So I thought about the kinds of data that I _can_ store in a data frame
and generate by flattening. Strings! So I decided to use my
aggregator_making_string function to make a string representation of the
set of days, and to write a set-inclusion function that compared these
sets against sets represented as vectors:
includes2 <- function( a_as_string, b )
{
a <- as.numeric( unlist( strsplit( a_as_string, split="," ) ) )
return( setequal( union( a, b ), a ) )
}
Here are some example calls:
> includes2( '1,2,3', c(1) )
[1] TRUE
> includes2( '1,2,3', c(1,2) )
[1] TRUE
> includes2( '1,2,3', c(1,2,4) )
[1] FALSE
> includes2( '1,2,3', c(3) )
[1] TRUE
> includes2( '1,2,3', c(0,3) )
[1] FALSE
>
I then tried using it:
df <- data.frame( ID = c( 1, 1, 1, 1, 2, 2, 3, 3, 3, 3 )
, Day = c( 1, 2, 4, 7, 2, 3, 1, 3, 4, 8 )
)
aggregator_making_string <- function( ... )
{
return( toString( ... ) )
}
flattened <- dcast( df, ID~. , fun.aggregate=aggregator_making_string )
# Which patients have a day 1?
flattened[ includes2( flattened$. , c(1) ), ]
Unfortunately, that didn't work. The final statement selected every row of
'flattened'. I eventually realised that I had to vectorise 'includes2':
includes3 <- Vectorize( includes2, "a_as_string" )
And that did work:
> flattened[ includes3( flattened$. , c(1) ), ]
ID .
1 1 1, 2, 4, 7
3 3 1, 3, 4, 8
> flattened[ includes3( flattened$. , c(1,2) ), ]
ID .
1 1 1, 2, 4, 7
> flattened[ includes3( flattened$. , c(1,3) ), ]
ID .
3 3 1, 3, 4, 8
> flattened[ includes3( flattened$. , c(2) ), ]
ID .
1 1 1, 2, 4, 7
2 2 2, 3
The moral of this email tale is that sets are really useful for filtering
data, and dcast ought to be really useful for generating sets, but R
refuses to let me store them in the data frame that dcast generates. I can
fudge it by representing the sets as strings, but is there a cleaner way
to solve the problem?
Cheers,
Jocelyn Ireson-Paine
07768 534 091
http://www.jocelyns-cartoons.uk
http://www.j-paine.org
______________________________________________
R-help@r-project.org mailing list -- To UNSUBSCRIBE and more, see
https://stat.ethz.ch/mailman/listinfo/r-help
PLEASE do read the posting guide http://www.R-project.org/posting-guide.html
and provide commented, minimal, self-contained, reproducible code.