Factors in R: more than an annoyance?

后端 未结 7 1477
梦如初夏
梦如初夏 2020-11-28 19:30

One of the basic data types in R is factors. In my experience factors are basically a pain and I never use them. I always convert to characters. I feel oddly like I\'m missi

相关标签:
7条回答
  • 2020-11-28 20:03

    You should use factors. Yes they can be a pain, but my theory is that 90% of why they're a pain is because in read.table and read.csv, the argument stringsAsFactors = TRUE by default (and most users miss this subtlety). I say they are useful because model fitting packages like lme4 use factors and ordered factors to differentially fit models and determine the type of contrasts to use. And graphing packages also use them to group by. ggplot and most model fitting functions coerce character vectors to factors, so the result is the same. However, you end up with warnings in your code:

    lm(Petal.Length ~ -1 + Species, data=iris)
    
    # Call:
    # lm(formula = Petal.Length ~ -1 + Species, data = iris)
    
    # Coefficients:
    #     Speciessetosa  Speciesversicolor   Speciesvirginica  
    #             1.462              4.260              5.552  
    
    iris.alt <- iris
    iris.alt$Species <- as.character(iris.alt$Species)
    lm(Petal.Length ~ -1 + Species, data=iris.alt)
    
    # Call:
    # lm(formula = Petal.Length ~ -1 + Species, data = iris.alt)
    
    # Coefficients:
    #     Speciessetosa  Speciesversicolor   Speciesvirginica  
    #             1.462              4.260              5.552  
    

    Warning message: In model.matrix.default(mt, mf, contrasts) :

    variable Species converted to a factor

    One tricky thing is the whole drop=TRUE bit. In vectors this works well to remove levels of factors that aren't in the data. For example:

    s <- iris$Species
    s[s == 'setosa', drop=TRUE]
    #  [1] setosa setosa setosa setosa setosa setosa setosa setosa setosa setosa
    # [11] setosa setosa setosa setosa setosa setosa setosa setosa setosa setosa
    # [21] setosa setosa setosa setosa setosa setosa setosa setosa setosa setosa
    # [31] setosa setosa setosa setosa setosa setosa setosa setosa setosa setosa
    # [41] setosa setosa setosa setosa setosa setosa setosa setosa setosa setosa
    # Levels: setosa
    s[s == 'setosa', drop=FALSE]
    #  [1] setosa setosa setosa setosa setosa setosa setosa setosa setosa setosa
    # [11] setosa setosa setosa setosa setosa setosa setosa setosa setosa setosa
    # [21] setosa setosa setosa setosa setosa setosa setosa setosa setosa setosa
    # [31] setosa setosa setosa setosa setosa setosa setosa setosa setosa setosa
    # [41] setosa setosa setosa setosa setosa setosa setosa setosa setosa setosa
    # Levels: setosa versicolor virginica
    

    However, with data.frames, the behavior of [.data.frame() is different: see this email or ?"[.data.frame". Using drop=TRUE on data.frames does not work as you'd imagine:

    x <- subset(iris, Species == 'setosa', drop=TRUE)  # susbetting with [ behaves the same way
    x$Species
    #  [1] setosa setosa setosa setosa setosa setosa setosa setosa setosa setosa
    # [11] setosa setosa setosa setosa setosa setosa setosa setosa setosa setosa
    # [21] setosa setosa setosa setosa setosa setosa setosa setosa setosa setosa
    # [31] setosa setosa setosa setosa setosa setosa setosa setosa setosa setosa
    # [41] setosa setosa setosa setosa setosa setosa setosa setosa setosa setosa
    # Levels: setosa versicolor virginica
    

    Luckily you can drop factors easily with droplevels() to drop unused factor levels for an individual factor or for every factor in a data.frame (since R 2.12):

    x <- subset(iris, Species == 'setosa')
    levels(x$Species)
    # [1] "setosa"     "versicolor" "virginica" 
    x <- droplevels(x)
    levels(x$Species)
    # [1] "setosa"
    

    This is how to keep levels you've selected out from getting in ggplot legends.

    Internally, factors are integers with an attribute level character vector (see attributes(iris$Species) and class(attributes(iris$Species)$levels)), which is clean. If you had to change a level name (and you were using character strings), this would be a much less efficient operation. And I change level names a lot, especially for ggplot legends. If you fake factors with character vectors, there's the risk that you'll change just one element, and accidentally create a separate new level.

    0 讨论(0)
  • 2020-11-28 20:05

    tapply (and aggregate) rely on factors. The information-to-effort ratio of these functions is very high.

    For instance, in a single line of code (the call to tapply below) you can get mean price of diamonds by Cut and Color:

    > data(diamonds, package="ggplot2")
    
    > head(dm)
    
       Carat     Cut    Clarity Price Color
    1  0.23     Ideal     SI2   326     E
    2  0.21   Premium     SI1   326     E
    3  0.23      Good     VS1   327     E
    
    
    > tx = with(diamonds, tapply(X=Price, INDEX=list(Cut=Cut, Color=Color), FUN=mean))
    
    > a = sort(1:diamonds(tx)[2], decreasing=T)  # reverse columns for readability
    
    > tx[,a]
    
             Color
    Cut         J    I    H    G    F    E    D
    Fair      4976 4685 5136 4239 3827 3682 4291
    Good      4574 5079 4276 4123 3496 3424 3405
    Very Good 5104 5256 4535 3873 3779 3215 3470
    Premium   6295 5946 5217 4501 4325 3539 3631
    Ideal     4918 4452 3889 3721 3375 2598 2629
    
    0 讨论(0)
  • 2020-11-28 20:06

    Factors are fantastic when one is doing statistical analysis and actually exploring the data. However, prior to that when one is reading, cleaning, troubleshooting, merging and generally manipulating the data, factors are a total pain. More recently, as in the past few years a lot of the functions have improved to handle the factors better. For instance, rbind plays nicely with them. I still find it a total nuisance to have left over empty levels after a subset function.

    #drop a whole bunch of unused levels from a whole bunch of columns that are factors using gdata
    require(gdata)
    drop.levels(dataframe)
    

    I know that it is straightforward to recode levels of a factor and to rejig the labels and there are also wonderful ways to reorder the levels. My brain just cannot remember them and I have to relearn it every time I use it. Recoding should just be a lot easier than it is.

    R's string functions are quite easy and logical to use. So when manipulating I generally prefer characters over factors.

    0 讨论(0)
  • 2020-11-28 20:06

    Factors are an excellent "unique-cases" badging engine. I've recreated this badly many times, and despite a couple of wrinkles occasionally, they are extremely powerful.

    library(dplyr)
    d <- tibble(x = sample(letters[1:10], 20, replace = TRUE))
    
    ## normalize this table into an indexed value across two tables
    id <- tibble(x_u = sort(unique(d$x))) %>% mutate(x_i = row_number())
    di <- tibble(x_i = as.integer(factor(d$x)))
    
    
    ## reconstruct d$x when needed
    d2 <- inner_join(di, id) %>% transmute(x = x_u)
    identical(d, d2)
    ## [1] TRUE
    

    If there's a better way to do this task I'd love to see it, I don't see this capability of factor discussed.

    0 讨论(0)
  • 2020-11-28 20:07

    A factor is most analogous to an enumerated type in other languages. Its appropriate use is for a variable which can only take on one of prescribed set of values. In these cases, not every possible allowed value may be present in any particular set of data and the "empty" levels accurately reflect that.

    Consider some examples. For some data which was collected all across the United States, the state should be recorded as a factor. In this case, the fact that no cases were collected from a particular state is relevant. There could have been data from that state, but there happened (for whatever reason, which may be a reason of interest) to not be. If hometown was collected, it would not be a factor. There is not a pre-stated set of possible hometowns. If data were collected from three towns rather than nationally, the town would be a factor: there are three choices that were given at the outset and if no relevant cases/data were found in one of those three towns, that is relevant.

    Other aspects of factors, such as providing a way to give an arbitrary sort order to a set of strings, are useful secondary characteristics of factors, but are not the reason for their existence.

    0 讨论(0)
  • 2020-11-28 20:10

    What a snarky title!

    I believe many estimation functions allow you to use factors to easily define dummy variables... but I don't use them for that.

    I use them when I have very large character vectors with few unique observations. This can cut down on memory consumption, especially if the strings in the character vector are longer-ish.

    PS - I'm joking about the title. I saw your tweet. ;-)

    0 讨论(0)
提交回复
热议问题