如何按R中的行中的值选择列

时间:2015-05-06 03:33:24

标签: r subset trigram

我有一个大型数据框,用于标记字符串中出现三元组,其中字符串是行,三元组是列,值表示三角形是否出现在字符串中。

这样的事情:

strs <- c('this', 'that', 'chat', 'chin')
thi <- c(1, 0, 0, 0)
tha <- c(0, 1, 0, 0)
hin <- c(0, 0, 0, 1)
hat <- c(0, 1, 1, 0)
df <- data.frame(strs, thi, tha, hin, hat)
df

#  strs thi tha hin hat
#1 this   1   0   0   0
#2 that   0   1   0   1
#3 chat   0   0   0   1
#4 chin   0   0   1   0

我希望得到给定行或给定字符串的所有列/三元组。

因此,对于第2行,字符串&#39;&#39;,结果将是一个如下所示的数据框:

  str tha hat
1 this  0   0
2 that  1   1
3 chat  0   1
4 chin  0   0

我怎么能这样做?

2 个答案:

答案 0 :(得分:4)

这将为您提供所需的输出df。

givenStr <- "that"
row <- df[df$strs==givenStr,]
df[,c(1,1+which(row[,-1]==1))]

答案 1 :(得分:1)

在一个班轮中:

df[as.logical(df[df$strs=='that',])]

#  strs tha hat
#1 this   0   0
#2 that   1   1
#3 chat   0   1
#4 chin   0   0