Функция atleast_1d

Функция atleast_1d принимает на вход один или несколько объектов (скаляры, списки, массивы) и приводит каждый из них к одномерному массиву. Если переданный объект уже является одномерным массивом, функция возвращает его без изменений. Скаляры преобразуются в массивы с одним элементом, а многомерные массивы - уплощаются до одномерной формы.

Синтаксис

np.atleast_1d(*arrays)

Параметр arrays - произвольное количество объектов, которые требуется привести к одномерному массиву. Функция возвращает одномерный массив или кортеж из одномерных массивов, если передано несколько аргументов.

Пример

Передадим функции скалярное значение:

import numpy as np

res = np.atleast_1d(10)
print(res)

Результат выполнения кода:

[10]

Пример

Приведем список к одномерному массиву:

import numpy as np

res = np.atleast_1d([1, 2, 3, 4, 5])
print(res)

Результат выполнения кода:

[1 2 3 4 5]

Пример

Передадим многомерный массив:

import numpy as np

arr = np.array([[1, 2, 3], [4, 5, 6]])
res = np.atleast_1d(arr)
print(res)

Результат выполнения кода:

[1 2 3 4 5 6]

Пример

Передадим несколько аргументов разного типа:

import numpy as np

res = np.atleast_1d(5, [7, 8, 9], np.array([[1, 2], [3, 4]]))
print(res)

Результат выполнения кода:

[array([5]), array([7, 8, 9]), array([1, 2, 3, 4])]

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

Пример

Передадим одномерный массив и убедимся, что он возвращается без изменений:

import numpy as np

arr = np.array([1, 2, 3, 4, 5])
res = np.atleast_1d(arr)
print(res)
print(res is arr)

Результат выполнения кода:

[1 2 3 4 5]
True

В этом примере функция возвращает тот же объект, а не его копию.

Смотрите также

функцию atleast_2d,
которая приводит входные данные к двумерным массивам
функцию atleast_3d,
которая приводит входные данные к трехмерным массивам
функцию expand_dims,
которая расширяет форму массива, добавляя новые оси
функцию squeeze,
которая удаляет оси массива с размером 1