Более одного истинного в одном тензорном волокне - PullRequest
0 голосов
/ 21 апреля 2020

Учитывая массив a_ij с формой (n,m), чья сумма столбца равна единице:

>>> a
array([[1, 1, 0],
       [0, 0, 1],
       [0, 0, 0]])

Без циклов for , я хотел бы найти 1-d np.ndarray b_j с формой (m,), чей j '-й элемент содержит True, если в той же i -ой строке, которая уже имеет j' -й, есть хотя бы одно другое значение True. в нем значение True столбца. (Ничего себе, это было глотком, и даже не так ясно ...).

Ответ будет:

>>> b
array([1,1,0])

Чтение b: значение True в первом элемент означает, что в первой строке a (т.е. во втором столбце) было по крайней мере еще одно значение True. Третий элемент b, являющийся False, сигнализирует об отсутствии другого значения True во втором ряду массива a.

1 Ответ

1 голос
/ 21 апреля 2020

Если я все правильно понимаю:

(a.sum(axis=1) > 1) @ a
Добро пожаловать на сайт PullRequest, где вы можете задавать вопросы и получать ответы от других членов сообщества.
...