как добавить столбец в numpy

Аватар пользователя Maksim Litvinov
Maksim Litvinov
27 августа 2024

Для добавления нового столбца в массив NumPy можно воспользоваться функцией numpy.insert() или просто использовать оператор индексирования.

Вот примеры обоих методов:

  1. Используя numpy.insert():
import numpy as np

# Создаем исходный массив
arr = np.array([[1, 2], [3, 4], [5, 6]])

# Добавляем новый столбец с помощью функции insert()
new_column = np.array([7, 8, 9])
arr = np.insert(arr, 1, new_column, axis=1)
print(arr)
  1. Используя оператор индексирования:
import numpy as np

# Создаем исходный массив
arr = np.array([[1, 2], [3, 4], [5, 6]])

# Создаем новый столбец
new_column = np.array([7, 8, 9])

# Добавляем новый столбец с помощью оператора индексирования
arr = np.c_[arr, new_column]
print(arr)

Оба эти метода добавят новый столбец в конец массива arr. При этом первый индекс указывает на позицию, на которую нужно вставить новый столбец (в случае метода numpy.insert()), а оператор np.c_ объединяет два массива по столбцам.

1 0
Познакомьтесь с основами аналитики данных бесплатно