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

Question

Maksim Litvinov · Accepted Answer

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

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

Ответы

Каталог