我是C ++和Rcpp的新手,我想知道如何一次比较两个不同向量的每个元素而没有循环。
我的目标是通过引用其他向量来更改v1
的元素。
当前代码为
v1 = {6,7,8,9,10}
v2 = {2,4,6,8,10}
v3 = {a,b,a,b,c}
v4 = {0,0,0,0,0}
v5 = {a,b,c}
v6 = {1,2,3}
for (i in 1:5){
if (v1[i] > v2[i]){
for (j in 1:3){
if (v5[j] == v3[i]){
v4[i] = v2[i] + v6[j]
if (v1[i] > v4[i]){
v1[i] = v4[i]
}
}
}
}
}
结果值得
v1 = {3,6,7,9,10}
实际上,v1, v2, v3, v4
和v5, v6
在R中是不同的dataframe
。v1
的每个元素都与v2
进行比较。如果i
中的元素v1
大于i
中的v2
元素,则v1
的元素变为{的i
元素的总和{1}}和v1
的元素由相应的v6
和v3
组成。然后将新估计的值v5
与v4[i]
进行比较。
我在v1[i]
和v1~v5
中有很多案件。在这种情况下,使用v5~v6
会花费很长时间。是否可以比较不同的向量而没有循环?或如何估计和引用另一个向量的元素?
答案 0 :(得分:2)
我看不到这里需要使用Rcpp或C ++。以我理解您的要求的方式,您正在尝试操纵两组等长向量。对于“等长集”向量,通常使用data.frame
或其扩展名之一。在这里,我将基数R,data.table
和dplyr
与tibble
一起使用。亲自查看您喜欢哪种语法。一般来说,data.table
对于大型数据集很有可能会更快。
设置数据:
v1 <- c(6,7,8,9,10)
v2 <- c(2,4,6,8,10)
v3 <- c("a","b","a","b","c")
v5 <- c("a","b","c")
v6 <- c(1,2,3)
基本R:
df1 <- data.frame(v1, v2, v3)
df2 <- data.frame(v5, v6)
df1 <- merge(df1, df2, by.x = "v3", by = "v5")
df1$v4 <- df1$v2 + df1$v6
df1$v1 <- ifelse(df1$v1 > df1$v2 & df1$v1 > df1$v4, df1[["v4"]], df1[["v1"]])
df1
#> v3 v1 v2 v6 v4
#> 1 a 3 2 1 3
#> 2 a 7 6 1 7
#> 3 b 6 4 2 6
#> 4 b 9 8 2 10
#> 5 c 10 10 3 13
data.table
:
library(data.table)
dt1 <- data.table(v1, v2, v3, key = "v3")
dt2 <- data.table(v5, v6, key = "v5")
dt1[dt2, v4 := v2 + v6]
dt1[v1 > v2 & v1 > v4, v1 := v4]
dt1
#> v1 v2 v3 v4
#> 1: 3 2 a 3
#> 2: 7 6 a 7
#> 3: 6 4 b 6
#> 4: 9 8 b 10
#> 5: 10 10 c 13
dplyr
:
suppressPackageStartupMessages(library(dplyr))
t1 <- tibble(v1, v2, v3)
t2 <- tibble(v5, v6)
t1 %>%
inner_join(t2, by = c("v3" = "v5")) %>%
mutate(v4 = v2 + v6) %>%
mutate(v1 = case_when(
v1 > v2 & v1 > v4 ~ v4,
TRUE ~ v1
))
#> # A tibble: 5 x 5
#> v1 v2 v3 v6 v4
#> <dbl> <dbl> <chr> <dbl> <dbl>
#> 1 3 2 a 1 3
#> 2 6 4 b 2 6
#> 3 7 6 a 1 7
#> 4 9 8 b 2 10
#> 5 10 10 c 3 13
由reprex package(v0.2.1)于2019-04-19创建
总体思路始终相同:
v4
作为v2
和v6
的总和v1
更新为v4
的值,其中v1 > v2
和v1 > v4
请注意,基数R和data.table
不会保留顺序,因此将输出放入另一列会更有意义。