df为1个data.frame对象,有stratum和psu两列,这里统计stratum列计数
方法1:
1
|
cnt = table( df $stratum)
|
方法2:
1
|
cnt = tapply( df $psu, INDEX= df $stratum, FUN=length)
|
在方法2的基础上,只要改变FUN函数就可以实现分组求和、求均值等功能,如下
分组求均值:
1
2
|
tapply( df $psu, INDEX= df $stratum, FUN=mean)
#(等价于python中的df.groupby('stratum').psu.mean)
|
补充:R语言 | 自定义函数对数据集(data.frame)的列进行条件判断计算
1.使用iris数据集
1
2
3
4
5
6
7
8
|
> iris_10 <- head (iris, n = 10)
## 自定义函数:如果x >= 5.0, z = y *10
> get_With_function <- function (x, y, z){
+ if (x >= 5.0){
+ z <- y * 10
+ }
+ c(zlie = z )
+ }
|
2.保险起见,设定z列为0,可能也不需要
1
|
> iris_10$z <- 0
|
3.运用自定义函数,对data.frame的x行进行判断,对y列进行运算,赋值到z列
4…注意Map的使用
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
|
> iris_10$z <- with(
+ iris_10,
+ Map(
+ get_With_function,
+ iris_10$Sepal.Length,
+ iris_10$Sepal.Width,
+ z
+ )
+ )
> iris_10
Sepal.Length Sepal.Width Petal.Length Petal.Width
1 5.1 3.5 1.4 0.2
2 4.9 3.0 1.4 0.2
3 4.7 3.2 1.3 0.2
4 4.6 3.1 1.5 0.2
5 5.0 3.6 1.4 0.2
6 5.4 3.9 1.7 0.4
7 4.6 3.4 1.4 0.3
8 5.0 3.4 1.5 0.2
9 4.4 2.9 1.4 0.2
10 4.9 3.1 1.5 0.1
Species z
1 setosa 35
2 setosa 0
3 setosa 0
4 setosa 0
5 setosa 36
6 setosa 39
7 setosa 0
8 setosa 34
9 setosa 0
10 setosa 0
|
以上为个人经验,希望能给大家一个参考,也希望大家多多支持服务器之家。如有错误或未考虑完全的地方,望不吝赐教。
原文链接:https://blog.csdn.net/Asher117/article/details/111442012