Question

Дан масив, елементами якого є нулі та одиниці. Створити новий масив, замінив нулі на одиниці, одиниці на нулі.

Дан масив, елементами которого есть нули и еденицы. Создать новый масив, в котором нули заменили на еденицы, а еденицы на нули.

Answer 1

Відповідь:

Python3:

import numpy as np

arr = np.array([0, 1, 1, 0, 0, 1])

new_arr = np.where(arr == 0, 1, 0)

print(new_arr)

Пояснення:

Імпортуємо бібліотеку "numpy", яка дозволить працювати з масивами та математичними операціями. Далі створюємо масив з нулів та одиниць за допомогою функції "np.array()". Після цього створюємо новий масив, інвертувавши значення оригінального масиву за допомогою функції "np.where()". Функція "np.where()" приймає три аргументи: умову, значення для випадку True і значення для випадку False. У нашому випадку умова "arr == 0" перевіряє, чи є елемент масиву рівним нулю, якщо так, то функція повертає 1, якщо ні - то повертає 0. На фіналі друкуємо новий масив за допомогою функції "print()".