1.3 获取数据结构的子集

1.3.1 问题

你想得到一个由向量、矩阵或数据框里元素组成的子集。

1.3.2 方案

为了基于一些条件准则获得子集，可以使用 subset() 函数或者是方括号索引。 两种方式均在下方举例说明。

  1. v <- c(1, 4, 4, 3, 2, 2, 3)
  2. subset(v, v < 3)
  3. #> [1] 1 2 2
  4. v[v < 3]
  5. #> [1] 1 2 2
  7. # 另一个向量
  8. t <- c("small", "small", "large", "medium")
  10. # 删除 'small' 这一项
  11. subset(t, t != "small")
  12. #> [1] "large"  "medium"
  13. t[t != "small"]
  14. #> [1] "large"  "medium"

这两种方法之间的一个重要区别在于方括号索引可以为元素赋值，而 subset() 不可以。

  1. v[v < 3] <- 9
  2. subset(v, v < 3) <- 9
  3. #> Error in subset(v, v < 3) <- 9: 没有"subset<-"这个函数

数据框：

  1. # 样本数据框
  2. data <- read.table(header = T, text = "
  3.  subject sex size
  4.      1   M    7
  5.      2   F    6
  6.      3   F    9
  7.      4   M   11
  8.  ")
  9. subset(data, subject < 3)
  10. #>   subject sex size
  11. #> 1       1   M    7
  12. #> 2       2   F    6
  13. data[data$subject < 3, ]
  14. #>   subject sex size
  15. #> 1       1   M    7
  16. #> 2       2   F    6
  19. # 特定行和列的子集
  20. subset(data, subject < 3, select = -subject)
  21. #>   sex size
  22. #> 1   M    7
  23. #> 2   F    6
  24. subset(data, subject < 3, select = c(sex, size))
  25. #>   sex size
  26. #> 1   M    7
  27. #> 2   F    6
  28. subset(data, subject < 3, select = sex:size)
  29. #>   sex size
  30. #> 1   M    7
  31. #> 2   F    6
  32. data[data$subject < 3, c("sex", "size")]
  33. #>   sex size
  34. #> 1   M    7
  35. #> 2   F    6
  37. # 逻辑与的两个条件
  38. subset(data, subject < 3 & sex == "M")
  39. #>   subject sex size
  40. #> 1       1   M    7
  41. data[data$subject < 3 & data$sex == "M", ]
  42. #>   subject sex size
  43. #> 1       1   M    7
  46. # 逻辑或的两个条件
  47. subset(data, subject < 3 | sex == "M")
  48. #>   subject sex size
  49. #> 1       1   M    7
  50. #> 2       2   F    6
  51. #> 4       4   M   11
  52. data[data$subject < 3 | data$sex == "M", ]
  53. #>   subject sex size
  54. #> 1       1   M    7
  55. #> 2       2   F    6
  56. #> 4       4   M   11
  59. # 基于转换数据的条件
  60. subset(data, log2(size) > 3)
  61. #>   subject sex size
  62. #> 3       3   F    9
  63. #> 4       4   M   11
  64. data[log2(data$size) > 3, ]
  65. #>   subject sex size
  66. #> 3       3   F    9
  67. #> 4       4   M   11
  69. # 当元素在另一个向量里时的子集
  70. subset(data, subject %in% c(1, 3))
  71. #>   subject sex size
  72. #> 1       1   M    7
  73. #> 3       3   F    9
  74. data[data$subject %in% c(1, 3), ]
  75. #>   subject sex size
  76. #> 1       1   M    7
  77. #> 3       3   F    9