Как упаковать матрицы в ячейки, а затем сформировать диагональную матрицу с помощью Python
Мне нужно создать диагональную матрицу в python из матрицы X, которая повторяется 3 раза. В Matlab я делаю это следующим образом:
X=[1 2 3;
4 5 6;
7 8 9]
for i=1:1:3
Brep{i}=X;
end
Mdiag=blkdiag(Brep{:})
Mdiag =
1 2 3 0 0 0 0 0 0
4 5 6 0 0 0 0 0 0
7 8 9 0 0 0 0 0 0
0 0 0 1 2 3 0 0 0
0 0 0 4 5 6 0 0 0
0 0 0 7 8 9 0 0 0
0 0 0 0 0 0 1 2 3
0 0 0 0 0 0 4 5 6
0 0 0 0 0 0 7 8 9
Но я не знаю, как это сделать в Пихтоне.
Буду признателен за любую помощь.
--------------------------------- ОТВЕТ:--------------- -----------------------
Принимая во внимание ответ Фабрицио Бернини, я изменил его код, чтобы сделать его более общим. Теперь код может повторять матрицу X (любое измерение) n раз по диагонали. Например:
import numpy as np
n=5
x = np.array([[1, 2, 3],[4, 5, 6],[7, 8, 9]])
Mdiag = np.zeros((n*x.shape[0], n*x.shape[1]))
for i in range(n):
Mdiag[x.shape[0]*i:x.shape[0]*i+x.shape[0],x.shape[1]*i:x.shape[1]*i+x.shape[1]]= x
Mdiag =[1., 2., 3., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.],
[4., 5., 6., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.],
[7., 8., 9., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.],
[0., 0., 0., 1., 2., 3., 0., 0., 0., 0., 0., 0., 0., 0., 0.],
[0., 0., 0., 4., 5., 6., 0., 0., 0., 0., 0., 0., 0., 0., 0.],
[0., 0., 0., 7., 8., 9., 0., 0., 0., 0., 0., 0., 0., 0., 0.],
[0., 0., 0., 0., 0., 0., 1., 2., 3., 0., 0., 0., 0., 0., 0.],
[0., 0., 0., 0., 0., 0., 4., 5., 6., 0., 0., 0., 0., 0., 0.],
[0., 0., 0., 0., 0., 0., 7., 8., 9., 0., 0., 0., 0., 0., 0.],
[0., 0., 0., 0., 0., 0., 0., 0., 0., 1., 2., 3., 0., 0., 0.],
[0., 0., 0., 0., 0., 0., 0., 0., 0., 4., 5., 6., 0., 0., 0.],
[0., 0., 0., 0., 0., 0., 0., 0., 0., 7., 8., 9., 0., 0., 0.],
[0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 1., 2., 3.],
[0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 4., 5., 6.],
[0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 7., 8., 9.]
Спасибо всем за предоставленные ответы.
5 ответов
Вы можете сделать что-то подобное с помощью numpy:
import numpy
x = np.array([[1, 2, 3],[4, 5, 6],[7, 8, 9]])
Mdiag = np.zeros((9, 9))
for i in range(3):
Mdiag[3*i:3*i+3, 3*i:3*i+3] = x
вы также можете сгенерировать свою исходную матрицу, используя np.reshape:
x = np.arange(1, 10)
x = x.reshape((3, 3))
В общем, если у вас есть подматрица nxn и вы хотите создать диагональную блочную матрицу MxM, где M = c*n, вы можете сделать:
import numpy
x = ... define here your nxn matrix
Mdiag = np.zeros((M, M))
for i in range(n):
Mdiag[n*i:n*i+n, n*i:n*i+n] = x
Существует множество встроенных функций для копирования блочных матриц и генерации блочных матриц, но я не знаю ни одной, генерирующей диагональную блочную матрицу с помощью всего одной команды. См. Также это для справки:
[1]: https://numpy.org/doc/stable/reference/generated/numpy.matlib.repmat.html и [2]: https://numpy.org/doc/stable/reference/generated/numpy.block.html?highlight=block%20matrix
Вы просто можете использовать scipy.linalg.block_diag
:
from scipy.linalg import block_diag
X=[[1, 2, 3],
[4,5,6],
[7, 8, 9]]
block_diag(X, X, X)
Выход:
array([[1, 2, 3, 0, 0, 0, 0, 0, 0],
[4, 5, 6, 0, 0, 0, 0, 0, 0],
[7, 8, 9, 0, 0, 0, 0, 0, 0],
[0, 0, 0, 1, 2, 3, 0, 0, 0],
[0, 0, 0, 4, 5, 6, 0, 0, 0],
[0, 0, 0, 7, 8, 9, 0, 0, 0],
[0, 0, 0, 0, 0, 0, 1, 2, 3],
[0, 0, 0, 0, 0, 0, 4, 5, 6],
[0, 0, 0, 0, 0, 0, 7, 8, 9]], dtype=int32)
Другой вариант, используя np.bmat
, np.asmatrix
а также np.asarray
, может быть это:
import numpy as np
X=np.asmatrix([[1, 2, 3],
[4,5,6],
[7, 8, 9]])
y = np.asmatrix(np.zeros((3, 3)))
r=np.asarray(np.bmat('X, y, y; y, X, y; y, y, X'))
print(r)
Один из способов сделать это - создать большую нулевую матрицу и напрямую вставить x
по диагонали:
N = 3
y = np.zeros((x.shape[0] * N, x.shape[1] * N))
for i in range(N):
y[i * x.shape[0] : (i + 1) * x.shape[0],
i * x.shape[1] : (i + 1) * x.shape[1]] = x
Другое решение: создать горизонтальный "пояс" из x
и столько нулей, сколько нужно. Затем сложите три копии этого ремня со сдвигом:
belt = np.hstack([x, np.zeros((x.shape[0], (x.shape[1] - 1) * N))])
np.vstack([np.roll(belt, i) for i in range(0, x.shape[0] * N, x.shape[0])])
Вы не пометили какой-либо "расширенный" модуль ( scipy, numpy,...), поэтому приступим:
Вы можете создать функцию, которая сделает это самостоятельно. Матрицы представлены в виде списков списков в Python (если вы не входите в numpy и т. Д.):
def diag_size(what, size = 4):
"""Create a diagonal matrix of size len(what)*size where the list of list
'what' is on the diagonal of the embiggended result"""
rv = [[]]
s = len(what) # quadratic => len(what) == len(what[0])
for row in range(size*s):
for col in range(size*s):
if row//s == col//s:
rv[-1].append(base[row%s][col%s])
else:
rv[-1].append(0)
rv.append([])
return rv[:-1]
n = 3 # size of the initial "matrix"
# create base case
base = [list(range(i+1,i+n+1)) for i in range(0,n*n,n)]
# create the diagonal one
print(*diag_size(base),sep="\n")
Выход:
# with size=4 you get a 4 wide diagonal matrix
[1, 2, 3, 0, 0, 0, 0, 0, 0, 0, 0, 0]
[4, 5, 6, 0, 0, 0, 0, 0, 0, 0, 0, 0]
[7, 8, 9, 0, 0, 0, 0, 0, 0, 0, 0, 0]
[0, 0, 0, 1, 2, 3, 0, 0, 0, 0, 0, 0]
[0, 0, 0, 4, 5, 6, 0, 0, 0, 0, 0, 0]
[0, 0, 0, 7, 8, 9, 0, 0, 0, 0, 0, 0]
[0, 0, 0, 0, 0, 0, 1, 2, 3, 0, 0, 0]
[0, 0, 0, 0, 0, 0, 4, 5, 6, 0, 0, 0]
[0, 0, 0, 0, 0, 0, 7, 8, 9, 0, 0, 0]
[0, 0, 0, 0, 0, 0, 0, 0, 0, 1, 2, 3]
[0, 0, 0, 0, 0, 0, 0, 0, 0, 4, 5, 6]
[0, 0, 0, 0, 0, 0, 0, 0, 0, 7, 8, 9]
Попробуйте этот код.
import numpy as np
X = np.array([[1, 2, 3], [4, 5, 6], [7, 8, 9]])
M = np.zeros(shape = (9, 9))
for i in range(3):
M[3*i:3*(i+1), 3*i:3*(i+1)] = X