🧩 Как сложить элементы двух массивов в Python: простое руководство

array1 = [1, 2, 3]
array2 = [4, 5, 6]

sum_array = [x + y for x, y in zip(array1, array2)]
print(sum_array)
   

Как сложить элементы двух массивов в Python

Добро пожаловать! В этой статье мы рассмотрим, как сложить элементы двух массивов в Python. У нас есть два массива, и мы хотим их объединить, чтобы получить новый массив, содержащий сумму элементов из каждого переданного массива.

Метод 1: Использование цикла for

Первый способ - использовать цикл for для прохода по элементам каждого массива и складывать соответствующие элементы. Давайте рассмотрим пример:

    def sum_arrays(arr1, arr2):
        # Создаем пустой массив для хранения суммы элементов
        result = []
        
        # Проверяем, что оба массива имеют одинаковую длину
        if len(arr1) != len(arr2):
            print("Длины массивов не совпадают!")
            return None
        
        # Проходим по элементам каждого массива и складываем их
        for i in range(len(arr1)):
            sum_elem = arr1[i] + arr2[i]
            result.append(sum_elem)
        
        return result
    

Теперь мы можем вызвать эту функцию, передав два массива, и она вернет новый массив, содержащий сумму элементов:

    array1 = [1, 2, 3]
    array2 = [4, 5, 6]
    
    # Вызываем функцию sum_arrays и сохраняем результат в переменной sum_result
    sum_result = sum_arrays(array1, array2)
    
    # Выводим результат
    print(sum_result)
    # Результат: [5, 7, 9]
  

Метод 2: Использование генератора списка (List comprehension)

Второй способ - использовать генератор списка (List comprehension) для более компактного кода. Генератор списка позволяет создавать новый список, применяя определенное выражение к каждому элементу старого списка. Давайте посмотрим на пример:

    def sum_arrays(arr1, arr2):
        # Используем генератор списка для создания нового списка,
        # содержащего сумму элементов из каждого массива
        result = [arr1[i] + arr2[i] for i in range(len(arr1))]
        
        return result
  

Теперь мы можем вызвать эту функцию так же, как и в предыдущем примере, и она также вернет новый массив с суммой элементов:

    array1 = [1, 2, 3]
    array2 = [4, 5, 6]
    
    # Вызываем функцию sum_arrays и сохраняем результат в переменной sum_result
    sum_result = sum_arrays(array1, array2)
    
    # Выводим результат
    print(sum_result)
    # Результат: [5, 7, 9]
  

Вывод

В этой статье мы рассмотрели два способа сложения элементов из двух массивов в Python. Первый способ использует цикл for для прохода по элементам каждого массива и складывания их. Второй способ - использование генератора списка (List comprehension), который позволяет создавать новый список, содержащий сумму элементов из каждого массива.

Оба способа являются эффективными и могут быть использованы в зависимости от ваших предпочтений и требований. Попробуйте использовать эти методы в своих проектах и улучшите свои навыки программирования на Python!