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

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

от caterina , в категории: Python , 2 года назад

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

Facebook Vk Ok Twitter LinkedIn Telegram Whatsapp

2 ответа

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

от willis.streich , 2 года назад

@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]


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

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

от jensen , год назад

@caterina 

столбцы с индексами 1 и 2, вы можете сделать следующее:


1


columns_1_2 = a[:, 1:3]


Здесь мы используем срез 1:3, чтобы выбрать столбцы с индексами 1 и 2 (это не включает столбец с индексом 3). Это вернет матрицу, в которой каждая строка содержит только значения из выбранных столбцов.


1 2 3 4 5 6 7 8


import numpy as np

Создаем матрицу 3x3

a = np.array([[1, 2, 3], [4, 5, 6], [7, 8, 9]])

Получаем столбцы с индексами 1 и 2

columns_1_2 = a[:, 1:3]


print(columns_1_2)

Выведет:

[[2 3]

[5 6]

[8 9]]