Assign multiple columns using := in data.table, by group

by

This now works in v1.8.3 on R-Forge. Thanks for highlighting it!

x <- data.table(a = 1:3, b = 1:6) 
f <- function(x) {list("hi", "hello")} 
x[ , c("col1", "col2") := f(), by = a][]
#    a b col1  col2
# 1: 1 1   hi hello
# 2: 2 2   hi hello
# 3: 3 3   hi hello
# 4: 1 4   hi hello
# 5: 2 5   hi hello
# 6: 3 6   hi hello

x[ , c("mean", "sum") := list(mean(b), sum(b)), by = a][]
#    a b col1  col2 mean sum
# 1: 1 1   hi hello  2.5   5
# 2: 2 2   hi hello  3.5   7
# 3: 3 3   hi hello  4.5   9
# 4: 1 4   hi hello  2.5   5
# 5: 2 5   hi hello  3.5   7
# 6: 3 6   hi hello  4.5   9 

mynames = c("Name1", "Longer%")
x[ , (mynames) := list(mean(b) * 4, sum(b) * 3), by = a]
#     a b col1  col2 mean sum Name1 Longer%
# 1: 1 1   hi hello  2.5   5    10      15
# 2: 2 2   hi hello  3.5   7    14      21
# 3: 3 3   hi hello  4.5   9    18      27
# 4: 1 4   hi hello  2.5   5    10      15
# 5: 2 5   hi hello  3.5   7    14      21
# 6: 3 6   hi hello  4.5   9    18      27


x[ , get("mynames") := list(mean(b) * 4, sum(b) * 3), by = a][]  # same
#    a b col1  col2 mean sum Name1 Longer%
# 1: 1 1   hi hello  2.5   5    10      15
# 2: 2 2   hi hello  3.5   7    14      21
# 3: 3 3   hi hello  4.5   9    18      27
# 4: 1 4   hi hello  2.5   5    10      15
# 5: 2 5   hi hello  3.5   7    14      21
# 6: 3 6   hi hello  4.5   9    18      27

x[ , eval(mynames) := list(mean(b) * 4, sum(b) * 3), by = a][]   # same
#    a b col1  col2 mean sum Name1 Longer%
# 1: 1 1   hi hello  2.5   5    10      15
# 2: 2 2   hi hello  3.5   7    14      21
# 3: 3 3   hi hello  4.5   9    18      27
# 4: 1 4   hi hello  2.5   5    10      15
# 5: 2 5   hi hello  3.5   7    14      21
# 6: 3 6   hi hello  4.5   9    18      27

Older version using the with argument (we discourage this argument when possible):

x[ , mynames := list(mean(b) * 4, sum(b) * 3), by = a, with = FALSE][] # same
#    a b col1  col2 mean sum Name1 Longer%
# 1: 1 1   hi hello  2.5   5    10      15
# 2: 2 2   hi hello  3.5   7    14      21
# 3: 3 3   hi hello  4.5   9    18      27
# 4: 1 4   hi hello  2.5   5    10      15
# 5: 2 5   hi hello  3.5   7    14      21
# 6: 3 6   hi hello  4.5   9    18      27

More Related Contents:

  1. Aggregate / summarize multiple variables per group (e.g. sum, mean)
  2. Fastest way to replace NAs in a large data.table
  3. Fastest way to add rows for missing time steps?
  4. How to replace NA values in a table for selected columns
  5. Why is plyr so slow?
  6. Merge rows in a dataframe where the rows are disjoint and contain NAs
  7. Replace logical values (TRUE / FALSE) with numeric (1 / 0)
  8. When should I use the := operator in data.table?
  9. Cleaning `Inf` values from an R dataframe
  10. write.csv for large data.table
  11. DT[!(x == .)] and DT[x != .] treat NA in x inconsistently
  12. Merge overlapping ranges into unique groups, in dataframe
  13. Convert a data frame to a data.table without copy
  14. What you can do with a data.frame that you can’t with a data.table?
  15. how to create a dataframe in R from ” Min. 1stQu Median Mean 3rdQu Max. NA’s”? [closed]
  16. count number of rows in a data frame in R based on group [duplicate]
  17. How to plot all the columns of a data frame in R
  18. R Apply() function on specific dataframe columns
  19. Count NAs per row in dataframe [duplicate]
  20. Add (insert) a column between two columns in a data.frame
  21. Add a prefix to column names
  22. Create a Data Frame of Unequal Lengths
  23. apply a function over groups of columns
  24. No RTools compatible with R version 3.5.0 was found
  25. How to sort all dataframes in a list of dataframes on the same column?
  26. can the value.var in dcast be a list or have multiple value variables?
  27. Finding non-numeric data in a data frame or vector
  28. Find columns with all missing values
  29. How to transpose a dataframe in tidyverse?
  30. Subset data.table by logical column

Leave a Comment