嵌套字典中包含的向量的相关性

import numpy as np
from scipy.stats import pearsonr

# 示例嵌套字典
nested_dict = {
    'group1': {'vector1': [1, 2, 3], 'vector2': [4, 5, 6]},
    'group2': {'vector3': [7, 8, 9], 'vector4': [10, 11, 12]}
}

# 提取所有向量
vectors = []
for group in nested_dict.values():
    for vector in group.values():
        vectors.append(vector)

# 计算向量之间的相关性
correlations = []
for i in range(len(vectors)):
    for j in range(i + 1, len(vectors)):
        corr, _ = pearsonr(vectors[i], vectors[j])
        correlations.append((i, j, corr))

print("向量相关性结果：")
for idx1, idx2, corr in correlations:
    print(f"向量 {idx1} 和 向量 {idx2} 的相关系数: {corr}")