python - python中的列矩阵表示

Question

如何在 python 中表示列矩阵和行矩阵？

A =[1,2,3,4]

和

 1
 2
 3
 4

Answer 1

矩阵是二维结构。在纯 Python 中，矩阵最自然的表示是列表的列表。

因此，您可以将行矩阵编写为：

[[1, 2, 3, 4]]

并将列矩阵写为：

[[1],
 [2],
 [3],
 [4]]

这也很好地扩展到mxn矩阵：

[[10, 20],
 [30, 40],
 [50, 60]]

有关如何在纯 Python 中开发完整矩阵包的示例，请参见matfunc.py 。它的文档在这里。

这是一个使用列表列表表示在普通 python 中进行矩阵乘法的有效示例：

>>> from pprint import pprint
>>> def mmul(A, B):
        nr_a, nc_a = len(A), len(A[0])
        nr_b, nc_b = len(B), len(B[0])
        if nc_a != nr_b:
            raise ValueError('Mismatched rows and columns')
        return [[sum(A[i][k] * B[k][j] for k in range(nc_a))
                 for j in range(nc_b)] for i in range(nr_a)]

>>> A = [[1, 2, 3, 4]]
>>> B = [[1],
         [2],
         [3],
         [4]]

>>> pprint(mmul(A, B))
[[30]]

>>> pprint(mmul(B, A), width=20)
[[1, 2, 3, 4],
 [2, 4, 6, 8],
 [3, 6, 9, 12],
 [4, 8, 12, 16]]

正如另一位受访者所提到的，如果您认真对待矩阵运算，那么您应该安装直接支持许多矩阵运算的numpy ：

Answer 2

这是我对矩阵函数的实现，它采用矩阵的行数、列数和起始值

def matrix(rows, cols, start=0):
    return [[c + start + r * cols for c in range(cols)] for r in range(rows)]

用法：

>>> m = matrix(5, 1)
>>> m
[[0], [1], [2], [3], [4]]
>>> m = matrix(3, 3, 10)
>>> m
[[10, 11, 12], [13, 14, 15], [16, 17, 18]]