Fügen Sie in R data.table eine Spalte hinzu, deren Name auf einer Zeichenfolge basiert.

Question

Ich möchte eine Spalte zu einer data.table hinzufügen. Der Name dieser neuen Spalte muss jedoch aus einem Zeichenvektor extrahiert werden. Ich schreibe dies:

add_var=function(index){ 
    label=c("products","price") 
    var_name=label[index] 
    df=data.frame(x=c(1,2,5,9),y=c(5,2,6,7)) 
    dt=as.data.table(df) 
    dt[,(as.name(var_name)):=5] 
    return(dt) 
} 
new_ds=add_var(1) 

Und ich erwartete etwas wie

x y products 
1 5  5 
2 2  5 
5 6  5 
9 7  5 

Aber stattdessen bekam ich diese Fehlermeldung:

Error in `[.data.table`(dt, , `:=`((as.name(var_name)), 5)) : 
    LHS of := must be a symbol, or an atomic vector (column names or positions). 

Wer weiß, wie meine Funktion zu beheben, es zu machen Arbeit?

Answer 1

Sie müssen nur das:

label <- c("products","price") 
df <- data.frame(x=c(1,2,5,9),y=c(5,2,6,7)) 


setDT(df)[ , (label) := 5] 


#> x y label products price 
#> 1: 1 5  5  5  5 
#> 2: 2 2  5  5  5 
#> 3: 5 6  5  5  5 
#> 4: 9 7  5  5  5