R 使用 for 和 while 循环创建向量

Question

早上好， 我有以下问题。 我的 Data.frame “数据”具有以下格式：

Type    amount
1       2
2       0
3       3

我想创建一个格式如下的向量：

1
1
3
3
3

这意味着我想转换我的数据。 我创建了一个向量并为我在 R 中的转换编写了以下代码：

向量

for (i in 1:3){
     k <- 1
     while (k <= data[i,2]){
         vector[k] <- data[i,1]
         k <- k+1
 }
 }

问题是，我得到以下结果，但我不知道我错在哪一部分……

3
3
3
0
0

在 R 中解决这个特定问题可能有很多不同的方法，但我很好奇为什么我的解决方案不起作用。我很感谢其他选择，但真的很想知道我的错误是什么。 感谢您的帮助！

Answer 1

试试这个解决方案：

df <- data.frame(type = c(1, 2, 3), amount = c(2, 0, 3))
result <- unlist(mapply(function(x, y) rep.int(x, y), df[, "type"], df[, "amount"]))
result

输出如下：

# [1] 1 1 3 3 3

您的代码确实有问题。正确的代码应如下所示：

df <- data.frame(type = c(1, 2, 3), amount = c(2, 0, 3))
vector <- numeric(5)
k <- 1
for (i in 1:3) {
    j <- 1
    while (j <= df[i, 2]) {
        vector[k] <- df[i, 1]
        k <- k + 1
        j <- j + 1
    }
}
vector
# [1] 1 1 3 3 3

Answer 2

可能是获得此结果的最快和最优雅的方式之前已在@akrun 的评论中发布：

with(data, rep(Type, amount))
[1] 1 1 3 3 3

但是，如果您想使用 for/while 循环执行此操作，则在开始时不知道条目数的情况下使用列表可能会有所帮助。

这是一个对您的代码进行了最少修改的示例：

my_list <- vector("list", 3)
for (i in 1:3) {
  k <- 1
  while (k <= data[i,2]){
    my_list[[i]][k] <- data[i,1]
    k <- k + 1
  }
}
vector <- unlist(my_list)
#> vector
#[1] 1 1 3 3 3

您的代码不起作用的原因本质上是您试图将太多信息放入单个变量k。它不能同时用作输出向量的索引，也不能用作data 第一列中各个条目的计数器；每次while 循环完成时重置为1 的计数器。