How to add index of a List item after melt() in R

Question

I am working with a list as follows:

  > l <- list(c(2,4,9), c(4,2,6,1))
  > m <- melt(l)
  > m 
  value L1
      2  1
      4  1
      9  1
      4  2
      2  2
      6  2
      1  2

i want to add index i for my resulting data frame m looks like this:

> m
i  value L1
1     2  1
2     4  1
3     9  1
1     4  2
2     2  2
3     6  2
4     1  2

i indicating 3 values belongs to first list element and 4 values belongs to the second list element.

How can i archive it please, can anyone help?

Do it after melting, `m$i <- with(m, ave(L1, L1, FUN=seq_along))` — user20650, Feb 05 '15 at 22:42
`setDT(m)[, i := seq_len(.N), L1]` with the `data table` pacage — David Arenburg, Feb 05 '15 at 23:03

akrun · Answer 1 · 2015-02-08T09:49:06.733

2

You could use splitstackshape

library(splitstackshape)
getanID(m, 'L1')[]
#   value L1 .id
#1:     2  1   1
#2:     4  1   2
#3:     9  1   3
#4:     4  2   1
#5:     2  2   2
#6:     6  2   3
#7:     1  2   4

Or using base R

transform(stack(setNames(l, seq_along(l))), .id= rapply(l, seq_along))

edited Feb 08 '15 at 09:49

answered Feb 06 '15 at 00:32

akrun

874,273
37
540
662

1

did not know about sequence, really good!! – Colonel Beauvel Feb 06 '15 at 08:36
ow, behind sequnce hides sapply ...! Thks for this discover! – Colonel Beauvel Feb 07 '15 at 19:09

score 2 · Answer 2 · answered Feb 07 '15 at 19:29

Just for completeness, some other options

data.table (which is basically what getanID is doing)

library(data.table)
setDT(m)[, i := seq_len(.N), L1]

dplyr

library(dplyr)
m %>%
  group_by(L1) %>%
  mutate(i = row_number())

Base R (from comments by @user20650)

transform(m, i = ave(L1, L1, FUN = seq_along))

score 1 · Answer 3 · edited Feb 07 '15 at 19:06

1

Less elegant than ave but does the work:

transform(m, i=unlist(sapply(rle(m$L1)$length, seq_len)))
#  value L1 i
#1     2  1 1
#2     4  1 2
#3     9  1 3
#4     4  2 1
#5     2  2 2
#6     6  2 3
#7     1  2 4

Or

 m$i <- sequence(rle(m$L1)$lengths)

edited Feb 07 '15 at 19:06

akrun

874,273
37
540
662

answered Feb 05 '15 at 22:57

Colonel Beauvel

30,423
11
47
87

How to add index of a List item after melt() in R

3 Answers3