R循环将输出分配给新向量

时间:2018-09-04 18:48:24

标签: r for-loop assign

我正在R中尝试使用for循环生成几个不同的向量。

首先,我创建了一个名为df的可复制的小型示例数据框。

cluster.assignment <- c("1 Unknown", "1 Unknown", "2 Neuron","3 
PBMC","4 Basket")
Value1 <- c("a","b","c","d","e")
Value2 <- c("191","234","178","929","123")
df <- data.frame(cluster.assignment,Value1,Value2)

df

  cluster.assignment Value1 Value2
1          1 Unknown      a    191
2          1 Unknown      b    234
3           2 Neuron      c    178
4             3 PBMC      d    929
5           4 Basket      e    123 . 

接下来,我创建一个名为clusters的变量,其中包含指向我感兴趣的数据集的键。

clusters <- c("1 ","4 ")

这是我尝试使用for循环在df中提取感兴趣的数据的行名。

for (COI in clusters) { 
  name2 <- c(gsub(" ","", paste("Cluster", COI, sep = "_")))
  assign(Cluster_1, name2, envir = parent.frame())
  name2 <- grep(COI, df$cluster.assignment)
}

所需的输出是两个向量,分别为Cluster_1Cluster_4

Cluster_1将包含值12

Cluster_4将包含值5

我似乎无法弄清楚如何将COI变量的名称指定为输出向量的名称。

2 个答案:

答案 0 :(得分:1)

我建议您不要使用assign。相反,我将创建一个命名列表。 See this answer for a long discussion of why lists are better than sequentially named variables。如果您决定随时将列表转换为全局环境中的对象,则可以使用list2env,但这样做可能只会做更多的工作。

## subset the data to the parts we care about, use `split` to separate it
## into a list
subdf = df[grepl(paste(clusters, collapse = "|"), df$cluster.assignment), ]
result = split(subdf, subdf$cluster.assignment, drop = TRUE)
result
# $`1 Unknown`
#   cluster.assignment Value1 Value2
# 1          1 Unknown      a    191
# 2          1 Unknown      b    234
# 
# $`4 Basket`
#   cluster.assignment Value1 Value2
# 5           4 Basket      e    123

## name the list as desired
names(result) = paste("Cluster", trimws(clusters), sep = "_")
result
# $`Cluster_1`
#   cluster.assignment Value1 Value2
# 1          1 Unknown      a    191
# 2          1 Unknown      b    234
# 
# $Cluster_4
#   cluster.assignment Value1 Value2
# 5           4 Basket      e    123

## if only the row names are needed, use lapply
result = lapply(result, row.names)
result
# $`Cluster_1`
# [1] "1" "2"
# 
# $Cluster_4
# [1] "5"

其他一些注意事项-我假设您在clusters中包含空格,以防止"1""12 foo"匹配。您可能会考虑改用正则表达式单词边界"\\b1\\b",因为"1 "仍会匹配,例如"11 foo""21 bar"。更好的是,您可以使用strplit或类似方法仅使用要匹配的数字键来创建新列。

答案 1 :(得分:0)

除非您有自己的理由,否则我认为没有必要为此创建一个for循环,但是以下代码为您提供了所需的内容:

library(data.table)
Cluster_1<-df[df$cluster.assignment %like% "1 ", c("Value1", "Value2")]
Cluster_2<-df[df$cluster.assignment %like% "4 ", c("Value1", "Value2")]
View(Cluster_1);View(Cluster_2)

您可以删除或更改c(“ Value1”,“ Value2”)以获取最终输出中所需的列。