Как векторизовать scipy.integrate.quad для вычисления поэлементного интеграла матрицы ⇐ Python

[Anonymous]

Я хочу интегрировать матрицу так, чтобы каждый элемент выходной матрицы был целым числом соответствующего элемента подынтегральной матрицы. Фрагменты кода приведены ниже:

Код: Выделить всё

import numpy as np
from scipy.integrate import quad
N=3
A = np.random.rand(N,N)
evs = np.linalg.eigvals(A)
evs = -np.sort(-evs)
Anew = A/(evs[0]+1) - np.eye(N)
B = np.eye(N)

def integrand(t,A,B):
prod = np.multiply(sp.linalg.expm(A*t),B)
return np.multiply(prod,prod.T)

This gives a square matrix with each element a function of t. I use the following to integrate:

Код: Выделить всё

np.vectorize(quad)(integrand,0,1,args=(Anew,B))

However, I receive the following error message:

Код: Выделить всё

integrand() missing 1 required positional argument: 'B'

Although this states that 'B' is missing, I don't understand it as I am providing B as an argument. I am also not sure if I am implementing vectorization correctly.


Источник: https://stackoverflow.com/questions/565 ... l-of-a-mat