Установите элементы, которые меньше нуля в столбцах 1 и 3, в ноль в R

Question

У меня 4 на 5 случайных матриц. Как я могу установить элементы, которые меньше нуля в столбцах 1 и 3 на ноль в R? Я пытался использовать replace() функция.

r <- rnorm(20, mean=2, sd=2)
r1 <- matrix(r, ncol=5, nrow=4, byrow=T)
replace(r1, r1[,c(1,3)] < 0, 0) 

Спасибо за любую помощь!

Answer 1

Вы были близки, код должен быть

r1[, c(1,3)] <- replace(r1[, c(1,3)], r1[, c(1,3)] < 0, 0)
#          [,1]     [,2]     [,3]        [,4]     [,5]
#[1,] 0.2061709 2.369698 5.175691 -0.26075135 1.839496
#[2,] 2.2648406 3.415909 1.520604  5.96894787 1.722426
#[3,] 2.8353015 3.963506 1.214609 -0.07933795 5.564458
#[4,] 0.0000000 3.757209 2.071613  4.02565738 2.864530

---

В replace вам нужно указать столбцы 1 и 3 для x аргумент, а также list аргумент. Проблема с вашим кодом заключается в том, что list аргумент get переработан, и вы получите неправильные результаты.

replace(r1,r1[,c(1,3)] <0, 0)
#          [,1]     [,2]     [,3]        [,4]     [,5]
#[1,] 0.2061709 2.369698 5.175691 -0.26075135 1.839496
#[2,] 2.2648406 3.415909 1.520604  5.96894787 1.722426
#[3,] 2.8353015 3.963506 1.214609 -0.07933795 5.564458
#[4,] 0.0000000 3.757209 0.000000  4.02565738 0.000000

данные

set.seed(2)
(r1 <- matrix(rnorm(20, mean=2, sd=2), ncol=5, nrow=4, byrow=T))
#           [,1]     [,2]     [,3]        [,4]     [,5]
#[1,]  0.2061709 2.369698 5.175691 -0.26075135 1.839496
#[2,]  2.2648406 3.415909 1.520604  5.96894787 1.722426
#[3,]  2.8353015 3.963506 1.214609 -0.07933795 5.564458
#[4,] -2.6221382 3.757209 2.071613  4.02565738 2.864530

Answer 2

Мне действительно нравится использовать функции min и max для таких вещей. В этом случае мы хотим pmax в Р.

r1[,c(1,3)] <- pmax(r1[,c(1,3)], 0)

Или немного более запутанный:

r1[,c(1,3)][r1[,c(1,3)]<0] <- 0

эталонный тест

И просто, чтобы удовлетворить вопрос, похоже, что более запутанный путь самый быстрый, но replace тоже довольно быстро

library(microbenchmark)

microbenchmark(
  r1[,c(1,3)] <- replace(r1[, c(1,3)], r1[, c(1,3)] < 0, 0),
  r1[,c(1,3)][r1[,c(1,3)]<0] <- 0,
  r1[,c(1,3)] <- pmax(r1[,c(1,3)], 0)
)

Unit: microseconds
                                                          expr    min      lq     mean median      uq    max neval
 r1[, c(1, 3)] <- replace(r1[, c(1, 3)], r1[, c(1, 3)] < 0, 0)  3.601  3.9005  4.97602  4.201  4.5020 25.501   100
                         r1[, c(1, 3)][r1[, c(1, 3)] < 0] <- 0  3.200  3.3020  4.75701  3.601  3.9020 24.201   100
                       r1[, c(1, 3)] <- pmax(r1[, c(1, 3)], 0) 12.601 13.1010 17.42398 13.901 15.5015 49.601   100