R×purrr::lmap リストのリスト要素に関数を適用する

by
R言語で、リストのリスト要素に関数を適用する方法について解説します。ここではtidyverseパッケージに含まれているpurrrパッケージのlmap()を使用した方法についてお伝えします。

目次

lmap()の概要

lmap()は、リストのリスト要素に関数を適用するための関数です。lmap()は、map()と似ていますが、.x[[i]]をマッピングする代わりに、.x[i]をマッピングします。これにはいくつかの利点があります。

  • リストのみを受け取る関数を扱えるようになります。
  • .fがnames()のように、カプセル化されたリストの属性にアクセスできるようになります。
  • .fが、出力のサイズを変更することで、受け取ったリストよりも大きいリストまたは小さいリストを返すことができます。

lmap()の使い方


lmap(.x, .f, ...)

lmap()の引数

.x

リストまたはデータフレームを指定します。

.f

長さ1のリストを受け取り、任意の長さのリストを返す関数を指定します。

…

マッピングされた関数に渡される追加の引数を指定します。
現在、.fに追加の（定数）引数を渡すために…を使用することは、一般的に推奨されていません。代わりに、省略形の匿名関数を使用してください。


# 次の代わりに
x |> map(f, 1, 2, collapse = ",")
# 次のように実行してください。
x |> map(\(x) f(x, 1, 2, collapse = ","))

これにより、どの引数がどの関数に属しているかがわかりやすくなり、エラーメッセージもわかりやすくなります。

lmap()の戻り値

.xに一致するリストまたはデータフレームを返します。長さは保証されません。

準備

あらかじめ、tidyverseパッケージを読み込んでおきます。


library(tidyverse)

使用例

リストの要素名と値を同時に変更する

次は、3つのユーザー情報(user01, user02, user03)を持つリストを作成し、各要素名(user01, user02, user03)の末尾に「A」を追加し、nameを大文字に変換するコードです。


list(user01 = list(id = 100, name = "satou", age = 30),
     user02 = list(id = 101, name = "suzuki", age = 40),
     user03 = list(id = 102, name = "takahashi", age = 50)) |>
  lmap(function(x) {
    names(x) <- paste0(names(x), "A")
    x[[1]]$name <- str_to_upper(x[[1]]$name)
    x
  })


$user01A
$user01A$id
[1] 100

$user01A$name
[1] "SATOU"

$user01A$age
[1] 30


$user02A
$user02A$id
[1] 101

$user02A$name
[1] "SUZUKI"

$user02A$age
[1] 40


$user03A
$user03A$id
[1] 102

$user03A$name
[1] "TAKAHASHI"

$user03A$age
[1] 50