2012-06-30 119 views
11

我們希望將數組零中的所有值設置爲負數。 我嘗試了很多東西,但還沒有實現可行的解決方案。 我想過一個for循環的條件,但是這似乎不起作用。R:將負值替換爲零

#pred_precipitation is our array 
pred_precipitation <-rnorm(25,2,4)  

for (i in nrow(pred_precipitation)) 
{ 
    if (pred_precipitation[i]<0) {pred_precipitation[i] = 0} 
    else{pred_precipitation[i] = pred_precipitation[i]} 
} 

感謝您的提示!

回答

38

感謝您的可重複的例子。這是非常基本的R的東西。您可以分配到(注意數組有尺寸的,什麼你給是一個向量不是數組)向量的選擇的元素:

> pred_precipitation[pred_precipitation<0] <- 0 
> pred_precipitation 
[1] 1.2091281 0.0000000 7.7665555 0.0000000 0.0000000 0.0000000 0.5151504 0.0000000 1.8281251 
[10] 0.5098688 2.8370263 0.4895606 1.5152191 4.1740177 7.1527742 2.8992215 4.5322934 6.7180530 
[19] 0.0000000 1.1914052 3.6152333 0.0000000 0.3778717 0.0000000 1.4940469 

基準戰爭!

@James找到了更快的方法,並將其留在評論中。我贊成他,如果只是因爲我知道他的勝利將是短命的。

首先,我嘗試編譯,但似乎並沒有幫助任何人:

p <- rnorm(10000) 
gsk3 <- function(x) { x[x<0] <- 0; x } 
jmsigner <- function(x) ifelse(x<0, 0, x) 
joshua <- function(x) pmin(x,0) 
james <- function(x) (abs(x)+x)/2 
library(compiler) 
gsk3.c <- cmpfun(gsk3) 
jmsigner.c <- cmpfun(jmsigner) 
joshua.c <- cmpfun(joshua) 
james.c <- cmpfun(james) 

microbenchmark(joshua(p),joshua.c(p),gsk3(p),gsk3.c(p),jmsigner(p),james(p),jmsigner.c(p),james.c(p)) 
      expr  min  lq median  uq  max 
1  gsk3.c(p) 251.782 255.0515 266.8685 269.5205 457.998 
2  gsk3(p) 256.262 261.6105 270.7340 281.3560 2940.486 
3 james.c(p) 38.418 41.3770 43.3020 45.6160 132.342 
4  james(p) 38.934 42.1965 43.5700 47.2085 4524.303 
5 jmsigner.c(p) 2047.739 2145.9915 2198.6170 2291.8475 4879.418 
6 jmsigner(p) 2047.502 2169.9555 2258.6225 2405.0730 5064.334 
7 joshua.c(p) 237.008 244.3570 251.7375 265.2545 376.684 
8  joshua(p) 237.545 244.8635 255.1690 271.9910 430.566 

compiled comparison

別急!德克寫了這個Rcpp的東西。一個完整的C++能力不足可以閱讀他的JSS論文,調整他的例子,並且寫出他們最快的功能?敬請期待,親愛的聽衆。

library(inline) 
cpp_if_src <- ' 
    Rcpp::NumericVector xa(a); 
    int n_xa = xa.size(); 
    for(int i=0; i < n_xa; i++) { 
    if(xa[i]<0) xa[i] = 0; 
    } 
    return xa; 
' 
cpp_if <- cxxfunction(signature(a="numeric"), cpp_if_src, plugin="Rcpp") 
microbenchmark(joshua(p),joshua.c(p),gsk3(p),gsk3.c(p),jmsigner(p),james(p),jmsigner.c(p),james.c(p), cpp_if(p)) 
     expr  min  lq median  uq  max 
1 cpp_if(p) 8.233 10.4865 11.6000 12.4090 69.512 
2  gsk3(p) 170.572 172.7975 175.0515 182.4035 2515.870 
3 james(p) 37.074 39.6955 40.5720 42.1965 2396.758 
4 jmsigner(p) 1110.313 1118.9445 1133.4725 1164.2305 65942.680 
5 joshua(p) 237.135 240.1655 243.3990 250.3660 2597.429 

with rcpp comparison

這是肯定的,隊長。

即使您未分配,也會修改輸入p。如果你想避免這種行爲,你必須克隆:

cpp_ifclone_src <- ' 
    Rcpp::NumericVector xa(Rcpp::clone(a)); 
    int n_xa = xa.size(); 
    for(int i=0; i < n_xa; i++) { 
    if(xa[i]<0) xa[i] = 0; 
    } 
    return xa; 
' 
cpp_ifclone <- cxxfunction(signature(a="numeric"), cpp_ifclone_src, plugin="Rcpp") 

這不幸的是殺死了速度優勢。

+0

阿里和@DirkEddelbuettel:它是否真的修改'p'不分配?這似乎並沒有當我嘗試它。 – Aaron

+0

@Aaron請參閱Dirk的解釋:http://stackoverflow.com/questions/11300048/rcpp-pass-by-reference-vs-by-value –

4

另外,您還可以使用ifelse

ifelse(pred_precipitation < 0, 0, pred_precipitation) 
11

我會用pmax因爲ifelse可以在時間和集替換有點慢創建一個額外的載體(可以是大型數據集的問題) 。

set.seed(21) 
pred_precipitation <- rnorm(25,2,4) 
p <- pmax(pred_precipitation,0) 

子集替換是遠雖然最快:

library(rbenchmark) 
gsk3 <- function(x) { x[x<0] <- 0; x } 
jmsigner <- function(x) ifelse(x<0, 0, x) 
joshua <- function(x) pmin(x,0) 
benchmark(joshua(p), gsk3(p), jmsigner(p), replications=10000, order="relative") 
     test replications elapsed relative user.self sys.self 
2  gsk3(p)  10000 0.215 1.000000  0.216 0.000 
1 joshua(p)  10000 0.444 2.065116  0.416 0.016 
3 jmsigner(p)  10000 0.656 3.051163  0.652 0.000 

autoplot microbenchmark

+0

+1。添加了一個時間圖(使用'taRifx'包中的'autoplot.microbenchmark') –

+0

@ gsk3:哇,你做了什麼使我的解決方案變得更糟? :P –

+0

Kvit你的垂涎欲滴或面對我的人talonz。儘管如此,我再次運行它,似乎是rbenchmark結果和'microbenchmark'結果之間的一致性差異,至少在我的系統上是這樣。 〜2x與〜3x時間差異。 –