我有两个向量,base <- 1:10
, 和 power <- 1:10
;我想要的是使用 for
循环以使用功率向量提高基本向量的每个元素的功率。
即 c([1^1,1^2,...],[2^1, 2^2,...],[3^1,3^2,...]...)
我试过嵌套 for
循环,但由于我是 R 的新手,所以我不知道如何让它工作。我的代码是:
for (i in 1:10) {
for (m in 1:10) {
rows <- i^m
}
}
最佳答案
我们可以用 outer
来做到这一点。
b <- 1:4; p <- 1:3
outer(b, p, `^`)
# [,1] [,2] [,3]
# [1,] 1 1 1
# [2,] 2 4 8
# [3,] 3 9 27
# [4,] 4 16 64
如果需要向量,as.vector
outer(b, p, `^`) |> as.vector()
还有一个列表
outer(b, p, `^`) |> as.data.frame() |> as.list()
# $V1
# [1] 1 2 3 4
#
# $V2
# [1] 1 4 9 16
#
# $V3
# [1] 1 8 27 64
循环可能看起来像这样,
b <- 1:4; p <- 1:3
res <- array(dim=c(length(b), length(p)))
for (i in seq_along(b)) {
for (j in seq_along(p)) {
res[i, j] <- b[i]^p[j]
}
}
res
# [,1] [,2] [,3]
# [1,] 1 1 1
# [2,] 2 4 8
# [3,] 3 9 27
# [4,] 4 16 64
但它在 R 中非常效率低下。
不过,我们可以在 Rcpp
中完成。
Rcpp::sourceCpp(code='
#include <Rcpp.h>
using namespace Rcpp;
// [[Rcpp::export]]
NumericMatrix powv(NumericVector b, NumericVector p) {
int blen = b.size();
int plen = p.size();
NumericMatrix m(blen, plen);
for (int i = 0; i < blen; i++) {
for (int j = 0; j < plen; j++) {
m(i, j) = std::pow(b[i], p[j]);
}
}
return m;
}
')
powv(b, p)
# [,1] [,2] [,3]
# [1,] 1 1 1
# [2,] 2 4 8
# [3,] 3 9 27
# [4,] 4 16 64
微基准测试
b <- seq(0, 5, .01); p <- seq(0, 5, .01)
microbenchmark::microbenchmark(
outer=outer(b, p, `^`),
powv=powv(b, p),
Power_rui=Power(b, p),
`for`=forfun(b, p),
check='equal'
)
# Unit: milliseconds
# expr min lq mean median uq max neval
# outer 9.700440 9.922495 10.341630 9.981673 10.081983 13.71488 100
# powv 6.389418 6.418062 6.552114 6.434775 6.478917 8.35611 100
# Power_rui 11.074284 11.131459 11.433523 11.178858 11.267215 13.77799 100
# for 39.902641 40.970678 42.190528 42.313148 42.825829 47.53356 100
关于将一个向量提升为另一个向量的幂,我们在Stack Overflow上找到一个类似的问题: https://stackoverflow.com/questions/73839290/