Как получить столбец матрицы в numpy?

Пользователь

от caterina , в категории: Python , месяц назад

Как получить столбец матрицы в numpy?

Facebook Vk Ok Twitter LinkedIn Telegram Whatsapp

1 ответ

Пользователь

от willis.streich , месяц назад

@caterina 

Чтобы получить столбец матрицы в NumPy, вы можете использовать индексирование с помощью срезов (slicing). Например, если у вас есть матрица a, вы можете получить столбец с индексом i следующим образом:

1
column_i = a[:, i]


Обратите внимание, что в данном случае мы используем двоеточие (:) в качестве первого индекса, что означает, что мы выбираем все строки в матрице. Это означает, что мы получаем вектор, который содержит все элементы столбца с индексом i.


Например:

1
2
3
4
5
6
7
8
9
import numpy as np

# Создаем матрицу 3x3
a = np.array([[1, 2, 3], [4, 5, 6], [7, 8, 9]])

# Получаем столбец с индексом 1 (это второй столбец)
column_1 = a[:, 1]

print(column_1)  # Выведет: [2 5 8]


Вы также можете использовать срезы, чтобы выбрать несколько столбцов. Например, чтобы получить