python - 从列创建 sympy 矩阵

Question

我试图通过从现有矩阵中选择列来创建一个sympy矩阵(用于计算主次数)。目前我正在这样做:

>>> A = Matrix(3,5,[2,3,4,1,34,23,12,54,5,0,0,0,3,4,5])
>>> l = [A[:,i].T for i in [2,3,0]]
>>> M = Matrix(l).T
>>> M
Matrix([
[ 4, 1,  2],
[54, 5, 23],
[ 3, 4,  0]])

但这对我来说似乎很浪费(尤其是需要转置两次。我不知道这是否很耗时)。有没有更好的办法？如果我只需要行列式，是否有更好的方法？

Answer 1

您可以使用[2, 3, 0]作为索引。

>>> A = Matrix(3, 5, [2,3,4,1,34,23,12,54,5,0,0,0,3,4,5])
>>> A[:, [2,3,0]]
Matrix([
[ 4, 1,  2],
[54, 5, 23],
[ 3, 4,  0]])

<小时/>

低版本不支持使用list作为索引，可以使用Matrix.hstack :

>>> Matrix.hstack(*(A.col(i) for i in [2,3,0]))
Matrix([
[ 4, 1,  2],
[54, 5, 23],
[ 3, 4,  0]])

或Matrix.row_join:

>>> # from functools import reduce  # For Python 3.x
>>> reduce(Matrix.row_join, (A.col(i) for i in [2,3,0]), Matrix(3,0,[]))
Matrix([
[ 4, 1,  2],
[54, 5, 23],
[ 3, 4,  0]])