как добавить столбец в numpy
Ответы
Maksim Litvinov
27 августа 2024
Для добавления нового столбца в массив NumPy можно воспользоваться функцией numpy.insert()
или просто использовать оператор индексирования.
Вот примеры обоих методов:
- Используя
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)
- Используя оператор индексирования:
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