Python|获取矩阵的第 K 列

有时，在使用Python Matrix 时，可能会遇到需要找到 Matrix 的第 K 列的问题。这是机器学习领域中一个非常流行的问题，解决这个问题很有用。让我们讨论一些可以解决这个问题的方法。

方法#1：使用列表推导
这个问题可以使用列表解析来解决，我们可以遍历所有行并有选择地收集出现在第 K 个索引处的所有元素。

# Python3 code to demonstrate working of
# Get Kth Column of Matrix
# using list comprehension
  
# initialize list
test_list = [[4, 5, 6], [8, 1, 10], [7, 12, 5]]
  
# printing original list
print("The original list is : " + str(test_list))
  
# initialize K
K = 2
  
# Get Kth Column of Matrix
# using list comprehension
res = [sub[K] for sub in test_list]
  
# printing result
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 code to demonstrate working of
# Get Kth Column of Matrix
# using zip()
  
# initialize list
test_list = [[4, 5, 6], [8, 1, 10], [7, 12, 5]]
  
# printing original list
print("The original list is : " + str(test_list))
  
# initialize K
K = 2
  
# Get Kth Column of Matrix
# using zip()
res = list(zip(*test_list)[K])
  
# printing result
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]