Изучение Numpy: стекирование массивов в новом измерении

В этой статье блога мы углубимся в мощные возможности библиотеки NumPy в Python и рассмотрим различные методы объединения массивов в новом измерении. Операции со стеками играют основополагающую роль в манипулировании данными и могут помочь вам комбинировать и изменять массивы в соответствии с вашими конкретными потребностями. Мы предоставим примеры кода и познакомим вас с различными методами выполнения этой задачи. Итак, начнём!

Методы сложения массивов в новом измерении:

  1. np.newaxis:
    Один из самых простых способов добавить новое измерение в массив — использовать атрибут np.newaxis. Это позволяет вам явно указать новое положение оси. Вот пример:

    import numpy as np
    arr = np.array([1, 2, 3])  # Original array
    # Stacking in a new dimension using np.newaxis
    stacked_arr = arr[np.newaxis, :]  # Adding a new dimension along rows
    print(stacked_arr)
  2. np.expand_dims:
    Функция np.expand_dims— еще один удобный метод вставки новой оси в массив. В качестве входных данных он принимает массив и положение оси. Вот пример:

    import numpy as np
    arr = np.array([1, 2, 3])  # Original array
    # Stacking in a new dimension using np.expand_dims
    stacked_arr = np.expand_dims(arr, axis=1)  # Adding a new dimension along columns
    print(stacked_arr)
  3. np.vstack и np.hstack:
    Функции np.vstackи np.hstackпозволяют складывать массивы по вертикали и горизонтали соответственно при создании новое измерение. Вот пример:

    import numpy as np
    arr1 = np.array([1, 2, 3])  # First array
    arr2 = np.array([4, 5, 6])  # Second array
    # Stacking vertically using np.vstack
    stacked_arr = np.vstack((arr1, arr2))
    print(stacked_arr)
  4. np.concatenate:
    Функция np.concatenateможет использоваться для укладки массивов вдоль указанной оси. Если для параметра оси установлено значение None, создается новое измерение. Вот пример:

    import numpy as np
    arr1 = np.array([1, 2, 3])  # First array
    arr2 = np.array([4, 5, 6])  # Second array
    # Stacking in a new dimension using np.concatenate
    stacked_arr = np.concatenate((arr1, arr2), axis=None)
    print(stacked_arr)
  5. np.stack:
    Функция np.stack— это более гибкий способ складывать массивы в новое измерение. Он позволяет указать ось, вдоль которой должно быть создано новое измерение. Вот пример:

    import numpy as np
    arr1 = np.array([1, 2, 3])  # First array
    arr2 = np.array([4, 5, 6])  # Second array
    # Stacking in a new dimension using np.stack
    stacked_arr = np.stack((arr1, arr2), axis=1)
    print(stacked_arr)

В этой статье мы рассмотрели несколько методов размещения массивов в новом измерении с использованием мощной библиотеки NumPy. Мы рассмотрели такие методы, как np.newaxis, np.expand_dims, np.vstack, np.hstack, np.concatenate, и np.stack. Каждый метод предлагает разные способы выполнения задачи, что позволяет вам выбрать тот, который лучше всего соответствует вашим потребностям. Освоив эти методы, вы сможете эффективно манипулировать массивами и изменять их форму, чтобы раскрыть весь потенциал ваших данных.