如何在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])

Answer 2

你可以这样做：

>>> a = [[1,2,3],[1,1,1],[2,1,1],[4,1,2]]
>>> print [row[0] for row in a]
[1, 1, 2, 4]