如何在 R 中将二维数据框“展平"或“折叠"为一维数据框? [英] How to “flatten” or “collapse” a 2D data frame into a 1D data frame in R?

问题描述

我在 R 中的 data.frame 中有一个带有距离的二维表(从 csv 导入):

I have a two dimensional table with distances in a data.frame in R (imported from csv):

           CP000036   CP001063      CP001368
CP000036      0           a            b
CP001063      a           0            c
CP001368      b           c            0

我想压平"它.我在第一个列中有一个轴的值​​，在第二个列中有其他轴的值，然后在第三列中有距离:

I'd like to "flatten" it. that I have one axes's value in the first col, and the other axes's value in the second col, and then the distance in the third col:

Genome1      Genome2       Dist
CP000036     CP001063       a
CP000036     CP001368       b
CP001063     CP001368       c

以上是理想的，但如果重复输入矩阵中的每个单元格都有自己的行，那就完全没问题了:

Above is ideal, but it would be completely fine to have repetition such that each cell in the input matrix has it's own row:

Genome1      Genome2       Dist
CP000036     CP000036       0
CP000036     CP001063       a
CP000036     CP001368       b
CP001063     CP000036       a
CP001063     CP001063       0
CP001063     CP001368       c
CP001368     CP000036       b
CP001368     CP001063       c
CP001368     CP001368       0

这是一个 3x3 矩阵示例，但我的数据集 I 大得多(大约 2000x2000).我会在 Excel 中执行此操作，但我需要约 300 万行输出，而 Excel 的最大值为约 100 万行.

Here is an example 3x3 matrix, but my dataset I is much larger (about 2000x2000). I would do this in Excel, but I need ~3 million rows for the output, whereas Excel's maximum is ~1 million.

这个问题非常类似于如何将 2D Excel 表格展平"或折叠"为 1D?"1

This question is very similar to "How to "flatten" or "collapse" a 2D Excel table into 1D?"1

推荐答案

所以这是一个使用 reshape2 包中的 melt 的解决方案:

So this is one solution using melt from the package reshape2:

dm <- 
  data.frame( CP000036 = c( "0", "a", "b" ),
              CP001063 = c( "a", "0", "c" ),
              CP001368 = c( "b", "c", "0" ),
              stringsAsFactors = FALSE,
              row.names = c( "CP000036", "CP001063", "CP001368" ) )

# assuming the distance follows a metric we avoid everything below and on the diagonal
dm[ lower.tri( dm, diag = TRUE ) ]  <- NA
dm$Genome1 <- rownames( dm )

# finally melt and avoid the entries below the diagonal with na.rm = TRUE
library(reshape2) 
dm.molten <- melt( dm, na.rm= TRUE, id.vars="Genome1",
                   value.name="Dist", variable.name="Genome2" )

print( dm.molten )
   Genome1  Genome2 Dist
4 CP000036 CP001063    a
7 CP000036 CP001368    b
8 CP001063 CP001368    c

可能有更高效的解决方案，但我喜欢这个解决方案，因为它简单明了.

Probably there are more performant solutions but I like this one because it's plain and simple.

这篇关于如何在 R 中将二维数据框“展平"或“折叠"为一维数据框?的文章就介绍到这了，希望我们推荐的答案对大家有所帮助，也希望大家多多支持IT屋！