Рубрики

Python | Получить K-ю колонку с матрицей

Иногда во время работы с Python Matrix может возникнуть проблема, из-за которой нужно найти K-й столбец Matrix. Это очень популярная проблема в области машинного обучения, и ее решение полезно. Давайте обсудим некоторые способы решения этой проблемы.

Метод № 1: Использование понимания списка
Эта проблема может быть решена с использованием понимания списка, в котором мы можем перебирать все строки и выборочно собирать все элементы, встречающиеся в K-ом индексе.

# Python3 код для демонстрации работы
# Получить Kth столбец матрицы
# использование списка понимания

  
# инициализировать список

test_list = [[4, 5, 6], [8, 1, 10], [7, 12, 5]]

  
# печать оригинального списка

print("The original list is : " + str(test_list))

  
# инициализировать K

K = 2

  
# Получить Kth столбец матрицы
# использование списка понимания

res = [sub[K] for sub in test_list]

  
# результат печати

print("The Kth column of matrix is : " + str(res))

Выход :

The original list is : [[4, 5, 6], [8, 1, 10], [7, 12, 5]]
The Kth column of matrix is : [6, 10, 5]

Способ № 2: Использование zip()
Эту задачу также можно выполнить с помощью zip() . Это выполняет аналогичную задачу сбора элементов, как это делается с помощью описанного выше понимания списка, и предлагает компактное, но медленное выполнение. Работает только с Python2.

# Python-код для демонстрации работы
# Получить Kth столбец матрицы
# используя zip ()

  
# инициализировать список

test_list = [[4, 5, 6], [8, 1, 10], [7, 12, 5]]

  
# печать оригинального списка

print("The original list is : " + str(test_list))

  
# инициализировать K

K = 2

  
# Получить Kth столбец матрицы
# используя zip ()

res = list(zip(*test_list)[K])

  
# результат печати

print("The Kth column of matrix is : " + str(res))

Выход :

The original list is : [[4, 5, 6], [8, 1, 10], [7, 12, 5]]
The Kth column of matrix is : [6, 10, 5]

Рекомендуемые посты:

Python | Получить K-ю колонку с матрицей

0.00 (0%) 0 votes