numpy оператор для сравнения двух векторов - PullRequest
2 голосов
/ 01 апреля 2020

Мне интересно, есть ли оператор python - numpy для сравнения двух векторов одинаковой формы. В частности, enter image description here

Могу ли я получить результаты напрямую через numpy API и что это? Большое спасибо!

1 Ответ

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

Если ваши массивы a и b:

c = ((np.repeat(a, b.shape[0]).reshape(a.shape[0], b.shape[0]) - b) == 0).astype(int)

или, как сказал hpaulj и FBruzzesi:

c = (a[:, None] == b).astype(int)
Добро пожаловать на сайт PullRequest, где вы можете задавать вопросы и получать ответы от других членов сообщества.
...