python - 构 build 计矩阵 python

Question

假设我有一个 RxC contingency table 。这意味着有 R 行和 C 列。我想要一个维度为 RC × (R + C − 2) 的矩阵 X，其中包含行的 R − 1 “主效应”以及列的 C − 1 “主效应”。例如，如果您有 R=C=2 (R = [0, 1], C = [0, 1]) 和仅主效应，则有多种方法参数化设计矩阵 (X)，但以下是一种方法:

1 0
0 1
1 0
0 0

请注意，这是 4 x 2 = RC x (R + C - 2)，您省略了每行一个级别和每列一个级别。

对于 R 和 C 的任何值，即 R = 3、C = 4([0 1 2] 和 [0 1 2 3])，如何在 Python 中执行此操作？我只有 R 和 C 的值，但我可以使用它们来使用 np.arange(R) 和 np.arange(C) 构造数组。

Answer 1

以下应该有效:

R = 3
C = 2

ir = np.zeros((R, C))
ir[0, :] = 1
ir = ir.ravel()

mat = []
for i in range(R):
    mat.append(ir)
    ir = np.roll(ir, C)

ic = np.zeros((R, C))
ic[:, 0] = 1
ic = ic.ravel()

for i in range(C):
    mat.append(ic)
    ic = np.roll(ic, R)

mat = np.asarray(mat).T

结果是:

array([[ 1.,  0.,  0.,  1.,  0.],
       [ 1.,  0.,  0.,  0.,  1.],
       [ 0.,  1.,  0.,  1.,  0.],
       [ 0.,  1.,  0.,  0.,  1.],
       [ 0.,  0.,  1.,  1.,  0.],
       [ 0.,  0.,  1.,  0.,  1.]])

感谢大家的帮助!