16

如何在函数中写入一种方法来检测输出是否被分配（<-）给某物？原因是我想打印一条消息，如果它没有被分配，只是去控制台，但如果它被分配，我不希望它打印消息。

这是一个虚拟示例以及我希望它的行为方式：

fun <- function(x) {
    if (being_assigned) {
        print("message")
    }
    return(x)
}

#no assignment so message prints 
> fun(6)  
[1] "message"
[1] 6

#assignment so message does not prints
> x <- fun(6)

函数中的being_assigned是我想检测但不知道如何检测的虚构未知条件。

4

2 回答 2

14

我认为你能做的最好的就是为函数返回的对象定义一个特殊的打印方法：

## Have your function prepend "myClass" to the class of the objects it returns
fun <- function(x) {
    class(x) <- c("myClass", class(x))
    x
}

## Define a print method for "myClass". It will be dispatched to 
## by the last step of the command line parse-eval-print cycle.
print.myClass <- function(obj) {
    cat("message\n")
    NextMethod(obj)
}

> fun(1:10)
message
 [1]  1  2  3  4  5  6  7  8  9 10
attr(,"class")
[1] "myClass"
>
> out <- fun(1:10)
>
于 2012-10-24T19:51:55.553 回答
2

我喜欢 Josh 的想法，但对于未来的海报，我想展示我所做的，这是他方法的略微修改版本。他的方法打印出我唯一不喜欢的课程信息。他用NextMethod来避免无限递归打印。这导致

attr(,"class")
[1] "myClass"

要打印。所以为了避免这种情况，我首先打印消息，然后通过类对象的长度打印 1（使用索引）。

fun <- function(x) {
    class(x) <- 'beep'
    comment(x) <- "hello world"
    return(x)
}

print.beep<- function(beep) {
    cat(paste0(comment(beep), "\n"))
    print(beep[1:length(beep)])
}


> fun(1:10)
hello world
 [1]  1  2  3  4  5  6  7  8  9 10

再次感谢乔希的想法。

如果读者不想[1]打印小索引，他们可以cat在打印语句中输出如下：

print.beep<- function(beep) {
    cat(paste0(comment(beep), "\n"))
    cat(beep[1:length(beep)], "\n")
}
于 2012-11-03T14:56:27.280 回答