迭代并操作Sympy Matrices

问题描述 投票:0回答:1

我修改过的here代码块通过使用Sympy库的函数(XOR和AND)(下面的第一个代码块)来处理XOR'ing python列表。但是,我很难理解如何通过sympy矩阵进行迭代(下面的第二段代码)。

python列出的代码是:

from sympy import And, Xor
from sympy.logic import SOPform, simplify_logic
from sympy import symbols
def LogMatrixMult (A, B):
    rows_A = len(A)
    cols_A = len(A[0])
    rows_B = len(B)
    cols_B = len(B[0])

    if cols_A != rows_B:
      print ("Cannot multiply the two matrices. Incorrect dimensions.")
      return

    # Create the result matrix
    # Dimensions would be rows_A x cols_B
    C = [[0 for row in range(cols_B)] for col in range(rows_A)]
    for i in range(rows_A):
        for j in range(cols_B):
            for k in range(cols_A):
                # I can add Sympy's in simplify_logic(-)
                C[i][j] = Xor(C[i][j], And(A[i][k], B[k][j]))
    return C
b, c, d, e, f, w, x, y, z = symbols('b c d e f w x y z')
m1 = [[b,c,d,e]]
m2 = [[w,x],[x,z],[y,z],[z,w]]

result = simplify_logic(LogMatrixMult(m1, m2)[0][0])
print(result)

在下面的块中使用Sympy矩阵注意i,j,k和C,A,B定义是我试图修改使用迭代器,我不知道这是否需要或正确。

from sympy import And, Xor
from sympy.matrices import Matrix
from sympy.logic import SOPform, simplify_logic
from sympy import symbols, IndexedBase, Idx
from sympy import symbols

def LogMatrixMultArr (A, B):
    rows_A = A.rows
    cols_A = A.cols
    rows_B = B.rows
    cols_B = B.cols
    i,j,k = symbols('i j k', cls=Idx)
    C = IndexedBase('C')
    A = IndexedBase('A')
    B = IndexedBase('B')

    if cols_A != rows_B:
      print ("Cannot multiply the two matrices. Incorrect dimensions.")
      return

    # Create the result matrix
    # Dimensions would be rows_A x cols_B
    C = [[0 for row in range(cols_B)] for col in range(rows_A)]
    for i in range(rows_A):
        for j in range(cols_B):
            for k in range(cols_A):
                # I can add Sympy's in simplify_logic(-)
                C[i,j] = Xor(C[i,j], And(A[i,k], B[k,j]))
                # C[i][j] = Xor(C[i][j],And(A[i][k],B[k][j]))
    return C
b, c, d, e, f, w, x, y, z = symbols('b c d e f w x y z')
P = Matrix([w,x]).reshape(1,2)
Q = Matrix([y,z])
print(LogMatrixMultArr(P,Q))

我得到的错误是:TypeError:列表索引必须是整数或切片,而不是元组

C[i,j] = Xor(C[i,j], And(A[i,k], B[k,j]))

现在我相信我必须采用一些特殊的方式来进行迭代,但我仍然坚持如何让它在代码中运行 - 如果我甚至需要这种方法。

此外,如果有人知道如何使用XOR和And(非按位)而不是+和*运算符以更快的方式执行上述操作,请分享。

谢谢。

matrix sympy matrix-multiplication
1个回答
0
投票

我认为问题出在IndexedBase对象上。我不能胜任这些,但似乎你没有正确使用它们。如果你更换

    i,j,k = symbols('i j k', cls=Idx)
    C = IndexedBase('C')
    A = IndexedBase('A')
    B = IndexedBase('B')

通过

    C = zeros(rows_A, cols_B)

并删除C = [[0 for row in range(cols_B)] for col in range(rows_A)],然后它的工作原理。

© www.soinside.com 2019 - 2024. All rights reserved.