ifelse 条件 R 中的 sumif

Question

我有一个包含多列的 DT，我需要在 ifelse 中给出一个条件并相应地进行计算。我希望它按段分组进行计数/总和（计数）。这是DT

Segment  Count  Flag
A        23     Y
B        45     N
A        56     N 
B        212    Y

我希望第四列作为基于标志的段总计数的计数，因此输出应该看起来像这样。对于标志 N，它是每段计数的份额。对于标志 Y，如果 No(N) 变为 Yes(Y)，则计算收入百分比，在这种情况下是可以赚取的收入。我很抱歉，因为它很笨拙，但如果您有任何疑问，请在 cmets 中问我。

Segment  Count  Flag   Rev   Value
    A        23     Y  34    ((56/23)*34)/(34+69)
    B        45     N  48    45/(45+212)
    A        56     N  23    56/(56+23)
    B        212    Y  67    ((45/212)*67)/(67+12)
    A        65     Y  69     ...
    B        10     Y  12    ...

感谢任何帮助。谢谢！

Answer 1

我们可以通过data.table 做到这一点。将'data.frame'转换为'data.table'（setDT(DT)），按'Segment'分组，通过'Count'除以'Count'的sum来创建'Value'列，然后我们更新Flag 为 'N' 的'Value'

library(data.table)
setDT(DT)[, Value := Count/sum(Count), Segment
              ][Flag == "N", Value := Count/sum(Count), Segment]


DT
#   Segment Count Flag      Value
#1:       A    23    Y 0.18852459
#2:       B    45    N 1.00000000
#3:       A    56    N 1.00000000
#4:       B   212    Y 0.78810409
#5:       A    43    Y 0.35245902
#6:       B    12    Y 0.04460967

只需检查 OP 的预期输出“值”

> 23/122
#[1] 0.1885246
> 212/269
#[1] 0.7881041
> 43/122
#[1] 0.352459
> 12/269
#[1] 0.04460967

更新3

基于 Op 帖子中的更新 No:3

s1 <-  setDT(DT1)[, .(rn = .I[Flag == "Y"], Value = (Rev[Flag=="Y"] *
    (Count[Flag == "N"]/Count[Flag=="Y"]))/sum(Rev[Flag == "Y"])), Segment]
s2 <-  DT1[, .(rn = .I[Flag == "N"], Value = Count[Flag == "N"]/(Count[Flag == "N"] + 
               Count[Flag=="Y"][1])), Segment]

DT1[, Value := rbind(s1, s2)[order(rn)]$Value]
DT1
#   Segment Count Flag Rev     Value
#1:       A    23    Y  34 0.8037146
#2:       B    45    N  48 0.1750973
#3:       A    56    N  23 0.7088608
#4:       B   212    Y  67 0.1800215
#5:       A    65    Y  69 0.5771471
#6:       B    10    Y  12 0.6835443


>((56/23)*34)/(34+69)
#[1] 0.8037146
> 45/(45+212)
#[1] 0.1750973
>  56/(56+23)
#[1] 0.7088608
> ((45/212)*67)/(67+12)
#[1] 0.1800215

数据

DT <- structure(list(Segment = c("A", "B", "A", "B", "A", "B"), Count = c(23L, 
45L, 56L, 212L, 43L, 12L), Flag = c("Y", "N", "N", "Y", "Y", 
"Y")), .Names = c("Segment", "Count", "Flag"), row.names = c(NA, 
-6L), class = "data.frame")

DT1 <- structure(list(Segment = c("A", "B", "A", "B", "A", "B"), Count = c(23L, 
45L, 56L, 212L, 65L, 10L), Flag = c("Y", "N", "N", "Y", "Y", 
"Y"), Rev = c(34L, 48L, 23L, 67L, 69L, 12L)), .Names = c("Segment", 
"Count", "Flag", "Rev"), class = "data.frame", row.names = c(NA, 
-6L))

Answer 2

或者，我们也可以使用dplyr pkg 来实现...

根据@Aramis7d 提供的建议进行更新 - 谢谢！

library(data.table)
df <- fread("Segment  Count  Flag
 A        23     Y
    B        45     N
            A        56     N
            B        212    Y
            A        43     Y
            B        12     Y")

library(dplyr)

df %>% 
      group_by(Segment) %>% 
      mutate(Value = Count/sum(Count)) %>%
      group_by(Segment, Flag) %>%
      mutate(Value = if_else( Flag == "N", Count/sum(Count), Value))