Векторизация вложенных циклов

Question

Я пытаюсь векторизовать вложенный цикл с условием.ниже небольшая фиктивная программа того, что я пытаюсь сделать.полная версия кода будет расширять пределы памяти моего компьютера (больше вложенных циклов и более сложные условные выражения), поэтому я также собираюсь использовать широковещательную передачу, чтобы свести к минимуму использование памяти - т.е. просто удерживая конечные результаты, а не простобольшой файл с некоторыми элементами, помеченными для исключения.

я думаю, мне нужно 'где' или маска, но проблема в том, что код не повторяется по обоим массивам.любая помощь будет принята с благодарностью.: -)

#conventional python
import numpy as np
myList = []
count = 0
for a in range(5):
  for b in range(5):
    if a != b:
      myList.append([a, b])
    else:
      count += 1

print(myList)

#attempt at vectorised numpy
a = np.arange(5).reshape(1,5)
b = np.arange(5)
output = np.where(a != b, [a,b])
print(output)