Объединение двух матриц в Python - PullRequest
2 голосов
/ 16 июня 2020

Возможно ли объединение двух матриц в python? Я хочу иметь в одной матрице все элементы из двух других матриц, не повторяя ни одной из них. Например, если у нас есть:

A = [[1,2],[3,4],[5,6]]

B = [[5,6],[7,8]]

Объединение будет C = [[1,2],[3,4],[5,6],[7,8]]

Существует команда numpy для массивов: np.union1d, но я не могу найти ее для матриц. Я только что нашел np.concatenate и np.vstack, но они записывают дважды повторяющиеся элементы.

1 Ответ

1 голос
/ 16 июня 2020

Если я правильно понял ваш вопрос, вы можете добиться использования np.unique на объединенном результате A и B, как показано ниже

import numpy as np
A = np.array([[1,2],[3,4],[5,6]])
B = np.array([[5,6],[7,8]])
np.unique(np.concatenate([A, B]), axis=0)

выходы

array([[1, 2],
       [3, 4],
       [5, 6],
       [7, 8]])

или немного больше лаконичная укладка будет np.unique(np.r_[A,B], axis=0)

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