如何在python数组中打印列？

Question

我有一个每行3个数字,4列深的数组.我正在努力弄清楚如何编写代码来打印指定列而不是行中的所有数字.

我已经搜索了可以轻松解释这些内容的教程,但却找不到任何有帮助的内容.谁能指出我正确的方向？

Answer 1

如果您将python列表视为行和列,可能最好使用numpy数组(如果您还没有).然后,您可以轻松打印各种行和列,例如

import numpy as np
a = np.array([[1,2,6],[4,5,8],[8,3,5],[6,5,4]])
#Print first column
print(a[:,0])
#Print second row
print(a[1,:])

请注意,否则你有一个列表列表,你需要使用类似的东西,

b = [[1,2,6],[4,5,8],[8,3,5],[6,5,4]]
print([i[0] for i in b])