Python循环实现Kronecker积计算λ_i、ρ_i矩阵求助

根据定义，λ_i是n个2×2矩阵的克罗内克积，其中第i个位置为σ_i，其余位置为σ₀；ρ_i则是从右往左数第i个位置为σ_i，其余位置为σ₀。以下是基于你初始代码的循环实现：

完整代码 #

import numpy as np

n = 3
σ_0 = np.eye(2)
σ_1 = np.array([[1, 0 ],[0, -1]])
σ_2 = np.array([[0, -1],[1, 0]])
σ = np.array([σ_0 , σ_1 , σ_2])

# 存储计算结果的列表
λ_matrices = []
ρ_matrices = []

# 循环计算每个i对应的λ_i和ρ_i
for i in range(1, n+1):
    # 计算λ_i：第i个位置用σ_i，其余用σ_0
    λ_i = σ_0
    for pos in range(1, n+1):
        current_mat = σ[i] if pos == i else σ[0]
        λ_i = np.kron(λ_i, current_mat)
    λ_matrices.append(λ_i)
    
    # 计算ρ_i：从右数第i个位置用σ_i，其余用σ_0
    ρ_i = σ_0
    target_pos = n - i + 1
    for pos in range(1, n+1):
        current_mat = σ[i] if pos == target_pos else σ[0]
        ρ_i = np.kron(ρ_i, current_mat)
    ρ_matrices.append(ρ_i)

# 示例：打印λ₁和ρ₁
print("λ₁矩阵：")
print(λ_matrices[0])
print("\nρ₁矩阵：")
print(ρ_matrices[0])

代码说明 #

1. λ_i的构建逻辑：
   • 外层循环遍历i从1到n，对应每个待计算的λ_i
   • 内层循环逐个确定每个位置的矩阵：当位置等于i时用σ_i，否则用σ₀，逐步做克罗内克积
2. ρ_i的构建逻辑：
   • 和λ_i类似，只是目标位置改为n - i + 1（即从右往左数第i个位置）
3. 结果存储：用列表保存所有i对应的矩阵，方便后续调用或验证

以n=3为例，计算出的矩阵对应：

• λ₁ = σ₁ ⊗ σ₀ ⊗ σ₀，λ₂ = σ₀ ⊗ σ₁ ⊗ σ₀，λ₃ = σ₀ ⊗ σ₀ ⊗ σ₁
• ρ₁ = σ₀ ⊗ σ₀ ⊗ σ₁，ρ₂ = σ₀ ⊗ σ₁ ⊗ σ₀，ρ₃ = σ₁ ⊗ σ₀ ⊗ σ₀

内容的提问来源于stack exchange，提问作者Sara Mansoori