Как добавить матрицу перекрестно-диагональных элементов в Python? ⇐ Python

[Anonymous]

У меня есть матрицы емкостей, и у меня возникли проблемы с созданием недиагональных элементов для этой матрицы (cross_off_diagonal, особенно часть Cg матрицы).

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

def capacitance_matrix_inv_sqrt(Cg, Cc, Cja, N):
diagonal = np.ones(N) * (2 * Cja + Cg)
diagonal[0] = Cja + Cg + Cc  # Special condition for the first element
diagonal[-1] = Cja + Cg + Cc
off_diagonal = np.ones(N - 1) * (-Cja)
# Construct the capacitance matrix
C = np.diag(diagonal) + np.diag(off_diagonal, k=1) + np.diag(off_diagonal, k=-1)
return C

Я прикрепил снимок матриц, которые я хотел бы создать, извините, я не знаю, как их здесь показать.



Подробнее здесь: https://stackoverflow.com/questions/784 ... -in-python