Update data frame via function doesn't work
test
in your function is a copy of the object from your global environment (I'm assuming that's where it is defined). Assignment happens in the current environment unless specified otherwise, so any changes that happen inside the function apply only to the copy inside the function, not the object in your global environment.
And it's good form to pass all necessary objects as arguments to the function.
Personally, I would return(test)
at the end of your function and make the assignment outside of the function, but I'm not sure if you can do this in your actual situation.
test.fun <- function (x, test) {
test[test$v1==x,"v2"] <- 10
return(test)
}
test <- data.frame(v1=c(rep(1,3),rep(2,3)),v2=0)
(test <- test.fun(1, test))
# v1 v2
#1 1 10
#2 1 10
#3 1 10
#4 2 0
#5 2 0
#6 2 0
If it is absolutely necessary to modify an object outside your function directly, so you need to tell R that you want to assign the local copy of test
to the test
in the .GlobalEnv
.
test.fun <- function (x, test) {
test[test$v1==x,"v2"] <- 10
assign('test',test,envir=.GlobalEnv)
#test <<- test # This also works, but the above is more explicit.
}
(test.fun(1, test))
# v1 v2
#1 1 10
#2 1 10
#3 1 10
#4 2 0
#5 2 0
#6 2 0
Using assign
or <<-
in this fashion is fairly uncommon, though, and many experienced R programmers will recommend against it.
Changing the <- to <<- in your function, does the trick as well, see the R-manual . Quote from that page:
The operators <<- and ->> are normally only used in functions, and cause a search to made through parent environments for an existing definition of the variable being assigned. If such a variable is found (and its binding is not locked) then its value is redefined, otherwise assignment takes place in the global environment.
Your code should then be:
test <- data.frame(v1=c(rep(1,3),rep(2,3)),v2=0)
test.fun <- function (x) {
test[test$v1==x,"v2"] <<- 10
print(test)
}
test.fun(1)
It is good practice to not change global variables in functions, because this may have undesirable side effects. To avoid this in R, any changes to objects inside a function actually only change copies that are local to that function's environment
.
If you really want to change test, you have to assign the return value of the function to test (it would be better to write the function with a more explicit return value,
test <- test.fun(1)
Or choose the global environment to assign to within test.fun
,
test.fun <- function (x) {
test[test$v1==x,"v2"] <- 10
print(test)
assign("test",test,.GlobalEnv)
}