Cách thay thế các giá trị NA trong bảng cho các cột đã chọn

Có rất nhiều bài viết về việc thay thế các giá trị NA. Tôi biết rằng người ta có thể thay thế NAs trong bảng / khung sau bằng những thứ sau:

x[is.na(x)]<-0

Nhưng, điều gì sẽ xảy ra nếu tôi muốn giới hạn nó ở một số cột nhất định? Hãy để tôi chỉ cho bạn một ví dụ.

Đầu tiên, hãy bắt đầu với một tập dữ liệu.

set.seed(1234)
x <- data.frame(a=sample(c(1,2,NA), 10, replace=T),
                b=sample(c(1,2,NA), 10, replace=T), 
                c=sample(c(1:5,NA), 10, replace=T))

Cái nào mang lại:

    a  b  c
1   1 NA  2
2   2  2  2
3   2  1  1
4   2 NA  1
5  NA  1  2
6   2 NA  5
7   1  1  4
8   1  1 NA
9   2  1  5
10  2  1  1

Ok, vì vậy tôi chỉ muốn giới hạn sự thay thế cho cột 'a' và 'b'. Cố gắng của tôi là:

x[is.na(x), 1:2]<-0

và:

x[is.na(x[1:2])]<-0

Cái nào không hoạt động.

Nỗ lực data.table của tôi, trong đó y<-data.table(x), rõ ràng là không bao giờ hoạt động:

y[is.na(y[,list(a,b)]), ]

Tôi muốn chuyển các cột bên trong đối số is.na nhưng điều đó rõ ràng sẽ không hoạt động.

Tôi muốn thực hiện việc này trong data.frame và data.table. Mục tiêu cuối cùng của tôi là mã hóa lại 1: 2 thành 0: 1 trong 'a' và 'b' trong khi giữ nguyên 'c' vì nó không phải là một biến logic. Tôi có một loạt các cột nên tôi không muốn làm từng cột một. Và, tôi chỉ muốn biết cách làm điều này.

Bạn có đề nghị nào không?

Bạn có thể làm:

x[, 1:2][is.na(x[, 1:2])] <- 0

hoặc tốt hơn (IMHO), hãy sử dụng các tên biến:

x[c("a", "b")][is.na(x[c("a", "b")])] <- 0

Trong cả hai trường hợp, 1:2hoặc c("a", "b")có thể được thay thế bằng một vectơ được xác định trước.

Chỉnh sửa 2020-06-15

Kể từ ngày data.table1.12.4 (tháng 10 năm 2019), data.tablecó hai chức năng để hỗ trợ điều này: nafillvà setnafill.

nafill hoạt động trên các cột:

cols = c('a', 'b')
y[ , (cols) := lapply(.SD, nafill, fill=0), .SDcols = cols]

setnafill hoạt động trên các bảng (thay thế xảy ra theo tham chiếu / tại chỗ)

setnafill(y, cols=cols, fill=0)
# print y to show the effect
y[]

Điều này cũng sẽ hiệu quả hơn các tùy chọn khác; thấy ?nafillđể biết thêm, cuối cùng theo dõi bằng-thực-forward (LOCF) và (NOCB) phiên bản của lạc hậu tiếp theo quan sát-thực- NAsự đổ tội cho chuỗi thời gian.

Điều này sẽ hoạt động cho data.tablephiên bản của bạn :

for (col in c("a", "b")) y[is.na(get(col)), (col) := 0]

Ngoài ra, như David Arenburg chỉ ra bên dưới, bạn có thể sử dụng set(lợi ích phụ - bạn có thể sử dụng trên data.framehoặc data.table):

for (col in 1:2) set(x, which(is.na(x[[col]])), col, 0)

Dựa trên tidyr::replace_na()câu trả lời của @Robert McDonald , đây là một số dplyrtùy chọn để kiểm soát cột NAnào được thay thế:

library(tidyverse)

# by column type:
x %>%
  mutate_if(is.numeric, ~replace_na(., 0))

# select columns defined in vars(col1, col2, ...):
x %>%
  mutate_at(vars(a, b, c), ~replace_na(., 0))

# all columns:
x %>%
  mutate_all(~replace_na(., 0))

Điều này bây giờ là tầm thường trong ngăn nắp với Replace_na (). Hàm dường như hoạt động đối với data.tables cũng như data.frames:

tidyr::replace_na(x, list(a=0, b=0))

Không chắc liệu điều này có ngắn gọn hơn không, nhưng hàm này cũng sẽ tìm và cho phép thay thế các NA (hoặc bất kỳ giá trị nào bạn thích) trong các cột đã chọn của data.table:

update.mat <- function(dt, cols, criteria) {
  require(data.table)
  x <- as.data.frame(which(criteria==TRUE, arr.ind = TRUE))
  y <- as.matrix(subset(x, x$col %in% which((names(dt) %in% cols), arr.ind = TRUE)))
  y
}

Để áp dụng nó:

y[update.mat(y, c("a", "b"), is.na(y))] <- 0

Hàm tạo một ma trận gồm các cột và hàng đã chọn (tọa độ ô) đáp ứng tiêu chí đầu vào (trong trường hợp này là .na == TRUE).

Chúng tôi có thể giải quyết nó theo data.tablecách với tidyr::repalce_nachức năng vàlapply

library(data.table)
library(tidyr)
setDT(df)
df[,c("a","b","c"):=lapply(.SD,function(x) replace_na(x,0)),.SDcols=c("a","b","c")]

Bằng cách này, chúng ta cũng có thể giải quyết các cột dán với NAchuỗi. Đầu tiên, chúng tôi replace_na(x,""), sau đó chúng tôi có thể sử dụng stringr::str_cđể kết hợp các cột!

Đối với một cột cụ thể, có một giải pháp thay thế bằng sapply

DF <- data.frame(A = letters[1:5],
             B = letters[6:10],
             C = c(2, 5, NA, 8, NA))

DF_NEW <- sapply(seq(1, nrow(DF)),
                    function(i) ifelse(is.na(DF[i,3]) ==
                                       TRUE,
                                       0,
                                       DF[i,3]))

DF[,3] <- DF_NEW
DF

nó khá tiện dụng với {data.table} và {stringr}

library(data.table)
library(stringr)

x[, lapply(.SD, function(xx) {str_replace_na(xx, 0)})]

FYI

Bắt đầu từ data.table y, bạn có thể viết:
y[, (cols):=lapply(.SD, function(i){i[is.na(i)] <- 0; i}), .SDcols = cols]
Đừng quên library(data.table)trước khi tạo yvà chạy lệnh này.

Cái này làm việc tốt cho tôi

DataTable DT = new DataTable();

DT = DT.AsEnumerable().Select(R =>
{
      R["Campo1"] = valor;
      return (R);
}).ToArray().CopyToDataTable();