如何创建一个具有两列彼此特定相关性的数据集?我希望能够定义将要创建的值的数量,并指定输出应该具有的相关性。
问题与此类似:Generate numbers with specific correlation
其中一个答案是使用:
out <- mvrnorm(10, mu = c(0,0), Sigma = matrix(c(1,0.56,0.56,1),, ncol = 2),
mpirical = TRUE)
生成如下输出:
[,1] [,2]
[1,] -0.4152618 0.033311146
[2,] 0.7617759 -0.181852441
[3,] -1.6393045 -1.054752469
[4,] -1.7872420 -0.605214425
[5,] 0.9581152 2.511000955
[6,] 0.5048160 -0.278329145
[7,] 0.8656220 0.483521747
[8,] -0.1385699 0.017395548
[9,] 0.3261103 -0.932889606
[10,] 0.5639388 0.007808691
使用以下相关表cor(out):
[,1] [,2]
[1,] 1.00 0.56
[2,] 0.56 1.00
但我希望数据集包含更高,没有负数和更远的数字,例如:
x y
1 5 5
2 20 20
3 30 30
4 100 100
具有1:
的相关性 x y
x 1 1
y 1 1
在更远的地方,我的意思是“更多”随机而且价值更大,就像我上面的示例一样。
是否有(简单)方式存档这样的东西?
答案 0 :(得分:2)
相关性不受基础变量线性变换的影响。因此,获得所需内容的最直接方式可能是:
out <- as.data.frame(mvrnorm(10, mu = c(0,0),
Sigma = matrix(c(1,0.56,0.56,1),, ncol = 2),
empirical = TRUE))
out$V1.s <- (out$V1 - min(out$V1))*1000+10
out$V2.s <- (out$V2 - min(out$V2))*200+30
现在数据框out
已经“移位”了列V1.s
和V2.s
,这些列是非负数和“大”。您可以在我上面的代码中使用您想要的任何数字,而不是1000,10,200和30。相关性的答案仍为0.56。
> cor(out$V1.s, out$V2.s)
[1] 0.56
答案 1 :(得分:2)
谢谢Curt F.这对我生成一些模拟数据集很有帮助。我添加了一些选项来指定约。 X和Y所需的平均值和范围。它还提供输出,以便您可以检查斜率和截距以及绘制点和回归线。
library(MASS)
library(ggplot2)
# Desired correlation
d.cor <- 0.5
# Desired mean of X
d.mx <- 8
# Desired range of X
d.rangex <- 4
# Desired mean of Y
d.my <- 5
# Desired range of Y
d.rangey <- 2
# Calculations to create multipliation and addition factors for mean and range of X and Y
mx.factor <- d.rangex/6
addx.factor <- d.mx - (mx.factor*3)
my.factor <- d.rangey/6
addy.factor <- d.my - (my.factor*3)
# Generate data
out <- as.data.frame(mvrnorm(1000, mu = c(0,0),
Sigma = matrix(c(1,d.cor,d.cor,1), ncol = 2),
empirical = TRUE))
# Adjust so that values are positive and include factors to match desired means and ranges
out$V1.s <- (out$V1 - min(out$V1))*mx.factor + addx.factor
out$V2.s <- (out$V2 - min(out$V2))*my.factor + addy.factor
# Create liniear model to calculate intercept and slope
fit <- lm(out$V2.s ~ out$V1.s, data=out)
coef(fit)
# Plot scatterplot along with regression line
ggplot(out, aes(x=V1.s, y=V2.s)) + geom_point() + coord_fixed() + geom_smooth(method='lm')
# Produce summary table
summary(out)