Rのリストに成分名が設定されているとき、それぞれの成分に格納されているデータフレームの一部を選択した上で、成分名をファイル名にして出力させる方法です。

Tidyverseのpurrrパッケージのmap2()を使いました。

目次

環境

  • Mac
  • R
  • R Studio
  • Tidyverseパッケージ

データの例

想定しているのは、たとえば以下のような構造のリスト。(有名な「iris」データで代用)

data(iris)
library(tidyverse) # install済みを想定

df_list_iris <- list()
df_list_iris[["setosa"]] <- dplyr::filter(iris, Species == "setosa")
df_list_iris[["versicolor"]] <- dplyr::filter(iris, Species == "versicolor")
df_list_iris[["virginica"]] <- dplyr::filter(iris, Species == "virginica")

これで以下のようなリストを作成できました。

str(df_list_iris)
List of 3
 $ setosa    :'data.frame':	50 obs. of  5 variables:
  ..$ Sepal.Length: num [1:50] 5.1 4.9 4.7 4.6 5 5.4 4.6 5 4.4 4.9 ...
  ..$ Sepal.Width : num [1:50] 3.5 3 3.2 3.1 3.6 3.9 3.4 3.4 2.9 3.1 ...
  ..$ Petal.Length: num [1:50] 1.4 1.4 1.3 1.5 1.4 1.7 1.4 1.5 1.4 1.5 ...
  ..$ Petal.Width : num [1:50] 0.2 0.2 0.2 0.2 0.2 0.4 0.3 0.2 0.2 0.1 ...
  ..$ Species     : Factor w/ 3 levels "setosa","versicolor",..: 1 1 1 1 1 1 1 1 1 1 ...
 $ versicolor:'data.frame':	50 obs. of  5 variables:
  ..$ Sepal.Length: num [1:50] 7 6.4 6.9 5.5 6.5 5.7 6.3 4.9 6.6 5.2 ...
  ..$ Sepal.Width : num [1:50] 3.2 3.2 3.1 2.3 2.8 2.8 3.3 2.4 2.9 2.7 ...
  ..$ Petal.Length: num [1:50] 4.7 4.5 4.9 4 4.6 4.5 4.7 3.3 4.6 3.9 ...
  ..$ Petal.Width : num [1:50] 1.4 1.5 1.5 1.3 1.5 1.3 1.6 1 1.3 1.4 ...
  ..$ Species     : Factor w/ 3 levels "setosa","versicolor",..: 2 2 2 2 2 2 2 2 2 2 ...
 $ virginica :'data.frame':	50 obs. of  5 variables:
  ..$ Sepal.Length: num [1:50] 6.3 5.8 7.1 6.3 6.5 7.6 4.9 7.3 6.7 7.2 ...
  ..$ Sepal.Width : num [1:50] 3.3 2.7 3 2.9 3 3 2.5 2.9 2.5 3.6 ...
  ..$ Petal.Length: num [1:50] 6 5.1 5.9 5.6 5.8 6.6 4.5 6.3 5.8 6.1 ...
  ..$ Petal.Width : num [1:50] 2.5 1.9 2.1 1.8 2.2 2.1 1.7 1.8 1.8 2.5 ...
  ..$ Species     : Factor w/ 3 levels "setosa","versicolor",..: 3 3 3 3 3 3 3 3 3 3 ...

リスト成分名をファイル名にしてデータを保存する

やりたかったのは、このようなリストがあるときに、

  • 個々のリスト成分下のデータフレームからSepal.LengthPetal.Length列を抜き出し、それをcsvファイルに出力する
  • そして出力先のファイル名を、setosa_Length.csvversicolor_Length.csvのように、リスト成分名から取ること でした。

最初は、

# NG例
df_list_iris %>%
    map(
      ~write_csv(.x[c("Sepal.Length","Petal.Length")], 
      file = paste0(names(.x),"_Length.csv"))
      )

のようにすればいけるのかと思ったのですが、map内ではリストの成分名をこの記述で取り出すことはできず、エラーが出ます。

思い通りの結果を得るには、map2を使い、リストとリストの成分名の2つを引数にする必要がありました。

# 成功例
map2(
  df_list_iris, #引数x(リスト)
  names(df_list_iris), #引数y(リストの成分名)
  ~write_csv(
    .x[c("Sepal.Length","Petal.Length")], #任意の列を指定
    file = paste0(.y, "_Length.csv") #ファイル名を指定
  )
)

これで出来ました。

参考情報

上で書いたプログラムに関連する内容が含まれているwebページです。