如何使用R的apply

由于R语言的apply家族函数是用C写的,所以使用apply进行遍历的执行效率远远高于自己编写的循环语句。

函数介绍

apply的用法很简单:

apply(X, MARGIN, FUN, ...)

X: 是一个数组(array),也就是说输入必须都是相同类型的数据,要么都是数值型,要么都是字符型。如果是一个混合数据类型的data.frame,那么就会尝试用as.matrix强制转换数据

MARGIN:表示对行(1)或者是对列(2)应用函数。

FUN: 可是R自带函数,如mean,sum等。也可以是自己编写的函数。

...: 是FUN中的额外参数

简单案例

首先创建一个2行4列的矩阵。

ma <- matrix(c(1:4, 1, 6:8), nrow = 2)
> ma
     [,1] [,2] [,3] [,4]
[1,]    1    3    1    7
[2,]    2    4    6    8

对每一行求平均

> apply(ma,1,mean)
[1] 3 5

相当于

mean(ma[1,]) # 3
mean(ma[2,]) # 5

对每一列求平均

> apply(ma,2,mean)
[1] 1.5 3.5 3.5 7.5

相当于:

mean(ma[,1]) # 1.5
mean(ma[,2]) # 3.5
...

PS:于是我发现MARGIN的1对应ma[x,y]的x,MARGIN的2对应[x,y]的y.

进阶:...的作用

用法说明中,...适用于为函数的提供额外参数。具体的说就是如果一个函数有多个参数,那么...就是负责传入除了一个个参数以外的其他参数。
比如说mean(x, trim = 0, na.rm = FALSE, ...),就有一个如何处理na值的参数。

# 手动增加NA值
> ma[1,1] <- NA
> ma
     [,1] [,2] [,3] [,4]
[1,]   NA    3    1    7
[2,]    2    4    6    8

默认情况:

> apply(ma,1,mean)
[1] NA  5

添加参数na.rm=TRUE

> apply(ma,1,mean,na.rm=TRUE)
[1] 3.666667 5.000000

进阶:自定义函数

由于Python用惯了,所以在遍历数组,然后套用多个函数的时候,基本上首先都会想到用for循环。由于在R里面写循环非常低效,所以尽可能使用apply。那么如果在一次循环中涉及到多个运算,就不可能直接用R自带函数。不过好在apply允许我们自己编一个函数。

下面计算一个稍微复杂点的例子,按行循环,让矩阵的第1列加1,并计算出第1和第2列的均值,还有对每一行进行求和。
吐槽:我也不知道为什么要做这些操作,但是是为了演示函数把

ma
     [,1] [,2] [,3] [,4]
[1,]    1    3    1    7
[2,]    2    4    6    8

myfunc <- function(x){
  plusone <- x[1]+1
  mysum <- sum(x)
  mymean <- mean(c(x[1],x[2]))
  return(c(plusone,mysum,mymean))
}

apply(ma,1,myfunc)
     [,1] [,2]
[1,]    2    3
[2,]   12   20
[3,]    2    3
最后编辑于
©著作权归作者所有,转载或内容合作请联系作者
  • 序言:七十年代末,一起剥皮案震惊了整个滨河市,随后出现的几起案子,更是在滨河造成了极大的恐慌,老刑警刘岩,带你破解...
    沈念sama阅读 199,440评论 5 467
  • 序言:滨河连续发生了三起死亡事件,死亡现场离奇诡异,居然都是意外死亡,警方通过查阅死者的电脑和手机,发现死者居然都...
    沈念sama阅读 83,814评论 2 376
  • 文/潘晓璐 我一进店门,熙熙楼的掌柜王于贵愁眉苦脸地迎上来,“玉大人,你说我怎么就摊上这事。” “怎么了?”我有些...
    开封第一讲书人阅读 146,427评论 0 330
  • 文/不坏的土叔 我叫张陵,是天一观的道长。 经常有香客问我,道长,这世上最难降的妖魔是什么? 我笑而不...
    开封第一讲书人阅读 53,710评论 1 270
  • 正文 为了忘掉前任,我火速办了婚礼,结果婚礼上,老公的妹妹穿的比我还像新娘。我一直安慰自己,他们只是感情好,可当我...
    茶点故事阅读 62,625评论 5 359
  • 文/花漫 我一把揭开白布。 她就那样静静地躺着,像睡着了一般。 火红的嫁衣衬着肌肤如雪。 梳的纹丝不乱的头发上,一...
    开封第一讲书人阅读 48,014评论 1 275
  • 那天,我揣着相机与录音,去河边找鬼。 笑死,一个胖子当着我的面吹牛,可吹牛的内容都是我干的。 我是一名探鬼主播,决...
    沈念sama阅读 37,511评论 3 390
  • 文/苍兰香墨 我猛地睁开眼,长吁一口气:“原来是场噩梦啊……” “哼!你这毒妇竟也来了?” 一声冷哼从身侧响起,我...
    开封第一讲书人阅读 36,162评论 0 254
  • 序言:老挝万荣一对情侣失踪,失踪者是张志新(化名)和其女友刘颖,没想到半个月后,有当地人在树林里发现了一具尸体,经...
    沈念sama阅读 40,311评论 1 294
  • 正文 独居荒郊野岭守林人离奇死亡,尸身上长有42处带血的脓包…… 初始之章·张勋 以下内容为张勋视角 年9月15日...
    茶点故事阅读 35,262评论 2 317
  • 正文 我和宋清朗相恋三年,在试婚纱的时候发现自己被绿了。 大学时的朋友给我发了我未婚夫和他白月光在一起吃饭的照片。...
    茶点故事阅读 37,278评论 1 328
  • 序言:一个原本活蹦乱跳的男人离奇死亡,死状恐怖,灵堂内的尸体忽然破棺而出,到底是诈尸还是另有隐情,我是刑警宁泽,带...
    沈念sama阅读 32,989评论 3 316
  • 正文 年R本政府宣布,位于F岛的核电站,受9级特大地震影响,放射性物质发生泄漏。R本人自食恶果不足惜,却给世界环境...
    茶点故事阅读 38,583评论 3 303
  • 文/蒙蒙 一、第九天 我趴在偏房一处隐蔽的房顶上张望。 院中可真热闹,春花似锦、人声如沸。这庄子的主人今日做“春日...
    开封第一讲书人阅读 29,664评论 0 19
  • 文/苍兰香墨 我抬头看了看天上的太阳。三九已至,却和暖如春,着一层夹袄步出监牢的瞬间,已是汗流浃背。 一阵脚步声响...
    开封第一讲书人阅读 30,904评论 1 255
  • 我被黑心中介骗来泰国打工, 没想到刚下飞机就差点儿被人妖公主榨干…… 1. 我叫王不留,地道东北人。 一个月前我还...
    沈念sama阅读 42,274评论 2 345
  • 正文 我出身青楼,却偏偏与公主长得像,于是被迫代替她去往敌国和亲。 传闻我的和亲对象是个残疾皇子,可洞房花烛夜当晚...
    茶点故事阅读 41,856评论 2 339

推荐阅读更多精彩内容

  • 一、实验目的 学习使用 weka 中的常用分类器,完成数据分类任务。 二、实验内容 了解 weka 中 explo...
    yigoh阅读 8,379评论 5 4
  • 在挖掘分析的过程当中对字符串的处理是极为重要的,且出现也较为频繁,R语言作为当前最为流行的开源数据分析和可视化平台...
    果果哥哥BBQ阅读 5,769评论 0 8
  • 2601# Anthony Goldbloom: The jobs we'll lose to machines ...
    ucudrrad阅读 539评论 0 0
  • 我妈做饭手艺比较好,所以在我小的时候,总有各式各样的美食,不过多以肉食为主。可能新疆就是这个样子,肉才是主角,菜只...
    灿灿_e560阅读 360评论 10 4
  • 作者郝身洋 一夜风急万里汒, 轻于鹅毛重予雪。 笑看天公随意涂, 山河娇艳换玉妆。
    兰儿aa阅读 311评论 0 0