如何在python数组中打印列?
问题内容:
我每行有3个数字,深4列。我正在努力弄清楚如何编写代码以打印指定列而不是行中的所有数字。
我搜索了可以轻松解释此问题的教程,但找不到任何有用的帮助。谁能指出我正确的方向?
问题答案:
如果您将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])