簇向量在R中得到一个零一矩阵

Question

假设我有一个由任何聚类方法生成的聚类向量，如鸢尾花数据中的以下内容：

data(iris)
kmeans_res <- kmeans(x = iris[,c(1:4)], centers = 3)
kmeans_res$cluster

有没有一种有效的方法可以根据这个向量创建一个包含 0 和 1 的矩阵？

这个矩阵的行和列是数据集1到n的观测值。条目应该是一个，例如观测值 5 和 8 属于同一个集群，否则为零。

这个问题可以用一个循环来解决，但这看起来不太优雅。你能想到另一个解决方案吗？

Answer 1

您可以使用 outer.

M <- +(outer(v, v, `==`))
M[50:60, 50:60]
#       [,1] [,2] [,3] [,4] [,5] [,6] [,7] [,8] [,9] [,10] [,11]
#  [1,]    1    0    0    0    0    0    0    0    0     0     0
#  [2,]    0    1    1    0    1    1    1    1    1     1     1
#  [3,]    0    1    1    0    1    1    1    1    1     1     1
#  [4,]    0    0    0    1    0    0    0    0    0     0     0
#  [5,]    0    1    1    0    1    1    1    1    1     1     1
#  [6,]    0    1    1    0    1    1    1    1    1     1     1
#  [7,]    0    1    1    0    1    1    1    1    1     1     1
#  [8,]    0    1    1    0    1    1    1    1    1     1     1
#  [9,]    0    1    1    0    1    1    1    1    1     1     1
# [10,]    0    1    1    0    1    1    1    1    1     1     1
# [11,]    0    1    1    0    1    1    1    1    1     1     1

v[50:60]
# [1] 1 2 2 3 2 2 2 2 2 2 2

数据：

v <- kmeans_res$cluster

簇向量在R中得到一个零一矩阵

Cluster vector get a zero one matrix in R

r

cluster-analysis

vector

matrix