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

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

Сообщение 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, дающее тот же результат?

Вернуться в «Python»