arrays - 在 Julia 中检查随机样本组件的身份

Question

我在Julia中创建了一个混合的 k=3 简单正态分布，如下所示（Distributions.jl）：

mixm = MixtureModel(Normal[
     Normal(-2.0, 1.2),    
     Normal(0.0, 1.0),    
     Normal(2.0, 2.5)], 
     [0.2,0.5, 0.3])

然后我使用以下方法进行采样：

sample = rand(mixm, 10)

所以我们得到了属于 3 个法线中的一个的 10 个样本。这是一个 10 元素数组。如何找到每个元素来自哪个组件？

我认为这一定很容易，因为 Julia 随机选择两个法线之一来对数组中的元素 i 进行采样sample。

我想开发一个if循环，以便将每个组件的元素存储到两个单独的列表中。