python - 如何从Python中的协方差矩阵获取子协方差

Question

我在 python 中有一个协方差矩阵(作为 pandas DataFrame)，如下所示:

    a   b   c    
a   1   2   3    
b   2   10  4    
c   3   4   100

我想动态地仅选择矩阵协方差的子集。例如 A 和 C 的子集看起来像

    a   c
a   1   3    
c   3   100

有什么函数可以选择这个子集吗？

谢谢!

Answer 1

如果你的协方差矩阵是这样的 numpy 数组:

cov = np.array([[1, 2,  3],
                [2, 10, 4],
                [3, 4, 100]])

然后可以通过 advanced indexing 得到想要的子矩阵:

subset = [0, 2]  # a, c

cov[np.ix_(subset, subset)]

# array([[  1,   3],
#        [  3, 100]])

编辑:

如果您的协方差矩阵是 pandas DataFrame(例如，对于某些包含列 'a'、'b 的数据帧 df，通过 cov = df.cov() 获得', 'c', ...)，要获取 'a' 和 'c' 的子集，您可以执行以下操作:

cov.loc[['a','c'], ['a','c']]