diff --git a/sympyextensions.py b/sympyextensions.py index e89cf1374cec9259242736cbf8833b7cc266948c..d3b86fe588885b524bebe05c1053c63f3e98d31a 100644 --- a/sympyextensions.py +++ b/sympyextensions.py @@ -317,3 +317,12 @@ def countNumberOfOperations(term): visit(term) return result + + +def matrixFromColumnVectors(columnVectors): + """Creates a sympy matrix from column vectors. + :param columnVectors: nested sequence - i.e. a sequence of column vectors + """ + c = columnVectors + return sp.Matrix([list(c[i]) for i in range(len(c))]).transpose() +