從2D python列表中拉出某些元素？

Question

作爲一種簡單的方法來測試拉出一個大的地形設置，輸入在我的節目一個二維表的某些元素的正確方式，我把下面的到Python殼牌：

->>>test = [ [1,2,3],[4,5,6],[7,8,9] ] 

，並在努力來測試它，我得到這個：

->>>test1 = test[:][2] 

->>>test1 

[7, 8, 9] 

從輸入test[:][2]我預計TEST1返回[3,6,9]

有人能向我解釋爲什麼我[7,8,9]呢？

此外，如果你能解釋我將如何得到[3,6,9]的輸出？我知道我可以只說以下內容：

>>>test1 = [ test[0][2], test[1][2], test[2][2] ] 

，但如果我在我的程序的情況下寫的，這條線將是程較長且難以維護。如果比上面列出的方式更容易獲得[3,6,9]，我將非常感謝您獲得該信息的一些見解。

Answer 1

爲標準的Python，你可以這樣做：

>>> test = [[1,2,3],[4,5,6],[7,8,9]] 
>>> [x[2] for x in test] 
[3, 6, 9] 
>>> 

爲test1 = test[:][2]，這是因爲test[:]只是等於test（編輯：實際上，它是test精確副本，而不是同一個列表 ）

也看看numpy庫。

Answer 2

test[:]返回test的副本，所以[[1,2,3],[4,5,6],[7,8,9]]。所以test[:][2]是[7,8,9]。 使用列表comprehesion你可以得到你想要什麼輕鬆：

>>> [x[2] for x in test] 
[3, 6, 9]