如何将数据集转换为特定类型的矩阵? (RStudio)

问题描述 投票:0回答:1

我目前正在尝试转换此团队得分数据集:

      week AwayTeam AwayScore HomeTeam HomeScore
1    1        A         1        B         1
2    1        C         0        D         1
3    2        A         1        D         0
4    2        B         1        C         0
5    3        C         4        A         0
6    3        D         2        B         2

到看起来像这样的矩阵中:

A = matrix(c(1,-1,0,0,
             0,0,1,-1,
             1,0,0,-1,
             0,1,-1,0,
             -1,0,1,0,
             0,-1,0,1),6,4,byrow = TRUE)

矩阵A的行如下:“ 1”等于客队,“-1”等于主队,“ 0”是每一行的剩余空间。因此,矩阵A的第一行是A队(客队)vs B队(主场)。

[如果有人可以帮助将这些数据转换成这样的矩阵,将不胜感激。

r matrix
1个回答
0
投票

您可以尝试下面的代码

v <- LETTERS[1:4]
A <- matrix(0,nrow = nrow(df),ncol = 4)
A[cbind(1:nrow(df),match(df$AwayTeam,v))] <- 1
A[cbind(1:nrow(df),match(df$HomeTeam,v))] <- -1

诸如此类

> A
     [,1] [,2] [,3] [,4]
[1,]    1   -1    0    0
[2,]    0    0    1   -1
[3,]    1    0    0   -1
[4,]    0    1   -1    0
[5,]   -1    0    1    0
[6,]    0   -1    0    1

DATA

df <- structure(list(week = c(1L, 1L, 2L, 2L, 3L, 3L), AwayTeam = c("A", 
"C", "A", "B", "C", "D"), AwayScore = c(1L, 0L, 1L, 1L, 4L, 2L
), HomeTeam = c("B", "D", "D", "C", "A", "B"), HomeScore = c(1L, 
1L, 0L, 0L, 0L, 2L)), class = "data.frame", row.names = c("1", 
"2", "3", "4", "5", "6"))
© www.soinside.com 2019 - 2024. All rights reserved.