Creating dummy variables in R data.table

by

This seems to do what you’re looking for:

inds <- unique(test$index)
test[, (inds) := lapply(inds, function(x) index == x)]

which gives

      index        var1     a     b     c     d     e     f     g     h     i     j
   1:     a  0.25331851  TRUE FALSE FALSE FALSE FALSE FALSE FALSE FALSE FALSE FALSE
   2:     b -0.02854676 FALSE  TRUE FALSE FALSE FALSE FALSE FALSE FALSE FALSE FALSE
   3:     c -0.04287046 FALSE FALSE  TRUE FALSE FALSE FALSE FALSE FALSE FALSE FALSE
   4:     d  1.36860228 FALSE FALSE FALSE  TRUE FALSE FALSE FALSE FALSE FALSE FALSE
   5:     e -0.22577099 FALSE FALSE FALSE FALSE  TRUE FALSE FALSE FALSE FALSE FALSE
  ---                                                                              
 996:     f -1.02040059 FALSE FALSE FALSE FALSE FALSE  TRUE FALSE FALSE FALSE FALSE
 997:     g -1.31345092 FALSE FALSE FALSE FALSE FALSE FALSE  TRUE FALSE FALSE FALSE
 998:     h -0.49448088 FALSE FALSE FALSE FALSE FALSE FALSE FALSE  TRUE FALSE FALSE
 999:     i  1.75175715 FALSE FALSE FALSE FALSE FALSE FALSE FALSE FALSE  TRUE FALSE
1000:     j  0.05576477 FALSE FALSE FALSE FALSE FALSE FALSE FALSE FALSE FALSE  TRUE

Here’s another way:

dcast(test, index + var1 ~ index, fun = length)
# or, if you want to preserve row order
dcast(test[, r := .I], r + index + var1 ~ index, fun = length)[, r := NULL]

And another:

rs = split(seq(nrow(test)), test$index)
test[, names(rs) := FALSE ]
for (n in names(rs)) set(test, i = rs[[n]], j = n, v = TRUE )

More Related Contents:

  1. Split a string column into several dummy variables
  2. Select / assign to data.table when variable names are stored in a character vector
  3. Assign multiple columns using := in data.table, by group
  4. Efficient way to rbind data.frames with different columns
  5. How to replace NA values in a table for selected columns
  6. data.table “key indices” or “group counter”
  7. Select subset of columns in data.table R [duplicate]
  8. How to extract the first n rows per group?
  9. R: data.table cross-join not working
  10. Why is allow.cartesian required at times when when joining data.tables with duplicate keys?
  11. Dynamically build call for lookup multiple columns
  12. Select values from different columns based on a variable containing column names [duplicate]
  13. Complex non-equi merge in R
  14. ‘Embedded nul in string’ error when importing csv with fread
  15. eval and quote in data.table
  16. Efficiently computing a linear combination of data.table columns
  17. R fuzzy string match to return specific column based on matched string
  18. When should I use the := operator in data.table?
  19. Subsetting data.table by 2nd column only of a 2 column key, using binary search not vector scan
  20. Using dynamic column names in `data.table`
  21. How do you delete a column by name in data.table?
  22. How does one change the levels of a factor column in a data.table
  23. When should I use setDT() instead of data.table() to create a data.table?
  24. DT[!(x == .)] and DT[x != .] treat NA in x inconsistently
  25. In R data.table, how do I pass variable parameters to an expression?
  26. Merging two sets of data by data.table roll=’nearest’ function
  27. Convert a data frame to a data.table without copy
  28. Row sums over columns with a certain pattern in their name
  29. paste two data.table columns
  30. Removing multiple columns from R data.table with parameter for columns to remove

Leave a Comment