如何在列表列表中对向量进行子集化

Question

我有以下列表列表：

foo <- list(a = list(x = 1:10, y = 11:25), b = list(x = 1:10, y = 100:110))

它看起来像这样：

> foo
$a
$a$x
 [1]  1  2  3  4  5  6  7  8  9 10

$a$y
 [1] 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25


$b
$b$x
 [1]  1  2  3  4  5  6  7  8  9 10

$b$y
 [1] 100 101 102 103 104 105 106 107 108 109 110

我要做的是只选择产生收益的列表的前3个：

$a
$a$x
 [1]  1  2  3  

$a$y
 [1] 11 12 13 


$b
$b$x
 [1]  1  2  3  

$b$y
 [1] 100 101 102

我该如何实现？

Answer 1

出现一个疯狂的?rapply：

rapply(foo, f=head, n=3, how="list")
#$a
#$a$x
#[1] 1 2 3
#
#$a$y
#[1] 11 12 13
#
#
#$b
#$b$x
#[1] 1 2 3
#
#$b$y
#[1] 100 101 102

这是一个r递归apply，它将遍历每个列表，直到找到一个非列表，然后应用f=联合。 how="list"参数只是确保您再次获得相同的list结构。

Answer 2

使用嵌套的lapply

lapply(foo, lapply, `[`, 1:3)

#$a
#$a$x
#[1] 1 2 3

#$a$y
#[1] 11 12 13


#$b
#$b$x
#[1] 1 2 3

#$b$y
#[1] 100 101 102

或使用head

lapply(foo, lapply, head, 3)

Answer 3

library(purrr)
foo %>% map_depth(2,head,3)
#$a
#$a$x
#[1] 1 2 3
#
#$a$y
#[1] 11 12 13
#
#
#$b
#$b$x
#[1] 1 2 3
#
#$b$y
#[1] 100 101 102