代碼書寫01--argparse命令行參數(shù)解析

在命令行中運行R腳本,可以使用Rscript practice.R的方式后豫,其中practice.R是我們希望運行的腳本涌穆。在Linux和Unix系統(tǒng)中,有一種更簡介的方式独旷,即通過在腳本第一行添加#! /usr/bin/env Rscript的方式署穗,來告訴操作系統(tǒng)這段腳本的要調用Rscript來運行。此時只要進入到practice.R所在的路徑中嵌洼,使用命令./practice.R便可以直接運行這段代碼案疲。類似地,如果是一段Python3腳本的話麻养,我們可以使用#! /usr/bin/env Python3

argparse is an R package which provides a command line parser to be used with Rscript to write "#!" shebang scripts that gracefully accept positional and optional arguments and automatically generate usage.

argparse包分為三個步驟:

  1. 使用ArgumentParser()創(chuàng)建一個參數(shù)解析對象
  2. 使用add_argument()為前面的對象增加參數(shù)
  3. 通過parse_args()把解析對象賦值并在后續(xù)程序中調用

我們可以先把下面的腳本儲存成為example.R褐啡,這段腳本的作用是輸出若干個隨機數(shù),具體的隨機分布與輸出參數(shù)數(shù)量由命令行參數(shù)決定鳖昌。

#! /usr/bin/env Rscript
library(argparse)

# 創(chuàng)建參數(shù)解析對象
parser <- ArgumentParser()

# 設置參數(shù)
# 設置第一個參數(shù)verbose备畦,縮寫為v,其作用是告訴腳本是否打印完整的計算過程许昨,其缺省值為TRUE
parser$add_argument("-v", "--verbose", action="store_true", default=TRUE,
        help="Print extra output [default]")

# 設置第二個參數(shù)quietly懂盐,縮寫為q,其作用是修改verbose參數(shù)糕档,當調用改參數(shù)時莉恼,verbose被修改為FALSE,從而導致不再打印計算過程
parser$add_argument("-q", "--quietly", action="store_false", 
        dest="verbose", help="Print little output")

# 設置第三個參數(shù)count速那,縮寫為c俐银,這是一個整數(shù)參數(shù),缺省值5琅坡,在后續(xù)的代碼中被用作確定輸出隨機數(shù)的個數(shù)
parser$add_argument("-c", "--count", type="integer", default=5, 
        help="Number of random normals to generate [default %(default)s]",
        metavar="number")

# 設置第四個參數(shù)generator悉患,無縮寫,用于確定調用何種隨機分布榆俺,缺省值rnorm對應于正態(tài)分布
parser$add_argument("--generator", default="rnorm", 
        help = "Function to generate random deviates [default \"%(default)s\"]")

# 設置第五個參數(shù)mean售躁,無縮寫坞淮,浮點數(shù),用于確定正態(tài)分布的均值陪捷,缺省值為0
parser$add_argument("--mean", default=0, type="double",
        help="Mean if generator == \"rnorm\" [default %(default)s]")

# 設置第五個參數(shù)sd回窘,無縮寫,浮點數(shù)市袖,用于確定正態(tài)分布的標準差啡直,缺省值為1
parser$add_argument("--sd", default=1, type="double",
        metavar="standard deviation",
        help="Standard deviation if generator == \"rnorm\" [default %(default)s]")

# 調用解析器,此時args就被賦值為命令行參數(shù)輸入的相應值
args <- parser$parse_args()

# 根據(jù)verbose確定是否打印計算過程
if ( args$verbose ) { 
        write("writing some verbose output to standard error...\n", stderr()) 
}

# 根據(jù)其他參數(shù)苍碟,確定輸出隨機數(shù)的類型與次數(shù)
if( args$generator == "rnorm") {
        cat(paste(rnorm(args$count, mean=args$mean, sd=args$sd), collapse="\n"))
} else {
        cat(paste(do.call(args$generator, list(args$count)), collapse="\n"))
}
cat("\n")

首先修改腳本的權限酒觅,為其添加可執(zhí)行權限

sudo chmod +x example.R

然后便可以使用了

  • 查看幫助
./example.R --help
  • 執(zhí)行腳本
./practice.R

使用參數(shù)輸出3個正態(tài)分布隨機數(shù)

./practice.R --mean=10 --sd=10 --count=3

也可以輸出均勻分布隨機數(shù),并且選擇不打印計算過程

./practice.R --quiet -c 4 --generator="runif"

參考:
Data Science at the Command Line
argparse: Command line optional and positional argument parser

最后編輯于
?著作權歸作者所有,轉載或內容合作請聯(lián)系作者
禁止轉載微峰,如需轉載請通過簡信或評論聯(lián)系作者舷丹。
  • 序言:七十年代末,一起剝皮案震驚了整個濱河市蜓肆,隨后出現(xiàn)的幾起案子颜凯,更是在濱河造成了極大的恐慌,老刑警劉巖仗扬,帶你破解...
    沈念sama閱讀 206,126評論 6 481
  • 序言:濱河連續(xù)發(fā)生了三起死亡事件症概,死亡現(xiàn)場離奇詭異,居然都是意外死亡早芭,警方通過查閱死者的電腦和手機彼城,發(fā)現(xiàn)死者居然都...
    沈念sama閱讀 88,254評論 2 382
  • 文/潘曉璐 我一進店門,熙熙樓的掌柜王于貴愁眉苦臉地迎上來逼友,“玉大人精肃,你說我怎么就攤上這事≈钠颍” “怎么了司抱?”我有些...
    開封第一講書人閱讀 152,445評論 0 341
  • 文/不壞的土叔 我叫張陵,是天一觀的道長黎烈。 經常有香客問我习柠,道長,這世上最難降的妖魔是什么照棋? 我笑而不...
    開封第一講書人閱讀 55,185評論 1 278
  • 正文 為了忘掉前任资溃,我火速辦了婚禮,結果婚禮上烈炭,老公的妹妹穿的比我還像新娘溶锭。我一直安慰自己,他們只是感情好符隙,可當我...
    茶點故事閱讀 64,178評論 5 371
  • 文/花漫 我一把揭開白布趴捅。 她就那樣靜靜地躺著垫毙,像睡著了一般。 火紅的嫁衣襯著肌膚如雪拱绑。 梳的紋絲不亂的頭發(fā)上综芥,一...
    開封第一講書人閱讀 48,970評論 1 284
  • 那天,我揣著相機與錄音猎拨,去河邊找鬼膀藐。 笑死,一個胖子當著我的面吹牛红省,可吹牛的內容都是我干的额各。 我是一名探鬼主播,決...
    沈念sama閱讀 38,276評論 3 399
  • 文/蒼蘭香墨 我猛地睜開眼吧恃,長吁一口氣:“原來是場噩夢啊……” “哼臊泰!你這毒婦竟也來了?” 一聲冷哼從身側響起蚜枢,我...
    開封第一講書人閱讀 36,927評論 0 259
  • 序言:老撾萬榮一對情侶失蹤,失蹤者是張志新(化名)和其女友劉穎针饥,沒想到半個月后厂抽,有當?shù)厝嗽跇淞掷锇l(fā)現(xiàn)了一具尸體,經...
    沈念sama閱讀 43,400評論 1 300
  • 正文 獨居荒郊野嶺守林人離奇死亡丁眼,尸身上長有42處帶血的膿包…… 初始之章·張勛 以下內容為張勛視角 年9月15日...
    茶點故事閱讀 35,883評論 2 323
  • 正文 我和宋清朗相戀三年筷凤,在試婚紗的時候發(fā)現(xiàn)自己被綠了。 大學時的朋友給我發(fā)了我未婚夫和他白月光在一起吃飯的照片苞七。...
    茶點故事閱讀 37,997評論 1 333
  • 序言:一個原本活蹦亂跳的男人離奇死亡藐守,死狀恐怖,靈堂內的尸體忽然破棺而出蹂风,到底是詐尸還是另有隱情卢厂,我是刑警寧澤,帶...
    沈念sama閱讀 33,646評論 4 322
  • 正文 年R本政府宣布惠啄,位于F島的核電站慎恒,受9級特大地震影響,放射性物質發(fā)生泄漏撵渡。R本人自食惡果不足惜融柬,卻給世界環(huán)境...
    茶點故事閱讀 39,213評論 3 307
  • 文/蒙蒙 一、第九天 我趴在偏房一處隱蔽的房頂上張望趋距。 院中可真熱鬧粒氧,春花似錦、人聲如沸节腐。這莊子的主人今日做“春日...
    開封第一講書人閱讀 30,204評論 0 19
  • 文/蒼蘭香墨 我抬頭看了看天上的太陽。三九已至门怪,卻和暖如春骡澈,著一層夾襖步出監(jiān)牢的瞬間,已是汗流浹背掷空。 一陣腳步聲響...
    開封第一講書人閱讀 31,423評論 1 260
  • 我被黑心中介騙來泰國打工肋殴, 沒想到剛下飛機就差點兒被人妖公主榨干…… 1. 我叫王不留,地道東北人坦弟。 一個月前我還...
    沈念sama閱讀 45,423評論 2 352
  • 正文 我出身青樓护锤,卻偏偏與公主長得像,于是被迫代替她去往敵國和親酿傍。 傳聞我的和親對象是個殘疾皇子烙懦,可洞房花燭夜當晚...
    茶點故事閱讀 42,722評論 2 345

推薦閱讀更多精彩內容