R-包括cumsum函数的起点

作者: 薄情
发布时间: 2024-12-21 09:39:46 (11小时前)
转自：

4 条回复

0#
回复此人
我头上有犄角 | 2019-08-31 10-32

<div class =“post-text”itemprop =“text”> <P> 假设输入是 <code> df </code> 如最后的注释中可重复显示，试试这个。它会把任何东西都归零 <code> a </code> 价值 <code> b </code> 是0。 </p> <pre> <code> transform(df, cum = cumsum((b > 0) * a)) </code> </pre> <P> 赠送： </p> <pre> <code> a b cum 1 1 0 0 2 2 0 0 3 3 0 0 4 4 0 0 5 5 0 0 6 6 1 6 7 7 2 13 8 8 3 21 9 9 4 30 10 10 5 40 </code> </pre> <H2> 注意 </H2> <P> 我们假设此输入以可再现的形式显示： </p> <pre> <code> Lines <- " a b 1 0 2 0 3 0 4 0 5 0 6 1 7 2 8 3 9 4 10 5" df <- read.table(text = Lines, header = TRUE) </code> </pre> <H2> 更新 </H2> <P> <code> a </code> 和 <code> b </code> 被逆转了。有固定的。 </p> </DIV>

编辑
1#
回复此人
筱梨 | 2019-08-31 10-32

<div class =“post-text”itemprop =“text”> <P> 最好创建索引和更新 </p> <pre> <code> i1 <- df1$b > 0 df1$c[i1] <- with(df1, cumsum(a[i1])) </code> </pre> <HR /> <P> 或者在一行中 </p> <pre> <code> df1$c <- with(df1, cumsum(a * (b > 0))) df1$c #[1] 0 0 0 0 0 6 13 21 30 40 </code> </pre> </DIV>

编辑
2#
回复此人
听风～ | 2019-08-31 10-32

<div class =“post-text”itemprop =“text”> <P> 我真的很喜欢其他答案使用的清洁程度 <code> a * (b > 0) </code> 但对于较新的程序员来说，这有时会让人感到困惑。作为此语法的替代，您可以使用矢量化 <code> ifelse </code> 功能。 </p> <pre> <code> df <- data.frame(a=c(1:10), b=c(0,0,0,0,0,1,2,3,4,5)) # One way df$c <- cumsum(ifelse(df$b>0,df$a,0)) # Another way df$d <- with(df,cumsum(ifelse(b>0,a,0))) </code> </pre> </DIV>

编辑

登录后才能参与评论