在python中调用R的pwr功效函数

已封装为函数,

文件名 - PyPower.py

# 推荐使用conda管理环境
# conda create --name test python=3.6
# source activate test
# conda install rpy2  # should add conda-forge channel
# reference link: <http://www.cnblogs.com/cloudtj/articles/6372200.html>, <https://rpy2.readthedocs.io/en/version_2.8.x/index.html>

def pyPower_ttest(n='NULL', d='NULL', sig_level=0.05, power='NULL', type="two.sample", alternative='two.sided'):
    ''' 
    Compute power of tests or determine parameters to obtain target power. Exactly one of the parameters 'd','n','power' and 'sig_level' must be passed as 'NULL'. Results returned as a dictory. 

    This is an API function calling pwr.t.test function of R in Python using rpy2, pwr package (R environment) must be pre-installed.

    Arguments:
        n   
                Number of observations (per sample)
        d   
                Effect size (Cohen's d) - difference between the means divided by the pooled standard deviation
        sig_level   
                Significance level (Type I error probability)
        power   
                Power of test (1 minus Type II error probability)
        type    
                Type of t test : one- two- or paired-samples
        alternative 
                a character string specifying the alternative hypothesis, must be one of "two.sided" (default), "greater" or "less"

    Example:
    
        from PyPower import pyPower_ttest
        pyPower_ttest(d=0.2,n=60,sig_level=0.10,type="one.sample",alternative="two.sided")
        pyPower_ttest(d=2/2.8,n=30,sig_level=0.05,type="two.sample",alternative="two.sided")
        pyPower_ttest(d=0.3,power=0.75,sig_level=0.05,type="two.sample",alternative="greater")

    Note:
        
        More information please run ?pwr::pwr.t.test in R console
    '''

    # 载入r对象
    from rpy2 import robjects
    # 载入导入包函数
    from rpy2.robjects.packages import importr
    # 将pwr包导入为模块,记得提前在(test环境下)R中安装好pwr包
    pwr = importr('pwr')
# When one wants to create a vector from Python, either the class Vector or the convenience classes IntVector, FloatVector, BoolVector, StrVector can be used.
    
    # 将参数中的.替换为_,解决不兼容问题, 来自rpy2文档函数部分
    # rpy2默认支持该转换,可以删除该部分以简化代码
    def iamfeelinglucky(func):
        def f(*args, **kwargs):
            d = {}
            for k, v in kwargs.items():
                d[k.replace('_', '.')] = v
            return func(**d)
        return f

    # 矫正参数名
    power_ttest = iamfeelinglucky(pwr.pwr_t_test)

    # 重命名r中的as.null函数,用于将字符'NULL'转换为NULL对象
    as_null = robjects.r['as.null']
    # 转换
    args_list = [n, d, sig_level, power]
    for i,v in enumerate(args_list):
        if v == "NULL":
            args_list[i] = as_null(v)

    # 调用函数
    pwr_res = power_ttest(n=args_list[0], d=args_list[1], sig_level=args_list[2], power=args_list[3], type=type, alternative=alternative)

    # 输出分析结果       
    print(pwr_res)
    res = list(pwr_res)
    # 返回结果中需要的值构建字典
    res = {'n':res[0], 'd':res[1], 'sig_level':res[2], 'power':res[3], 'alternative':res[4], 'method':res[6]}
    # 返回的是robject对象,再转换为标准的python对象
    res['n'] = list(res['n'])[0]
    res['d'] = list(res['d'])[0]
    res['sig_level'] = list(res['sig_level'])[0]
    res['power'] = list(res['power'])[0]
    res['alternative'] = list(res['alternative'])[0]
    res['method'] = list(res['method'])[0]

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

推荐阅读更多精彩内容

  • 〇、前言 本文共108张图,流量党请慎重! 历时1个半月,我把自己学习Python基础知识的框架详细梳理了一遍。 ...
    Raxxie阅读 18,910评论 17 410
  • C++调用python 在C/C++中嵌入Python,可以使用Python提供的强大功能,通过嵌入Python可...
    Bruce_Szh阅读 13,756评论 1 7
  • Python 面向对象Python从设计之初就已经是一门面向对象的语言,正因为如此,在Python中创建一个类和对...
    顺毛阅读 4,207评论 4 16
  • 还记得那个一句我以用尽了洪荒之力的姑娘吗?我相信大家对这个姑娘应该不会陌生,她叫傅园慧是中国女子游泳队!对于这个这...
    大麦文化阅读 263评论 0 0
  • 细节的改变带来的是潜意识里的暗示,可以在别人意识不到的情况下对一个人进行潜移默化的影响。 因此,在为人处世时,多注...
    冰棍儿嗝阅读 245评论 3 2