将因子级别顺序从一列复制到另一列

时间:2017-09-24 13:37:40

标签: r levels

我在data.frame中有两列,应该以相同的顺序排序,但我不知道如何以简单的方式进行。

情况如下:

library(ggplot2)
library(dplyr)
library(magrittr)
set.seed(1)
df1 <- data.frame(rating = sample(c("GOOD","BAD","AVERAGE"),10,T),
                  div = sample(c("A","B","C"),10,T),
                  n = sample(100,10,T))

# I'm adding a label column that I use for plotting purposes
df1 <- df1 %>% group_by(rating) %>% mutate(label = paste0(rating," (",sum(n),")")) %>% ungroup
# # A tibble: 10 x 4
#     rating    div     n         label
#     <fctr> <fctr> <int>         <chr>
#  1     BAD      C    48     BAD (220)
#  2     BAD      B    87     BAD (220)
#  3     BAD      C    44     BAD (220)
#  4    GOOD      B    25     GOOD (77)
#  5 AVERAGE      B     8 AVERAGE (117)
#  6 AVERAGE      C    10 AVERAGE (117)
#  7 AVERAGE      A    32 AVERAGE (117)
#  8    GOOD      B    52     GOOD (77)
#  9 AVERAGE      C    67 AVERAGE (117)
# 10     BAD      C    41     BAD (220)

# rating levels are sorted
df1$rating <- factor(df1$rating,c("BAD","AVERAGE","GOOD"))

ggplot(df1,aes(x=rating,y=n,fill=div)) + geom_col() # plots in the order I want
ggplot(df1,aes(x=label,y=n,fill=div)) + geom_col()  # doesn't because levels aren't sorted

如何设法将因子顺序从一列复制到另一列? 我可以用这种方式工作,但我认为这真的很尴尬:

lvls <- df1 %>% select(rating,label) %>% unique %>% arrange(rating) %>% extract2("label")
df1$label <- factor(df1$label,lvls)
ggplot(df1,aes(x=label,y=n,fill=div)) + geom_col()

2 个答案:

答案 0 :(得分:3)

您可以坚持使用aes(x = label,而不是添加标签列并使用aes(x = rating,并在labels中创建scale_x_discrete

ggplot(df1, aes(x = rating, y = n, fill = div)) +
  geom_col() +
  scale_x_discrete(labels = df1 %>%
                     group_by(rating) %>%
                     summarize(n = sum(n)) %>%
                     mutate(lab = paste0(rating, " (", n, ")")) %>%
                     pull(lab))

enter image description here

答案 1 :(得分:3)

设置rating的级别后,您可以使用forcats按照label的顺序设置rating的级别......

library(forcats)
df1 <- df1 %>% group_by(rating) %>% 
               mutate(label=paste0(rating," (",sum(n),")")) %>% 
               ungroup %>% 
               arrange(rating) %>%              #sort by rating
               mutate(label=fct_inorder(label)) #set levels by order in which they appear

或者您可以使用forcats::fct_reorder做同样的事情......

df1$label <- fct_reorder(df1$label, as.numeric(df1$rating))

然后情节按正确的顺序排列。