CODEWITHSUNDEEP
rvectorattributessubsetpersistent
Brandon Bertelsen
Brandon Bertelsen

Reputation: 44658

subset() drops attributes on vectors; how to maintain/persist them?

Let's say I have a vector where I've set a few attributes:

vec <- sample(50:100,1000, replace=TRUE)
attr(vec, "someattr") <- "Hello World"

When I subset the vector, the attributes are dropped. For example:

tmp.vec <- vec[which(vec > 80)]
attributes(tmp.vec) # Now NULL

Is there a way to, subset and persist attributes without having to save them to another temporary object?

Bonus: Where would one find documentation of this behaviour?

Upvotes: 16

Views: 2199

Answers (2)

user63230
user63230

Reputation: 4698

Thanks to a similar answer to my question @G. Grothendieck, you can use collapse::fsubset see here.

library(collapse)
#tmp_vec <- fsubset(vec, vec > 80)
tmp_vec <- sbt(vec, vec > 80)     # Shortcut for fsubset
attributes(tmp_vec) 
# $someattr
# [1] "Hello World"

Upvotes: 0

Gavin Simpson
Gavin Simpson

Reputation: 174863

I would write a method for [ or subset() (depending on how you are subsetting) and arrange for that to preserve the attributes. That would need a "class" attribute also adding to your vector so that dispatch occurs.

vec <- 1:10
attr(vec, "someattr") <- "Hello World"
class(vec) <- "foo"

At this point, subsetting removes attributes:

> vec[1:5]
[1] 1 2 3 4 5

If we add a method [.foo we can preserve the attributes:

`[.foo` <- function(x, i, ...) {
    attrs <- attributes(x)
    out <- unclass(x)
    out <- out[i]
    attributes(out) <- attrs
    out
}

Now the desired behaviour is preserved

> vec[1:5]
[1] 1 2 3 4 5
attr(,"someattr")
[1] "Hello World"
attr(,"class")
[1] "foo"

And the answer to the bonus question:

From ?"[" in the details section:

Subsetting (except by an empty index) will drop all attributes except names, dim and dimnames.

Upvotes: 18

Related Questions