将数据转换为数据框

时间:2016-04-22 23:13:45

标签: r dataframe matching game-theory

我是R的新手和一般的编程。我有这些数据:screenshot enter image description here

我有12个'ID'(研究对象),编号为1-12。 'types'列告诉每个ID的'type'。例如,'类型'列的前5个数字指的是前5个ID的'类型',即前5个ID的'类型'分别是3,3,2,1,1。

'对'列描述了ID如何配对在一起。例如,6与9配对; 4与7配对; 1与11配对,依此类推。

所以我需要帮助的是我想使用这些数据创建三列。

第一栏列出ID(1-12)
第二列:返回对的ID (例如1与11配对,因此第二列应为11表示ID 1)
第三栏:告诉该对的'type'(因此'类型'的11是3.第三列应该显示。

以下是所需输出格式的可视化:output format enter image description here

非常感谢任何帮助。 提前谢谢!

1 个答案:

答案 0 :(得分:1)

你可以通过一些聪明的索引来做到这一点。我输入原始数据作为类型的向量,以及对的向量列表:

# Enter the raw data
type <- c(3, 3, 2, 1, 1, 1, 2, 3, 1, 1, 3, 1)
pairs <- list(c(6, 9), c(4, 7), c(1, 11), c(3, 10), c(2, 12), c(5, 8))

由此,您可以通过按原始顺序堆叠所有对,然后以相反的顺序再次堆叠所需输出的前两列。 (我使用lapply(pairs, rev)颠倒了每一对,它将rev命令应用于列表中的每一对。)

# Create a 12 x 2 matrix of the pairs
pairs.mat <- do.call(rbind, c(pairs, lapply(pairs, rev)))
pairs.mat
#       [,1] [,2]
#  [1,]    6    9
#  [2,]    4    7
#  [3,]    1   11
#  [4,]    3   10
#  [5,]    2   12
#  [6,]    5    8
#  [7,]    9    6
#  [8,]    7    4
#  [9,]   11    1
# [10,]   10    3
# [11,]   12    2
# [12,]    8    5

为了清洁结果,我将其转换为data.frame

# Convert to data frame
colnames(pairs.mat) <- c("id", "match")
df <- as.data.frame(pairs.mat)

最后,我们可以通过从我们刚刚创建的data.frame匹配列的顺序获取type来获取type_match列。

# Add in the type_match column
df$type_match <- type[df$match]

# Print results in order
df[order(df$id), ]
#    id match type_match
# 3   1    11          3
# 5   2    12          1
# 4   3    10          1
# 2   4     7          2
# 6   5     8          3
# 1   6     9          1
# 8   7     4          1
# 12  8     5          1
# 7   9     6          1
# 10 10     3          2
# 9  11     1          3
# 11 12     2          3

这应该会给你想要的输出。