我有一个包含数值和分类变量的数据集。数值变量的分布因每个类别而异。我想为每个分类变量绘制“密度图”,以便它们在视觉上低于整个密度图。
这与混合模型的组件类似,但没有计算混合模型(因为我已经知道分割数据的分类变量)。
如果我根据分类变量对ggplot进行分组,则四个密度中的每一个都是真实密度并整合为一个密度。
library(ggplot2)
ggplot(iris, aes(x = Sepal.Width)) + geom_density() + geom_density(aes(x = Sepal.Width, group = Species, colour = 'Species'))
我想要的是将每个类别的密度作为子密度(不整合到1)。类似于以下代码(我只对三种虹膜中的两种实施)
myIris <- as.data.table(iris)
# calculate density for entire dataset
dens_entire <- density(myIris[, Sepal.Width], cut = 0)
dens_e <- data.table(x = dens_entire[[1]], y = dens_entire[[2]])
# calculate density for dataset with setosa
dens_setosa <- density(myIris[Species == 'setosa', Sepal.Width], cut = 0)
dens_sa <- data.table(x = dens_setosa[[1]], y = dens_setosa[[2]])
# calculate density for dataset with versicolor
dens_versicolor <- density(myIris[Species == 'versicolor', Sepal.Width], cut = 0)
dens_v <- data.table(x = dens_versicolor[[1]], y = dens_versicolor[[2]])
# plot densities as mixture model
ggplot(dens_e, aes(x=x, y=y)) + geom_line() + geom_line(data = dens_sa, aes(x = x, y = y/2.5, colour = 'setosa')) +
geom_line(data = dens_v, aes(x = x, y = y/1.65, colour = 'versicolor'))
导致
上面我对数字进行了硬编码以减少y值。有没有办法用ggplot做到这一点?或计算它?
感谢您的想法。
答案 0 :(得分:1)
ggplot(iris, aes(x = Sepal.Width)) +
geom_density(aes(y = ..count..)) +
geom_density(aes(x = Sepal.Width, y = ..count..,
group = Species, colour = Species))
另一种选择可能是
ggplot(iris, aes(x = Sepal.Width)) +
geom_density(aes(y = ..density..)) +
geom_density(aes(x = Sepal.Width, y = ..density../3,
group = Species, colour = Species))