R语言 写一个函数

Posted 2021-04-18 龙氏逻辑

其实感觉R对我来说，想要去写一个函数的情况并不多。无论是在生信里还是在一般的统计中，数据格式都是固定好的。R会提供大量的各种各样的包去完成你想要完成的事。

但如果你不是仅仅把它当作一个数据分析软件，还想把它当作数据处理、整理软件，那么往往对于大数据来说，无可厚非的要用到各种循环，并且自定义一些你想要的功能。

这次我想要的是某次数据整理中，我想要创建一个拥有多个空白列的数据框。但是又不想按照正常的操作去做，因为会很麻烦，如下：

data <- data.frame(ID = ID, col1 = character(), col2 = character(), col3 = character(), coln = character())

在这里你想要创建多少个列，就要把参数输入多少次，太麻烦了。

《R数据科学》一书中有提到：当你想要做的重复操作超过3次时，就可以考虑写一个函数和循环了。

于是写了以下代码来替换上面的操作：

# 先创建一个只有一列的数据框。data <- data.frame(ID = ID)
# 函数内容fun_newcol <- function(x,y){ for (i in c(1:y)){ lieming <- paste('col_', i, sep = '') x <- transform(x, newcol = NA) colnames(x)[i] <- lieming } return(x)}
# 执行函数，为数据框增加46列（注意，由于我第一列是已存在，所以增加46列时参数是47，具体情况具体分析嘛）data <- fun_newcol(data, 47)# 此时为新的数据框添加列名colnames(data) <- c("ID", jiancha)

由于之前没有学过怎么编写函数，只是看过一些。

所以写起来报了很多次错。希望下次能改进。

其次，需要注意的地方：

1、for循环其实在R里效率低下，采用apply系列函数会更好。

这个网址讲的挺好，可以参考：http://blog.fens.me/r-apply/

2、写完函数后记得写return。。。。实在是太菜了我，这个问题我居然卡了好久，还在想怎么函数运行完怎么不出结果。