Question

使用RStudio，我有这个：

我想添加一个名为AGENT的新变量，它是：

AGENT=c("John", "Maria", "Pamela")

但问题在于，我希望我的每个Agent都能根据ID在初始数据帧中同等地传播。基本上，我想要这个：

GROUP NUM AGENT
A     45  John
A     78  Maria
A     79  Pamela
B     45  John 
B     47  Maria
B     99  Pamela
C     28  John
C     78  Maria
C     54  Pamela

我这里的例子是基本的，因为我拥有的代理人数量与我一样多。但是在我的情况下，我可能每个字母有70个（70 A，70 B和70 C），但仍然只有3个代理。我仍然希望它们尽可能平等地传播......

例如，如果我有6 A，我会：

GROUP NUM AGENT
A     45  John
A     78  Maria
A     79  Pamela
A     48  John
A     97  Maria
A     59  Pamela
...

如果我有7，那么第7个将被随机分配，或者只是列表中的下一个。

有什么想法吗？我一直在折磨自己。提前致谢！：P

Answer 1

如果“或只是列表中的下一个。”适用于群组较大时的任何溢出，您可以利用矢量回收，只需在一个作业中执行：

dat$newvar <- with(dat, ave(1:nrow(dat), GROUP, FUN=function(x) AGENT) )
dat
#  GROUP NUM newvar
#1     A  45   John
#2     A  78  Maria
#3     A  79 Pamela
#4     B  45   John
#5     B  47  Maria
#6     B  99 Pamela
#7     C  28   John
#8     C  78  Maria
#9     C  54 Pamela

如果群组与AGENT

的大小不完整匹配，请忽略您可能收到的任何警告也可以以类似的方式使用

data.table：

library(data.table)
setDT(dat)
dat[, newvar2 := AGENT, by=GROUP]

Answer 2

我想出了一个使用索引做一些复杂的方法。可能有一种更简单的方法。这是代码：

library(dplyr)
AGENT <- c("John", "Maria", "Paul") 

fun <- function(x){
 x %>% mutate(agent=AGENT[((1:nrow(.) - 1) %% 3) + 1])
}

df %>% 
  split(.$GROUP) %>%
  lapply(fun) %>%
  bind_rows()

  GROUP NUM agent
1     A  45  John
2     A  78 Maria
3     A  79  Paul
4     B  45  John
5     B  47 Maria
6     B  99  Paul
7     C  28  John
8     C  78 Maria
9     C  54  Paul

如果数据不是太多或GROUP的长度不是AGENT的倍数，则会按照AGENT的顺序创建变量。

df1 

GROUP NUM
1     A  45
2     A  78
3     B  45
4     C  28

df1 %>% 
  split(.$GROUP) %>%
  lapply(fun) %>%
  bind_rows()

  GROUP NUM agent
1     A  45  John
2     A  78 Maria
3     B  45  John
4     C  28  John

Answer 3

试试这个 -

# Data
df <- data.frame("GROUP" = c("A","A","A","A","B","B","C","C","C"),
             "NUM" = c(45,78,79,45,47,99,28,78,54))
AGENT=c("John", "Maria", "Pamela")

# Assign agents
df$agent <- NA
groups <- levels(factor(df$GROUP))

lapply(groups, function(x)
       {
        df[df$GROUP == x, "agent"] <<- 
        c(rep(AGENT, as.integer(length(df[df$GROUP == x, "NUM"]) / 3)),
          AGENT[0:(length(df[df$GROUP == x, "NUM"]) %% 3)])
       })

如果有超过3个代理，请按脚本（代理）

替换脚本中的3

同样向数据框添加新变量

3 个答案: