繁体   English   中英

将函数应用于向量中的每个第n个元素

[英]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 您可以随意指定nseq将在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.

 
粤ICP备18138465号  © 2020-2024 STACKOOM.COM