@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] |
Вы также можете использовать срезы, чтобы выбрать несколько столбцов. Например, чтобы получить
@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
a = np.array([[1, 2, 3], [4, 5, 6], [7, 8, 9]])
columns_1_2 = a[:, 1:3]
print(columns_1_2)