CODEWITHSUNDEEP
radjacency-matrix
Thomas
Thomas

Reputation: 1444

Create adjacency matrix based on binary variable

I have a data.frame with two columns. One indicating objects, the other one indicates if the objects are within some group. How can I create an adjacency matrix without writing a loop?

data("mtcars")
testdf <- data.frame(names=as.character(row.names(mtcars)[1:5]),
                 ingroup=0)

set.seed(123)
testdf$ingroup[testdf$names%in%row.names(mtcars)[sample(length(row.names(mtcars)[1:5]), 2)]] <- 1

testdf
              names ingroup
1         Mazda RX4       0
2     Mazda RX4 Wag       1
3        Datsun 710       0
4    Hornet 4 Drive       1
5 Hornet Sportabout       0

EDIT: The desired output should be a matrix indicating if objects are within a certain group:

data.frame(x1=c(0,0,0,0,0),
       x2=c(0,1,0,0,1),
       x3=c(0,0,0,0,0),
       x4=c(0,0,0,0,0),
       x5=c(0,1,0,0,1))
colnames(output) <- c("Datsun 710", "Hornet 4 Drive", "Hornet Sportabout", "Mazda RX4","Mazda RX4 Wag")
rownames(output) <- c("Datsun 710", "Hornet 4 Drive", "Hornet Sportabout", "Mazda RX4","Mazda RX4 Wag")

output
                  Datsun 710 Hornet 4 Drive Hornet Sportabout Mazda RX4 Mazda RX4 Wag
Datsun 710                 0              0                 0         0             0
Hornet 4 Drive             0              1                 0         0             1
Hornet Sportabout          0              0                 0         0             0
Mazda RX4                  0              0                 0         0             0
Mazda RX4 Wag              0              1                 0         0             1

Thanks, Thomas

Upvotes: 0

Views: 326

Answers (1)

emilliman5
emilliman5

Reputation: 5966

You just take the cross product of the ingroup variable:

res <- tcrossprod(testdf$ingroup)
rownames(res) <- testdf$names
colnames(res) <- testdf$names
res

# Mazda RX4 Mazda RX4 Wag Datsun 710 Hornet 4 Drive Hornet Sportabout
# Mazda RX4                 0             0          0              0                 0
# Mazda RX4 Wag             0             1          0              1                 0
# Datsun 710                0             0          0              0                 0
# Hornet 4 Drive            0             1          0              1                 0
# Hornet Sportabout         0             0          0              0                 0

Upvotes: 0

Related Questions