如何在Numpy中无需转置生成边界为1的n阶矩阵？

你的代码逻辑没问题，但确实有不用转置的更直接写法，而且Numpy有更简洁的索引方式来处理列赋值，完全没必要转置矩阵。

无需转置的直接实现 #

直接用Numpy的列索引语法[:, 0]和[:, -1]就能修改第一列和最后一列，不用转置：

import numpy as np

def decorate_matrix(n: int):
    matrix = np.zeros((n, n), dtype=int)
    # 给第一行和最后一行赋值1
    matrix[0, :] = 1
    matrix[-1, :] = 1
    # 给第一列和最后一列赋值1
    matrix[:, 0] = 1
    matrix[:, -1] = 1
    return matrix

这里用-1代替n-1是Numpy的便捷索引，读起来更直观；加上dtype=int能让矩阵元素是整数（原代码默认生成浮点数类型的0和1，按需选择即可）。

更高效简洁的写法 #

还有一种更省心的方式：先创建全1矩阵，再把内部区域改成0，只需要两步操作：

import numpy as np

def decorate_matrix(n: int):
    matrix = np.ones((n, n), dtype=int)
    # 把中间的(n-2)×(n-2)区域设为0
    matrix[1:-1, 1:-1] = 0
    return matrix

这种方法代码更短，效率也更高，因为不需要多次单独赋值行和列。

原代码的优缺点分析 #

你的原代码逻辑是通顺的，但转置操作完全是多余的——直接操作列索引就能完成的事，没必要绕转置这一步。另外，原代码生成的是浮点数矩阵，如果需要整数类型，记得指定dtype参数。

从性能上看，Numpy的转置其实是返回原数组的视图（不会实际复制数据），所以开销不大，但写法上不如直接操作索引直观，也没必要多这一步。

总的来说，推荐第二种先全1再改内部的写法，简洁高效；如果偏好先全0再改边界的逻辑，直接用行+列索引的写法就行，不用转置。

内容的提问来源于stack exchange，提问作者Maxim Savoskin