时间戳

首页 > 解决方案 > Postgres ARRAY 列类型到 R 中的 tbl 列表列，反之亦然

r - Postgres ARRAY 列类型到 R 中的 tbl 列表列，反之亦然

问题描述

假设我正在使用starwarsdplyr 包中的数据集，其中包含列表列（用于电影、车辆......）。为了简化，让我们只使用名称和电影数据：

library(dplyr)
ex_data <- starwars %>%
  select(name, films)

ex_data

# A tibble: 87 x 2
   name               films    
   <chr>              <list>   
 1 Luke Skywalker     <chr [5]>
 2 C-3PO              <chr [6]>
 3 R2-D2              <chr [7]>
 4 Darth Vader        <chr [4]>
 5 Leia Organa        <chr [5]>
 6 Owen Lars          <chr [3]>
 7 Beru Whitesun lars <chr [3]>
 8 R5-D4              <chr [1]>
 9 Biggs Darklighter  <chr [1]>
10 Obi-Wan Kenobi     <chr [6]>
# ... with 77 more rows

我想将此数据复制到 PostgreSQL 数据库，但我希望将电影列转换为 SQL 中的文本数组类型，如下所示：

starwars=# \d test
       Table "public.test"
   Column   |  Type  | Modifiers
------------+--------+-----------
 name       | text   | not null
 films      | text[] |

有什么简单的方法可以用dplyrand/or做到这一点dbplyr？

我也对相反的操作感兴趣，将一个具有数组类型列的表从 postgreSQL db 导入 R 并为数组获取一个列表类型列。

先感谢您！

标签： rpostgresqldplyr

解决方案

下面是穷人使用现有 R 数据库包和一些 SQL 代码的解决方法。我使用data.table包来处理数据（可以很容易地适应dplyr）。

# packages
require(data.table)
require(RPostgreSQL)

# data preparation
dt = dplyr::starwars
setDT(dt)
dt = dt[ , .SD, .SDcols = c('name', 'films') ]
# list conversion
dt2 = dt[ , .(films = unlist(films)), name ]

# database I/O
drv = dbDriver('PostgreSQL')
con = dbConnect(drv,
                dbname = dbname,
                user = user, password = password,
                host = host, port = port)
# write
dbWriteTable(con, name = 'test', value = dt2,
             overwrite = TRUE, row.names = FALSE)
# convert to array in database
q = "DROP TABLE IF EXISTS test2;
     CREATE TABLE test2 AS (
       SELECT name, array_agg(films) AS films
       FROM test
       GROUP BY name
     );"
dbSendQuery(con, q)

# read arrays
dt3 = dbReadTable(con, 'test')

# convert back to list columns
setDT(dt3)
dt4 = dt3[ , .(films = list(films)), name ]

dbDisconnect(con)
dbUnloadDriver(drv)

一些想法：如果RPostgreSQL或其他数据库包中存在这样的功能，那就太好了。它肯定不能很好地扩展。不幸的是，我缺乏 C 语言来帮助实现这一点。

推荐阅读