如何根据R中的索引填写空数据框？

Question

我正在尝试创建一个空数据框，然后根据此for循环填充它。

我希望通过维度5x10得到一个数据框，其中包含向量A和B中每个数字相乘的结果。

这就是我想要的结束数据框架。

到目前为止，我使用for循环来计算2个向量的乘积，但是我无法将我想要的结果插入到数据框中。

我哪里错了？

我的代码：

a <- c(1:10)
b <- c(1:5)

#Make a dummy dataframe filled with zeros, thats length(a) long and length(b) high     
dummy <- as.data.frame(matrix(0, ncol=5, nrow=10))

heatmap_prep <- function(vector_a,vector_b){
        for (i in 1:length(a)){
            first_number <- a[i]
        for(j in 1:length(b)){
            second_number <- b[j]
            result <- first_number*second_number
            dummy [i,j] <- result
            print(result)
        }
    }
}

谢谢！

Answer 1

函数不会修改函数之外的东西。您应该在函数内部创建dummy，并在函数末尾创建return最终修改版本：

heatmap_prep <- function(vector_a,vector_b){
    dummy <- as.data.frame(matrix(0, ncol=length(vector_b), nrow=length(vector_a)))
    for (i in 1:length(a)){
        first_number <- a[i]
        for(j in 1:length(b)){
            second_number <- b[j]
            result <- first_number*second_number
            dummy [i,j] <- result
            print(result)
        }
    }
    return(dummy)
}
heatmap_prep(a, b)
#    V1 V2 V3 V4 V5
# 1   1  2  3  4  5
# 2   2  4  6  8 10
# 3   3  6  9 12 15
# 4   4  8 12 16 20
# 5   5 10 15 20 25
# 6   6 12 18 24 30
# 7   7 14 21 28 35
# 8   8 16 24 32 40
# 9   9 18 27 36 45
# 10 10 20 30 40 50

但是，在这种情况下，内置outer函数更加简洁。输出为matrix，但您可以轻松将其强制转换为data.frame。

outer(a, b)
#       [,1] [,2] [,3] [,4] [,5]
#  [1,]    1    2    3    4    5
#  [2,]    2    4    6    8   10
#  [3,]    3    6    9   12   15
#  [4,]    4    8   12   16   20
#  [5,]    5   10   15   20   25
#  [6,]    6   12   18   24   30
#  [7,]    7   14   21   28   35
#  [8,]    8   16   24   32   40
#  [9,]    9   18   27   36   45
# [10,]   10   20   30   40   50

您还可以将此问题视为矩阵乘法。这将得到相同的结果。

 a %*% t(b)