如何生成具有定义相关性的双变量分类变量？

Question

假设我有两个分类变量A和B并且两者都具有三个等级， 1, 2, 3与概率0.2 ， 0.3和0.5为每个级别。 如何生成具有定义的相关性0.3的A和B的随机双变量数据列表？ 我知道对于单变量A或B我们可以做到

A=sample(1:3, 100, T, prob=c(0.2,0.3,0.5))
B=sample(1:3, 100, T, prob=c(0.2,0.3,0.5))

我的问题是如何以cor(A,B)=0.3采样cbind(A,B) ？

Answer 1

这是一个示例概率矩阵。 （您自己的将取决于您选择的型号。）

# describe probabilities over the space by a matrix
set.seed(1)
nA <- 3
nB <- 3
probmat <- matrix({r<-runif(nA*nB);r/sum(r)},ncol=nB)
#            [,1]       [,2]      [,3]
# [1,] 0.04868724 0.16654119 0.1732284
# [2,] 0.06823764 0.03698311 0.1211728
# [3,] 0.10504609 0.16474081 0.1153628

这是从中抽取样本的一种方法：

# rearrange
probs  <- c(probmat)
events <- as.matrix(expand.grid(A=1:nA,B=1:nB))

# draw samples
nSamp  <- 100
samp   <- events[sample.int(nA*nB,nSamp,prob=probs,replace=TRUE),]

Answer 2

下面是等效R代码里面此ARTICAL用MATLAB例子。 您可以将pnorm()和cut()应用于列，以获取相关的离散随机变量

# parameters
nrows  <-  10

# The desired correlation matrix
(M  <-    matrix(c(1.0,0.6,0.3,
                  0.6,1.0,0.5,
                  0.3,0.5,1.0),byrow=T,ncol=3))


#>      [,1] [,2] [,3]
#> [1,]  1.0  0.6  0.3
#> [2,]  0.6  1.0  0.5
#> [3,]  0.3  0.5  1.0

(U = chol(M))
#>      [,1] [,2]      [,3]
#> [1,]    1  0.6 0.3000000
#> [2,]    0  0.8 0.4000000
#> [3,]    0  0.0 0.8660254 

# generate a random matrix where columns have the desired correlatoin structure
matrix(rnorm(nrows*ncol(M)),ncol=ncol(M))%*%U

#>   -0.4326   -0.4089    0.0505
#>   -1.6656   -0.4187   -1.3665
#>    0.1253   -0.3955    0.4209
#>    0.2877    1.9192    2.3656
#>   -1.1465   -0.7970   -0.9976
#>    1.1909    0.8057    1.1459
#>    1.1892    1.5669    1.8695
#>   -0.0376    0.0248   -1.3678
#>    0.3273    0.1199   -1.1880
#>    0.1746   -0.5611    0.2141


# check that this works
cor(matrix(rnorm(1000000*ncol(M)),ncol=ncol(M))%*%U)

#>           [,1]      [,2]      [,3]
#> [1,] 1.0000000 0.5988445 0.2987633
#> [2,] 0.5988445 1.0000000 0.4992603
#> [3,] 0.2987633 0.4992603 1.0000000