Как векторизовать scipy.integrate.quad для вычисления поэлементного интеграла матрицы - PullRequest
0 голосов
/ 10 июня 2019

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

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) 

Это дает квадратную матрицу с каждым элементом функцией t.Для интеграции я использую следующее:

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

Однако я получаю следующее сообщение об ошибке:

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

Хотя это говорит о том, что «B» отсутствует, я его не понимаюкак я предоставляю B в качестве аргумента.Я также не уверен, правильно ли я реализую векторизацию.

1 Ответ

0 голосов
/ 11 июня 2019

Попробуйте scipy.integrate.quad_vec. Он еще не выпущен, поэтому вам понадобится версия scipy, находящаяся в разработке и доступная на github. Или дождитесь выхода Scipy 1.4.

Добро пожаловать на сайт PullRequest, где вы можете задавать вопросы и получать ответы от других членов сообщества.
...