Назначьте последовательные числа ячейкам массива numpy, удовлетворяющим определенному условию ⇐ Python

[Anonymous]

Этот скрипт Python:

импортировать numpy как np а = np.random.rand(8, 8) б = np.full_like(а, -1) п = 0 для i, val в np.ndenumerate(a): если значение < 0,666: б[я] = п п += 1 создает массив b с последовательными натуральными числами в местах, где a < 0,666 (заполнитель для произвольного условия) и -1 в противном случае. Есть ли волшебное выражение NumPy, дающее тот же результат?