利用subset()函数提取数据

最新推荐文章于 2026-04-15 08:36:51 发布

原创最新推荐文章于 2026-04-15 08:36:51 发布 · 4.7k 阅读

7 ·

本内容遵循CC 4.0 BY-SA版权协议

标签

#python #开发语言 #r语言

博客介绍了从矩阵中提取必要信息的方法，可利用subset()函数轻松实现这一操作，涉及Python、R语言等开发语言。

从矩阵中提取某些必要的信息时,可以利用subset()函数轻松实现

subset(data,sunbset,select)#data=要处理的数据,要显示的字符,选择显示哪些字符
#利用鸢尾花(iris)数据集举个例子
> a <- iris[1:150,]
> head(a)
  Sepal.Length Sepal.Width Petal.Length Petal.Width Species
1          5.1         3.5          1.4         0.2  setosa
2          4.9         3.0          1.4         0.2  setosa
3          4.7         3.2          1.3         0.2  setosa
4          4.6         3.1          1.5         0.2  setosa
5          5.0         3.6          1.4         0.2  setosa
6          5.4         3.9          1.7         0.4  setosa 
#查看一下Species有多少个分组
> unique(a$Species)
[1] setosa     versicolor virginica 
Levels: setosa versicolor virginica
#看看setosa组有多少个样本
> length(which(a$Species=="setosa"))
[1] 50
#利用subset()函数将"setosa"提取出来
> a_setsub <- subset(a,Species == "setosa")
> head(a_setsub)
  Sepal.Length Sepal.Width Petal.Length Petal.Width Species
1          5.1         3.5          1.4         0.2  setosa
2          4.9         3.0          1.4         0.2  setosa
3          4.7         3.2          1.3         0.2  setosa
4          4.6         3.1          1.5         0.2  setosa
5          5.0         3.6          1.4         0.2  setosa
6          5.4         3.9          1.7         0.4  setosa
#看一下是不是都提取出来了
> length(a_setsub$Species)
[1] 50
#现在只想显示最后两列
> a_setsub <- subset(a,Species == "setosa",select = c("Petal.Width","Species"))
> head(a_setsub)
  Petal.Width Species
1         0.2  setosa
2         0.2  setosa
3         0.2  setosa
4         0.2  setosa
5         0.2  setosa
6         0.4  setosa
#本次的例子中,length(),which()的用法也可以注意一下.