R語(yǔ)言函數(shù)如何返回多個(gè)值
問(wèn)題
之前想嘗試一下使用perl中的或者python風(fēng)格的方式來(lái)寫(xiě)R的函數(shù)的時(shí)候，想返回兩個(gè)參數(shù)
于是就這樣寫(xiě)了：
return_two_value <- function(a, b){
return(a, b)
}
運(yùn)行一下
return_two_value(2, 3)
結(jié)果
Error in return(a, b) : multi-argument returns are not permitted
報(bào)錯(cuò)了，之前的那種寫(xiě)法不可以，原因是return不能返回多個(gè)值。
于是又想到把這個(gè)值存到向量里面再取出來(lái)不就好了
return_two_value <- function(a, b){
l <- c(a, b)
return(l)
}
運(yùn)行
# 先賦值
result <- return_two_value(2, 3)
a <- result[1] # 2
b <- result[2] # 3
這樣就可以了，但是有時(shí)候函數(shù)的返回值根本就不是單個(gè)元素，很多時(shí)候是向量啊，數(shù)據(jù)框之類(lèi)的，這個(gè)時(shí)候在這么寫(xiě)可以試一下
# 試一下向量
result <- return_two_value(2, c(1,2,3))
a <- result[1] # 2
b <- result[2] # 1
# 試一下數(shù)據(jù)框
result <- return_two_value(2, data.frame(c("A", "B", "C"), c("a", "b", "c")))
a <- result[1] # [[1]] [1] 2
b <- result[2] # $c..A....B....C.. [1] A B C Levels: A B C
感覺(jué)已經(jīng)亂了亂了！這樣做只能用于兩個(gè)標(biāo)量的時(shí)候，對(duì)于傳入比較復(fù)雜的數(shù)據(jù)的時(shí)候返回結(jié)果就不行了，當(dāng)然你可能也會(huì)說(shuō)把結(jié)果放在數(shù)據(jù)框中，但是數(shù)據(jù)框會(huì)自動(dòng)填值，也不是個(gè)好辦法。
解決辦法
在stackoverflow上面找到了一個(gè)解決辦法，它是將數(shù)據(jù)放在列表里面的。
- 首先寫(xiě)一個(gè)示例函數(shù)，將返回值放在列表中，并且指明兩個(gè)返回值的標(biāo)題
return_two_value <- function(a, b){
# 當(dāng)然這里為了說(shuō)明，將返回結(jié)果及其簡(jiǎn)化，你可以按照自己的方式編寫(xiě)然后以類(lèi)似的方式返回
out <- list(one=a, two=b)
return(out)
}
- 使用
$根據(jù)標(biāo)題索引兩個(gè)元素（或者根據(jù)位置索引也可以啊）
result <- return_two_value(2, data.frame("group1" = c("A", "B", "C"), "group2" = c("a", "b", "c")))
# 或者 first = result[[1]]
first = result$one # [1] 2
# 或者 second = result[[2]]
second = result$two
# group1 group2
# 1 A a
# 2 B b
# 3 C c
??試了試排名第一的回答，發(fā)現(xiàn)這樣做好像不好使
library(gsubfn)
return_two_value <- function(a, b){
out <- list(one=a, two=b)
return(out)
}
list[first, second] = return_two_value(2, data.frame("group1" = c("A", "B", "C"), "group2" = c("a", "b", "c")))
好像報(bào)錯(cuò)了，不知道是我沒(méi)有安裝某些依賴(lài)的包，不過(guò)上面那種直接索引的方式還是可以的。
實(shí)際應(yīng)用
之前是因?yàn)橄雽?xiě)一個(gè)類(lèi)似于perl中的shift或者python中pop的功能的函數(shù)，這個(gè)函數(shù)就是將一個(gè)數(shù)組（對(duì)應(yīng)到R中可能就是向量）中的第一個(gè)元素取出并且返回。
- python
l = [1, 2, 3, 4]
l.pop(0)
print(l) # [2, 3, 4]
- perl
my @list = (1, 2, 3, 4);
shift(@list);
print join(",", @list); # 2,3,4
我首先是這么寫(xiě)的
- R
l <- c(1, 2, 3, 4)
# 傳入一個(gè)向量
shift <- function(l) {
first = l[1]
a = l[-1]
return(first)
}
i <- shift(l) # [1] 1
print(l) # [1] 1 2 3 4
我是先拿到第一個(gè)元素first = l[1]，然后去除第一個(gè)元素a = l[-1]，但是后來(lái)發(fā)現(xiàn)傳入的向量并沒(méi)有發(fā)生更改，就是函數(shù)好像不能改變外部的變量。
后來(lái)就嘗試使用另外一種方式，就是返回兩個(gè)東西，代碼已經(jīng)在開(kāi)頭示出了。
后來(lái)嘗試新的方法寫(xiě)了一下
l <- c(1, 2, 3, 4)
shift <- function(l) {
first = l[1]
a = l[-1]
return( list( one=first, two=a ) )
}
result <- shift(l)
first_item <- result$one # [1] 1
# 這個(gè)時(shí)候再將返回值賦值給l
l <- result$two # [1] 2 3 4
這種方法我想在處理大文件的時(shí)候可能不快，但是好歹也算弄出來(lái)了。
方法來(lái)源
- stackoverflow - How to assign from a function which returns more than one value?
參考
- philzang的回答 - R 如何實(shí)現(xiàn) 向量中某個(gè)元素的刪除?
