Как правильно добавить матрицу в конец существующей матрицы в Python? - PullRequest
1 голос
/ 07 апреля 2020

У меня есть две матрицы, A и B. Каждая из A и B имеет размер 100 x 100. Я пытаюсь создать комбинированную матрицу AB такую, которая равна 200 x 100, с элементами A в первых 100 x 100 и элементами B во втором 100 x 100.

Я пытался сделать следующее, но он говорит, что форма (2, 1, 500, 500), когда я делаю это в Python.

def get_bigAB(n, lamb):
    return np.array([[A], [get_B(n, lamb)]])

Мои записи являются числами с плавающей точкой, а не простыми целыми числами.

Моя функция get_B работает как положено, и, конечно, я использую Python 3.

Ответы [ 3 ]

2 голосов
/ 07 апреля 2020

Использование np.vstack закончилось работать. Спасибо за помощь!

def get_Alambda(n, lamb):
    B = get_lambdaI(n, lamb)
    AB = np.vstack((A, B))
    return AB
1 голос
/ 13 апреля 2020

Попробуйте использовать vstack(A, B), где матрица B добавлена ​​к нижней части матрицы A. Это даст вам необходимые размеры.

0 голосов
/ 07 апреля 2020

Вы пробовали numpy объединить

import numpy as np
AB = np.concatenate((A,B),axis=0)
print(AB)
print(AB.shape)
...