如何用 Python 制作矩阵:详细指南

liftword23小时前技术文章3

什么是矩阵?

在深入研究之前,先了解一下什么是矩阵:它是按行和列排列的数字矩形数组。在Python中,我们通常使用嵌套列表来表示矩阵,其中每个内部列表代表一行。

使用列表创建基本矩阵

让从创建矩阵的最简单方法开始:

# Method 1: Using list comprehension to create a 3x3 matrix of zeros
matrix = [[0 for _ in range(3)] for _ in range(3)]
print(matrix)  # Output: [[0, 0, 0], [0, 0, 0], [0, 0, 0]]

让我们一步步分解:
1. 外部 `for _ in range(3)` 创建三行2. 内部 `for _ in range(3)` 在每行中创建三列3.“0”是我们放入每个单元格中的值4. 我们使用 `_` 作为变量名,因为我们不使用循环变量 - 我们只是希望循环重复

这是创建具有特定值的矩阵的另一种方法:

# Method 2: Creating a matrix by explicitly listing values
matrix = [
    [1, 2, 3],  # First row
    [4, 5, 6],  # Second row
    [7, 8, 9]   # Third row
]

# Accessing elements
print(matrix[0][1])  # Output: 2 (row 0, column 1)

当我们访问“matrix[0][1]”时:
1. `matrix[0]` 获取第一行 `[1, 2, 3]` (记住 Python 使用基于 0 的索引)2. `[1]` 然后获取该行的第二个元素3. 所以`matrix[0][1]`得到值2

创建专门的矩阵

让看看如何创建具有特定模式的矩阵:

def create_identity_matrix(size):
    """
    Create an identity matrix of given size.
    An identity matrix has 1s on the diagonal and 0s everywhere else.
    """
    # This creates a size x size matrix
    return [[1 if i == j else 0 for j in range(size)] 
            for i in range(size)]

def create_triangular_matrix(size, upper=True):
    """
    Create an upper or lower triangular matrix of given size.
    Upper triangular: elements below diagonal are False
    Lower triangular: elements above diagonal are False
    """
    if upper:
        # For upper triangular, we set True when column >= row
        return [[j >= i for j in range(size)] 
                for i in range(size)]
    # For lower triangular, we set True when column <= row
    return [[j <= i for j in range(size)] 
            for i in range(size)]

# Let's see how these work
identity = create_identity_matrix(3)
print("Identity Matrix:")
for row in identity:
    print(row)
# Output:
# [1, 0, 0]
# [0, 1, 0]
# [0, 0, 1]

让我们详细检查一下“create_identity_matrix”:
1. 外循环“for i in range(size)”创建每一行2. 内部循环“for j in range(size)”创建每一列3. `1 if i == j else 0` 意思是:— 当行数等于列数(对角线)时,输入 1— 其他地方都放 04. 这给了我们独特的单位矩阵模式,对角线上有 1

对于三角矩阵:
1. 上三角中的`j >= i` 表示:— 当列号大于或等于行号时,置 True— 这将创建上三角图案,其中上面的所有内容(包括对角线)均为 True2. 下三角中的“j <= i”意味着:— 当列号小于或等于行号时,置 True— 这将创建下三角模式,其中以下所有内容(包括对角线)均为 True

矩阵运算

以下是执行基本矩阵运算的方法:

def matrix_addition(matrix1, matrix2):
    """
    Add two matrices of the same size.
    Returns a new matrix where each element is the sum of 
    corresponding elements from the input matrices.
    """
    # First, verify matrices have the same dimensions
    if len(matrix1) != len(matrix2) or len(matrix1[0]) != len(matrix2[0]):
        raise ValueError("Matrices must have the same dimensions")
    
    # Get dimensions
    rows = len(matrix1)
    cols = len(matrix1[0])
    
    # Create empty result matrix
    result = [[0 for _ in range(cols)] for _ in range(rows)]
    
    # Add corresponding elements
    for i in range(rows):
        for j in range(cols):
            result[i][j] = matrix1[i][j] + matrix2[i][j]
    
    return result

# Let's break down what happens with an example:
matrix1 = [[1, 2], [3, 4]]
matrix2 = [[5, 6], [7, 8]]

sum_matrix = matrix_addition(matrix1, matrix2)
print("Matrix Addition Result:")
for row in sum_matrix:
    print(row)
# Output:
# [6, 8]    # (1+5, 2+6)
# [10, 12]  # (3+7, 4+8)

让我们逐步检查“matrix_addition”:
1. 首先,我们检查矩阵的大小是否相同:— `len(matrix1)` 获取行数— `len(matrix1[0])` 获取列数— 如果其中一个不匹配matrix2,我们就无法将它们相加

2. 我们创建一个相同大小的结果矩阵:
— `[[0 for _ in range(cols)] for _ in range(rows)]` 创建一个用零填充的矩阵— 这为我们提供了一个存储结果的地方

3. 嵌套循环:
— 外循环“for i in range(rows)”遍历每一行— 内部循环 `for j in range(cols)` 遍历每一列— 在每个位置“[i][j]”,我们添加两个矩阵中的相应元素

4. 当我们添加 [[1, 2], [3, 4]] 和 [[5, 6], [7, 8]] 时:
— 位置 [0][0]:1 + 5 = 6— 位置 [0][1]:2 + 6 = 8— 位置 [1][0]:3 + 7 = 10— 位置 [1][1]:4 + 8 = 12

相关文章

面试必考的「矩阵快速幂」考点汇总

设想这样一个场景,面试官给了你一道算法题,你很快确定这是一道递推问题,并给出了 O(n) 的解题方法,然而面试官却继续问:“还能继续优化吗?”这样类似的场景并不少见,因为算法不仅追求「正确」,还追求「...

Python NumPy 进阶教程(中篇)

本篇(This intermediate tutorial)将进一步(deeper into)探讨——广播机制(Broadcasting)、统计函数(statistical functions)、线...

11-Python 库将数百行代码浓缩为一行

#寻找热爱表达的你#在今天的文章中,我们选取了 11 个 Python 库,测试了它们在效率和优化方面的流行说法是否真实。这些库以将数百行代码简化为几行甚至一行而闻名! 它们不仅让你的项目更加流畅,...

傻傻分不清楚的点积与矩阵乘法 Part3

作者:Minkyung Kang译者:知源觅流原文链接:https://github.com/mkang32/python-basics/blob/master/numpy/dot_vs_multip...

傻傻分不清楚的点积与矩阵乘法 Part2

作者:Minkyung Kang译者:知源觅流原文链接:https://github.com/mkang32/python-basics/blob/master/numpy/dot_vs_multip...