Как разбить массив на подмассивы в Python: простой способ

Вот простой способ разделить массив на подмассивы в Python:

def split_array(arr, n):
    return [arr[i:i+n] for i in range(0, len(arr), n)]

my_array = [1, 2, 3, 4, 5, 6, 7, 8, 9]
subarrays = split_array(my_array, 3)

print(subarrays)

В данном примере мы определяем функцию split_array, которая принимает на вход исходный массив (arr) и размер подмассивов (n). Затем мы используем генератор списка, чтобы разбить исходный массив на подмассивы.

В приведенном примере исходный массив [1, 2, 3, 4, 5, 6, 7, 8, 9] будет разделен на подмассивы размером 3 элемента каждый. Результат будет выведен на экран.

Как разбить массив на подмассивы в Python

Разделение массива на подмассивы является распространенной задачей в программировании. В Python существует несколько способов достичь этой цели. В этой статье мы рассмотрим несколько подходов с примерами кода.

1. Использование срезов (slicing)

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

array = [1, 2, 3, 4, 5, 6, 7, 8, 9, 10]
subarrays = [array[i:i+3] for i in range(0, len(array), 3)]

print(subarrays)  # [[1, 2, 3], [4, 5, 6], [7, 8, 9], [10]]

В приведенном выше примере мы разбиваем массив array на подмассивы длиной 3 элемента. Мы используем генератор списка и срезы, чтобы создать подмассивы.

2. Использование функции numpy.array_split

Если вам требуется более мощный и гибкий инструмент для разбиения массива на подмассивы, вы можете использовать функцию numpy.array_split из библиотеки NumPy.

import numpy as np

array = np.array([1, 2, 3, 4, 5, 6, 7, 8, 9, 10])
subarrays = np.array_split(array, 4)

print(subarrays)  # [array([1, 2, 3]), array([4, 5, 6]), array([7, 8]), array([ 9, 10])]

В этом примере мы импортируем библиотеку NumPy и создаем массив array. Затем мы используем функцию numpy.array_split с аргументом, указывающим количество подмассивов, на которые мы хотим разделить исходный массив. Функция возвращает список с разбитыми подмассивами.

3. Использование цикла

Если вам нужно более гибкое управление процессом разбиения массива на подмассивы, вы можете использовать цикл для создания и заполнения подмассивов.

array = [1, 2, 3, 4, 5, 6, 7, 8, 9, 10]
subarrays = []
subarray_size = 3

for i in range(0, len(array), subarray_size):
    subarray = array[i:i+subarray_size]
    subarrays.append(subarray)

print(subarrays)  # [[1, 2, 3], [4, 5, 6], [7, 8, 9], [10]]

В этом примере мы используем цикл for для прохода по массиву и создания подмассивов заданного размера subarray_size. Затем мы добавляем каждый подмассив в список subarrays.

4. Использование генератора функций

Если вы предпочитаете использовать генераторы функций для разбиения массива на подмассивы, вы можете определить собственную функцию-генератор.

def split_array(array, subarray_size):
    for i in range(0, len(array), subarray_size):
        yield array[i:i+subarray_size]

array = [1, 2, 3, 4, 5, 6, 7, 8, 9, 10]
subarrays = list(split_array(array, 3))

print(subarrays)  # [[1, 2, 3], [4, 5, 6], [7, 8, 9], [10]]

В этом примере мы определяем функцию-генератор split_array, которая возвращает подмассивы исходного массива. Мы используем ключевое слово yield для возврата очередного подмассива во время итерации.

Заключение

В этой статье мы рассмотрели несколько способов разбить массив на подмассивы в Python. Вам доступны различные подходы, включая использование срезов, функции numpy.array_split, цикла for и генераторов функций. Выберите подход, который наиболее удобен и соответствует вашим потребностям.