在Python中生成d阶（d为质数）全置换矩阵

嘿，我来帮你搞定生成d阶置换矩阵的问题！首先咱们先明确下核心定义——置换矩阵就是把d×d单位矩阵的行按1到d的某个全排列重新排列得到的矩阵，每行每列恰好只有一个1，其余元素全为0，这点你应该已经清楚了。

核心实现思路 #

其实生成所有d阶置换矩阵的逻辑很直接：

• 第一步：生成1到d的所有全排列，每个排列对应一个唯一的置换矩阵
• 第二步：对每个排列，构造对应的矩阵——对于第i行（编程里一般用0-based索引），在排列中第i个元素对应的列位置填入1，其余位置填0

举个简单的例子，当d=2时：

全排列有两个：[1,2] 和 [2,1]
对应置换矩阵分别是：

[[1, 0],
 [0, 1]]

和

[[0, 1],
 [1, 0]]

Python代码实现 #

这里用Python写一个通用的实现，用标准库的工具来高效生成全排列，避免自己写递归的冗余代码：

import itertools

def generate_permutation_matrices(d):
    # 生成1到d的所有全排列（这里用1-based的列号，方便对应数学定义）
    all_permutations = itertools.permutations(range(1, d + 1))
    permutation_matrices = []
    
    for perm in all_permutations:
        # 初始化一个d×d的零矩阵
        matrix = [[0 for _ in range(d)] for _ in range(d)]
        for row_idx, target_col in enumerate(perm):
            # 把1-based的列号转换成0-based索引，填入1
            matrix[row_idx][target_col - 1] = 1
        permutation_matrices.append(matrix)
    
    return permutation_matrices

# 测试d=3的情况（如果是质数比如d=5，直接改d的值就行）
if __name__ == "__main__":
    d = 3
    matrices = generate_permutation_matrices(d)
    for i, mat in enumerate(matrices, 1):
        print(f"第{i}个置换矩阵：")
        for row in mat:
            print(row)
        print("---")

关键细节说明 #

• 关于全排列：用itertools.permutations是因为它是Python标准库中经过优化的实现，比自己手写递归全排列更高效且不易出错
• 索引转换：代码里用了0-based的行/列索引（符合编程习惯），而数学定义里一般是1-based，所以通过target_col - 1完成转换
• 质数适配：你提到d是质数，但这个逻辑对任意正整数d都适用，完全不需要针对质数做特殊处理——置换矩阵的生成规则和d是否为质数无关
• 性能注意：当d较大时（比如d≥10），d!的数量会爆炸式增长，生成所有矩阵可能会占用大量内存，这时候要考虑按需生成而不是一次性存储所有矩阵

内容的提问来源于stack exchange，提问作者Patrick Lewis