Python:在所有元素的2D数组中显示索引

时间:2017-10-27 16:39:25

标签: python arrays

如果我有一个列表/数组,例如:

B = [[1,2,3,4],[2,3,4,5],[4,3,2,4]] 

(共有50个元素)

我只想显示每个元素的第二个值。例如3,4,2

我尝试了B([:,2])之类的内容,但我一直收到错误"TypeError: list indices must be integers or slices, not tuple"

我以为我可能不得不使用某种循环?

3 个答案:

答案 0 :(得分:3)

列出对救援的理解:

result = [i[2] for i in B]

答案 1 :(得分:1)

在Python 3.x

list(zip(*B))[2]

在Python 2.x

zip(*B)[2]

输出:

(3, 4, 2)

答案 2 :(得分:0)

使用List Comprehension

#`n` is the row
>>> [row[n] for row in B]

#driver value:

IN : B = [[1,2,3,4],[2,3,4,5],[4,3,2,4]]
IN : n = 2
OUT : [3, 4, 2]