Подготовить элемент к массиву numpy

У меня есть следующий массив numpy

import numpy as np

X = np.array([[5.], [4.], [3.], [2.], [1.]])

Я хочу вставить [6.] в начале. Я пробовал:

X = X.insert(X, 0)

как мне вставить в X?

Ответ 1

numpy имеет функцию insert, np.insert через np.insert с документацией.

Вы захотите использовать его в этом случае так:

X = np.insert(X, 0, 6., axis=0)

первый аргумент X указывает объект, который нужно вставить.

Второй аргумент 0 указывает где.

Третий аргумент 6. указывает, что нужно вставить.

Четвертый аргумент axis=0 указывает, что вставка должна быть в позиции 0 для каждого столбца. Мы могли бы выбрать строки, но ваш X - это вектор столбцов, поэтому я решил, что мы останемся последовательными.

Ответ 2

Вы можете попробовать следующее

X = np.append(arr = np.array([[6]]), values = X, axis= 0)

Вместо того, чтобы вставлять 6 в существующий X, добавьте 6 к X.

Итак, первый аргумент arr - это массив с числовым числом скаляра 6, второй аргумент - ваш массив, который нужно добавить, а третий - это место, где мы хотим добавить

Ответ 3

Я знаю, что это довольно старый, но короткое решение использует многократные трюки:

np.r_[[[6.]], X]

Если вам нужно сделать это во втором измерении, вы можете использовать np.c_.

Я думаю, что это наименее загроможденная версия, о которой я могу думать