2015-05-09 37 views
1

假设我有两个分类变量AB,并且都有三个级别,1, 2, 30.2,0.30.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)) 

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

+0

你要正确对待A和B作为连续计算相关性时?或者你想用什么样的相关度量来处理离散/分类数据?你是否假设序数据? – MrFlick

+0

只是假设皮尔逊的相关性。 –

+1

只需自己定义联合概率矩阵即可。从那里,抽样非常简单。 – Frank

回答

3

下面是一个例子概率矩阵。 (你自己将取决于您选择的型号。)

# 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),] 
2

下面的代码是this artical的等效R代码,用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 
+0

谢谢,@Jthorpe!但是我想这些变量是分类或序数。一些解决方案呢? –