來自多維列表的特定索引的新列表？

我基本上試圖從多維列表中提取一些數據到一個新列表中。雖然這段代碼做我想做的，但我想知道是否有更好的方法來獲得這個結果沒有循環通過原始列表？來自多維列表的特定索引的新列表？

aa = [ 
    ["a", "aa", "aaa"], 
    ["b", "bb", "bbb"], 
    ["c", "cc", "ccc"], 
] 
b = [] 
for a in aa: 
    b.append(a[1]) 
b 
>> Result: ['aaa', 'bbb', 'ccc']

來源

2013-03-02 Panupat

你現在的代碼實際上應該返回'['aa'，'bb'，'cc']'。這是故意的，還是你的意思是有'a [2]'而不是'a [1]'。 – TerryA 2013-03-02 05:04:44

啊是的，我想我抄了結果的錯誤部分 – Panupat 2013-03-02 05:47:57

>>> aa = [["a", "aa", "aaa"], 
      ["b", "bb", "bbb"], 
      ["c", "cc", "ccc"]] 
>>> b = zip(*aa)[2] 
>>> print b 
['aaa', 'bbb', 'ccc']

我很肯定zip仍然遍歷列表，但這是一個更加簡單的方法來獲得所需的結果。

注意的是Python 3，zip返回迭代器，所以你需要轉換到一個列表第一

list(zip(*aa))[2]

，或者使用列表理解

[i[2] for i in aa]

來源

2013-03-02 04:59:57 Volatility

你可以使用列表理解：

b = [aa[i][2] for i in range(len(aa))]

或更快和更短的攪拌器曾建議

b = [a[2] for a in aa]

來源

2013-03-02 04:58:41 Raufio

爲什麼不''[a [2]爲aa]'？ – Blender 2013-03-02 04:59:00

下面是與列表理解方式：

>>> aa = [ 
["a", "aa", "aaa"], 
["b", "bb", "bbb"], 
["c", "cc", "ccc"], 
] 
>>> b = [i[2] for i in aa] 
>>> print b 
['aaa', 'bbb', 'ccc']

來源

2013-03-02 04:59:35 TerryA

你也可以用numpy的做：

>>> >>> a = numpy.array([ 
...  ["a", "aa", "aaa"], 
...  ["b", "bb", "bbb"], 
...  ["c", "cc", "ccc"], 
... ]) 
>>> a[:,2] 
    array(['aaa', 'bbb', 'ccc'], 
     dtype='|S3')

來源

2013-03-02 05:02:29 Blender

來自多維列表的特定索引的新列表？

回答

相關問題