Двойной суммирования матричных элементов в Python

Question

Основываясь на упрощенном примере ниже

Я хотел бы в моем коде

from sympy import* 
import numpy as np 
init_printing() 

x, y = symbols('x, y') 

mat = Matrix([[x,1],[1,y]]) 

X = [1, 2, 3] 
Y = [[10, 20, 30], [40, 50, 60], [70, 80, 90]] 

заменить символическую x и y со значениями X и Y и, конечно, вычислить двойное суммирование заданной матрицы.

Я пытаюсь решить это, но у меня тяжелое время с заменой на каждом шагу. Любая помощь будет высоко оценена.

Answer 1

Вы импортировали как SymPy, так и NumPy, поэтому у вас есть выбор инструментов здесь. И для работы по объединению множества числовых матриц, numpy - правильный инструмент. Вот как суммирование происходит в NumPy:

sum([sum([np.array([[x,1], [1,y]]) for y in yr]) for x, yr in zip(X,Y)]) 

Здесь год стоит для ряда элементов Y. Внешняя сумма сверх я индекс, внутренняя закончилась у, хотя список понимание устраняет необходимость заклинания их.

Результатом является массив NumPy:

array([[ 18, 9], 
     [ 9, 450]]) 

но вы можете превратить его в матрицу SymPy просто поместив Matrix() вокруг него:

Matrix(sum([sum([np.array([[x,1], [1,y]]) for y in yr]) for x, yr in zip(X,Y)]))