How to get the nth element of each item of a list, which is itself a vector of unknown length

六月ゝ 毕业季﹏ 提交于 2019-12-30 09:42:17

问题


If we have a list, and each item can have different length. For example:

l <- list(c(1, 2), c(3, 4,5), c(5), c(6,7))

(In order to be clear, we will call objects in a list "items", and objects in the objects of list "elements".)

How can we extract, for example the first element of each item? Here, I want to extract:

1, 3, 5, 6

Then same question for the second element of each item:

2, 4, NA, 7

回答1:


We can create a function using sapply

fun1 <- function(lst, n){
         sapply(lst, `[`, n)
   }
fun1(l, 1)
#[1] 1 3 5 6

fun1(l, 2)
#[1]  2  4 NA  7



回答2:


data.table::transpose(l) will give you a list with vectors of all 1st elements, all 2nd elements, etc.

l <- list(1:2, 3:4, 5:7, 8:10)
b <- data.table::transpose(l)
b
# [[1]]
# [1] 1 3 5 8
# 
# [[2]]
# [1] 2 4 6 9
# 
# [[3]]
# [1] NA NA  7 10

If you don't want the NAs you can do lapply(b, function(x) x[!is.na(x)])



来源:https://stackoverflow.com/questions/43014782/how-to-get-the-nth-element-of-each-item-of-a-list-which-is-itself-a-vector-of-u

标签
易学教程内所有资源均来自网络或用户发布的内容,如有违反法律规定的内容欢迎反馈
该文章没有解决你所遇到的问题?点击提问,说说你的问题,让更多的人一起探讨吧!