r - r のベクトルから複雑な行列を作成する

Question

私は 4 つの母集団の標本サイズのベクトルを持っています: 3 2 1 4 合計すると 10 人になります。動的に（さまざまな状況で使用できるように）コードが必要です以下に示すように、残りの 0 など:

1 0 0 0
1 0 0 0
1 0 0 0
0 1 0 0
0 1 0 0
0 0 1 0
0 0 0 1
0 0 0 1
0 0 0 1
0 0 0 1

入力：

nSamples <- c(3,2,1,4)

Answer 1

これは、可能な 2 段階のアプローチです。

# create a matrix of zeros first
m <- matrix(0L, ncol = length(nSamples), nrow = sum(nSamples))
# then replace relevant 0s with 1s:
m[cbind(1:nrow(m), rep(seq_along(nSamples), nSamples))] <- 1L
m
#      [,1] [,2] [,3] [,4]
# [1,]    1    0    0    0
# [2,]    1    0    0    0
# [3,]    1    0    0    0
# [4,]    0    1    0    0
# [5,]    0    1    0    0
# [6,]    0    0    1    0
# [7,]    0    0    0    1
# [8,]    0    0    0    1
# [9,]    0    0    0    1
# [10,]   0    0    0    1

「トリック」は、もしそうなら、

rep(seq_along(nSamples), nSamples)
#[1] 1 1 1 2 2 3 4 4 4 4

Answer 2

を使用して計画マトリックスを作成できますmodel.matrix

f_nSamples <- factor(rep(nSamples, nSamples), orderd=TRUE)
model.matrix(~f_nSamples-1)
##    f_nSamples1 f_nSamples2 f_nSamples3 f_nSamples4
## 1            1           0           0           0
## 2            1           0           0           0
## 3            1           0           0           0
## 4            0           1           0           0
## 5            0           1           0           0
## 6            0           0           1           0
## 7            0           0           0           1
## 8            0           0           0           1
## 9            0           0           0           1
## 10           0           0           0           1
## attr(,"assign")
## [1] 1 1 1 1
## attr(,"contrasts")
## attr(,"contrasts")$f_nSamples
## [1] "contr.treatment"