R语言:Purrr包探索

看统计书,有一道题问n人之中至少有两人生日相同的概率。感觉这道题非常适合用来练习purrr包的基本用法,于是就试了下。

对于这道题,可以先算出所有人的生日都不相同的概率,然后再用1减去这个概率。当人数为2人时,概率是:

1 - (365 / 365) * (364 / 365)

## [1] 0.002739726

当人数为3人时,概率是:

1 - (365 / 365) * (364 / 365) * (363 / 365)

## [1] 0.008204166

写成数学公式的话,大概是这样:

image

为了方便使用,可以写成函数:

samebirth <- function(n) {
  
  temp <- 1 - prod(365:(366 - n)) / 365 ^ n 
  
  print(temp)
  
}

试一下:

samebirth(2)

## [1] 0.002739726

samebirth(3)

## [1] 0.008204166

没有问题。但是要想同时计算2人和3人的概率,就会出现问题:

samebirth(2:3)

## Warning in 365:(366 - n): numerical expression has 2 elements: only the
## first used

## [1] 0.002739726 0.997267780

此时就可以利用purrr包中的map函数,把.x参数所代表的数据中的每一个元素映射到.f位置的函数上:

library(purrr)

map(2:3, samebirth)

## [1] 0.002739726
## [1] 0.008204166

## [[1]]
## [1] 0.002739726
## 
## [[2]]
## [1] 0.008204166

这次同时算出了2人和3人情况下的概率,而且数值是正确的了,但不知道为什么数值都显示了两遍。因为samebirth函数比较短,所以可以直接在map中使用匿名函数。当然,也可以使用apply家族的函数,但map中可以写成公式式匿名函数,进而使代码更简洁:

map(2:3, ~ 1 - prod(365:(366 - .x)) / 365 ^ .x)

## [[1]]
## [1] 0.002739726
## 
## [[2]]
## [1] 0.008204166

这里的prod函数的作用是累乘,2和3依次被映射到了.x处,然后计算出概率。

如果要计算更多的人数,比如50,可以将3改为50,但因为map函数默认产生列表,结果输出会很长,这是可以将map函数改为map_dbl,指定输出数值型的结果:

map_dbl(2:50, ~ 1 - prod(365 : (366 - .x)) / 365 ^ .x)

##  [1] 0.002739726 0.008204166 0.016355912 0.027135574 0.040462484
##  [6] 0.056235703 0.074335292 0.094623834 0.116948178 0.141141378
## [11] 0.167024789 0.194410275 0.223102512 0.252901320 0.283604005
## [16] 0.315007665 0.346911418 0.379118526 0.411438384 0.443688335
## [21] 0.475695308 0.507297234 0.538344258 0.568699704 0.598240820
## [26] 0.626859282 0.654461472 0.680968537 0.706316243 0.730454634
## [31] 0.753347528 0.774971854 0.795316865 0.814383239 0.832182106
## [36] 0.848734008 0.864067821 0.878219664 0.891231810 0.903151611
## [41] 0.914030472 0.923922856 0.932885369 0.940975899 0.948252843
## [46] 0.954774403 0.960597973 0.965779609 0.970373580

这时任务就完成了,但一开始我并不知道prod函数的存在,所以写的稍微复杂了一些:

不过在展示最初写法之前,还要介绍下reduce函数。

reduce函数常用于合并列表中的数据框,如:

reduce(list(iris, iris, iris), rbind) %>% dim()

## [1] 450   5

3个本来只有150行iris数据框合并到一起变成了450行。另外,reduce配合*函数使用,就实现了prod函数的功能,如:

reduce(1:5, `*`)

## [1] 120

prod(1:5)

## [1] 120

R中的一些运算符号,如上面用到的*,加上反引号(backtick,键盘esc下面那个符号),就可以像普通函数那样使用,如:

5 + 6

## [1] 11

`+`(5, 6)

## [1] 11

sum(5, 6)

## [1] 11

最初的代码是这样的:

map_dbl(2:50, ~ 1 - map(1:.x, ~ (366 - .x) / 365) %>% reduce(`*`))

##  [1] 0.002739726 0.008204166 0.016355912 0.027135574 0.040462484
##  [6] 0.056235703 0.074335292 0.094623834 0.116948178 0.141141378
## [11] 0.167024789 0.194410275 0.223102512 0.252901320 0.283604005
## [16] 0.315007665 0.346911418 0.379118526 0.411438384 0.443688335
## [21] 0.475695308 0.507297234 0.538344258 0.568699704 0.598240820
## [26] 0.626859282 0.654461472 0.680968537 0.706316243 0.730454634
## [31] 0.753347528 0.774971854 0.795316865 0.814383239 0.832182106
## [36] 0.848734008 0.864067821 0.878219664 0.891231810 0.903151611
## [41] 0.914030472 0.923922856 0.932885369 0.940975899 0.948252843
## [46] 0.954774403 0.960597973 0.965779609 0.970373580

虽然要复杂一些,但也说明map函数是可以嵌套的。

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

推荐阅读更多精彩内容