在Rdata.table中分隔正負值（多列）

我想知道如何通過 data.table 中的符號拆分許多列。具體來說，假設我們有：

library(data.table)
DT = data.table(x = c(-1,-2,1,3),
                z = c(-1,-1,-1,-1))

我希望創建一個名為的新 data.table DT_new，它看起來像：

 DT_new
    x  z x_pos x_neg z_pos z_neg
1: -1 -1     0     1     0     1
2: -2 -1     0     2     0     1
3:  1 -1     1     0     0     1
4:  3 -1     3     0     0     1

我這樣做的原因是我想在回歸中分離出正變數和負變數。手動完成其中的一些很容易。但是我有數百個變數想要應用這種技術。所以我希望有一個“SDcols”解決方案。

謝謝！

uj5u.com熱心網友回復：

不需要使用 .SDcols ;-) 請在下面找到一個代表：

• 代碼

DT[,`:=` (x_pos = fifelse(x>0, x, 0),
          x_neg = fifelse(x<0, abs(x), 0),
          z_pos = fifelse(z>0, z, 0),
          z_neg = fifelse(z<0, abs(z), 0))][]

• 輸出

    x  z x_pos x_neg z_pos z_neg
1: -1 -1     0     1     0     1
2: -2 -1     0     2     0     1
3:  1 -1     1     0     0     1
4:  3 -1     3     0     0     1

作為您評論的后續行動

請在下面找到reprex。

• 代碼

vars <- c("x","z")
suffix <- c("_pos", "_neg")

DT[, CJ(vars, suffix, sorted = FALSE)[, paste0(vars, suffix)] := .(fifelse(x>0, x, 0),
                                                                   fifelse(x<0, abs(x), 0), 
                                                                   fifelse(z>0, z, 0),
                                                                   fifelse(z<0, abs(z), 0))][]

• 輸出

#>     x  z x_pos x_neg z_pos z_neg
#> 1: -1 -1     0     1     0     1
#> 2: -2 -1     0     2     0     1
#> 3:  1 -1     1     0     0     1
#> 4:  3 -1     3     0     0     1

^{由reprex 包(v2.0.1)于 2021 年 11 月 28 日創建}

uj5u.com熱心網友回復：

我們可以使用across帶case_when：

library(dplyr)
DT %>% 
  mutate(across(everything(), ~case_when(
    . < 0 ~ 0,
    TRUE ~ .), .names = "{col}_pos")) %>% 
  mutate(across(-contains("pos"), ~case_when(
    . < 0 ~ abs(.),
    TRUE ~ 0), .names = "{col}_neg"))

    x  z x_pos z_pos x_neg z_neg
1: -1 -1     0     0     1     1
2: -2 -1     0     0     2     1
3:  1 -1     1     0     0     1
4:  3 -1     3     0     0     1

uj5u.com熱心網友回復：

也許：

library(data.table)
DT = data.table(x = c(-1,-2,1,3),
                z = c(-1,-1,-1,-1))

col_nms <- c('x', 'z')
pos_nms <- paste0(col_nms, '_pos')
neg_nms <- paste0(col_nms, '_neg')

DT[, c(pos_nms) := lapply(.SD, function(.x) fifelse(.x > 0, .x, 0)), .SDcols = c('x', 'z')]
DT[, c(neg_nms) := lapply(.SD, function(.x) fifelse(.x < 0, -.x, 0)), .SDcols = c('x', 'z')]

DT
#>     x  z x_pos z_pos x_neg z_neg
#> 1: -1 -1     0     0     1     1
#> 2: -2 -1     0     0     2     1
#> 3:  1 -1     1     0     0     1
#> 4:  3 -1     3     0     0     1

^{由reprex 包(v2.0.1)于 2021 年 11 月 27 日創建}

uj5u.com熱心網友回復：

library(data.table)
DT = data.table(x = c(-1,-2,1,3),
                z = c(-1,-1,-1,-1))

vars <- names(DT)

DT <- DT[, sapply(.SD, function(j){
  list(ifelse(j<0, 0, j), 
       ifelse(j>0, 0, j))
})]

setnames(DT, paste(rep(vars, each=2), c("_pos", "_neg"), sep=""))

uj5u.com熱心網友回復：

另一個 dplyr 選項

library(data.table)
library(dplyr, warn.conflicts = F)
DT <- data.table(x = c(-1, -2, 1, 3),
                 z = c(-1, -1, -1, -1))

DT %>%
  mutate(across(everything(), list(
    pos = ~ if_else(. > 0, ., 0),
    neg = ~ if_else(. < 0, -., 0)
  )))
#>     x  z x_pos x_neg z_pos z_neg
#> 1: -1 -1     0     1     0     1
#> 2: -2 -1     0     2     0     1
#> 3:  1 -1     1     0     0     1
#> 4:  3 -1     3     0     0     1

^{由reprex 包(v2.0.1)于 2021 年 11 月 27 日創建}

如果您想使用此語法但在后臺執行 data.table 操作，則可以使用tidytable.

library(data.table)
library(tidytable, warn.conflicts = F)
DT <- data.table(x = c(-1, -2, 1, 3),
                 z = c(-1, -1, -1, -1))

DT %>%
  mutate.(across.(everything(), list(
    pos = ~ if_else(. > 0, ., 0),
    neg = ~ if_else(. < 0, -., 0)
  )))
#> # A tidytable: 4 × 6
#>       x     z x_pos x_neg z_pos z_neg
#>   <dbl> <dbl> <dbl> <dbl> <dbl> <dbl>
#> 1    -1    -1     0     1     0     1
#> 2    -2    -1     0     2     0     1
#> 3     1    -1     1     0     0     1
#> 4     3    -1     3     0     0     1

^{由reprex 包(v2.0.1)于 2021 年 11 月 27 日創建}

標籤：r dplyr 数据表 马格里特

上一篇：檢測字串型別并相應地創建新變數

下一篇：處理包含特殊字符的xts物件