[英]Apply a function to every nth element in vector
我在R中有一个向量:
> v <- c(5, 10, 15, 20, 25, 30, 35, 40, 45, 50)
我想将一个函数应用于向量的每个第n个元素,并让它返回新的向量。 例如,假设我想将向量的每个第三个元素乘以2.然后我会得到
> v
[1] 5 10 30 20 25 60 35 40 90 50
我设法使用向量逻辑提取元素:
> v[c(rep(FALSE,2),TRUE)]
[1] 15 30 45
意思是我已经弄清楚如何访问元素,我能够为他们做事,但我不知道如何让他们回到我原来的矢量v
。
我们需要分配
v[c(FALSE, FALSE, TRUE)] <- v[c(FALSE, FALSE, TRUE)]*2
另一种选择是使用seq
。 您可以随意指定n
, seq
将在n
处开始一个序列,由n
开始,直到向量的末尾( length(v)
)。
n <- 3
v[seq(n, length(v), n)] <- v[seq(n, length(v), n)]*2
> v
[1] 5 10 30 20 25 60 35 40 90 50
使用ifelse
解决方案:
ifelse(1:length(v) %% 3 == 0, v * 2, v)
# [1] 5 10 30 20 25 60 35 40 90 50
使用tidyverse:
library(tidyverse)
p <- seq(v) %% 3 == 0
f <- function(x) x*2
map_if(v, p, f) %>% as_vector
使用while循环:
i=0
while (i < (length(v)-1) ){
i=i+3
v[i]=v[i]*2
}
> v
[1] 5 10 30 20 25 60 35 40 90 50
声明:本站的技术帖子网页,遵循CC BY-SA 4.0协议,如果您需要转载,请注明本站网址或者原文地址。任何问题请咨询:yoyou2525@163.com.