Как заменить все значения в массиве по условию?
Возможно ли заменить в массиве одни значения на другие, например, все 4 на 0?
Ответы (2 шт):
Автор решения: MarianD
→ Ссылка
np.where(a == 4, 0, a)
Тест:
In[1]: a = np.random.randint(1, 5, 20)
In[2]: a
Out[2]: array([4, 1, 4, 2, 4, 3, 1, 2, 1, 2, 4, 1, 4, 2, 4, 3, 1, 1, 1, 3])
In[3]: np.where(a == 4, 0, a)
Out[3]: array([0, 1, 0, 2, 0, 3, 1, 2, 1, 2, 0, 1, 0, 2, 0, 3, 1, 1, 1, 3])