數組中的搜索元素

Question

對不起，如果我得到的術語錯誤 - 我只是剛剛開始學習Python，而且我接受來自朋友的指令而不是實際的課程。

我想搜索一個包含大量包含多個元素的數組的列表，並找到一些元素匹配的數組，但有一些不同。

在較不混淆的術語例如我有各自包含2個元素的陣列（我認爲這就是所謂的2D陣列？），以便列表：

list = [[1, 2], [2, 2], [3, 5], [4, 1], [5, 2], ...] 

以我具體實例中，每一子陣列中的第一元件剛上升線性，但是第二元素幾乎是隨機的。我想僅通過第二個數字查找或排序數組。我可以從每個數組中刪除第一個數字：

list = [2, 2, 5, 1, 2 ...] 

然後使用類似「如果list [X] == 1」發現「1」等
（邊注：我不是確定如果重複一個值，如何找到所有的值 - 我不記得我寫的是什麼，但它只會找到第一個匹配值的實例，例如它會檢測到第一個'2'，而不是第二個或第三個）

但我想保留每個數組中的第一個值。我的朋友告訴我，你可以用一個帶有值和鍵的字典，這對我的例子很有用，但我想知道更一般的方法是什麼。

所以在我的例子，我希望如果我寫這樣的：

if list[[?, x]] == [?, 1] 

然後它會找到數組，其中數組的第二個值是1，（即[4,1]在我的例子）而不關心第一個值。顯然它沒有工作，因爲'？'就我所知，它不是Python語法，但希望你能看到我正在嘗試做什麼？

因此，對於更一般的情況下，如果我有5個維數組列表，我想找到每個陣列的第二和第四值，我會寫：

if list[[?, x, ?, y, ?]] == [?, a, ?, b, ?] 

而且，它還將匹配任何其中第二個元素的值爲'a'，第四個元素的值爲'b'。

例如[3，a，4，b，7]，[20，a，1，b，9]，['cat'，a，'dog'，b，'fish']等都是可能的結果。命令。

所以我想知道是否有任何類似的方式來使用問號（但實際工作）的方法來表示數組中的元素可以有任何價值。

Answer 1

使用sorted如果你想保持原來的list不受影響

lst = [[1, 2], [2, 2], [3, 5], [4, 1], [5, 2]] 
In [103]: sorted(lst, key=lambda x: x[1]) 
Out[103]: [[4, 1], [1, 2], [2, 2], [5, 2], [3, 5]] 

使用別的list.sort排序當前列表中，並保持排序列表

In [106]: lst.sort(key=lambda x: x[1]) 
In [107]: lst 
Out[107]: [[4, 1], [1, 2], [2, 2], [5, 2], [3, 5]] 

或使用operator.itemgetter

from operator import itemgetter 
In [108]: sorted(lst, key=itemgetter(1)) 
Out[108]: [[4, 1], [1, 2], [2, 2], [5, 2], [3, 5]] 

Answer 2

你可以使用list comprehension構建所有期望項的列表：

In [16]: seq = [[1, 2], [2, 2], [3, 5], [4, 1], [5, 2]] 

要查找所有項，其中第二元素是1：

In [17]: [pair for pair in seq if pair[1] == 1] 
Out[17]: [[4, 1]] 

此發現的所有項目，其中所述第二元件是2：

In [18]: [pair for pair in seq if pair[1] == 2] 
Out[18]: [[1, 2], [2, 2], [5, 2]] 

---

而不是

if list[[?, x, ?, y, ?]] == [?, a, ?, b, ?] 

你可以使用

[item for item in seq if item[1] == 'a' and item[3] == 'b'] 

---

注意，但是，每次使用列表理解，Python有循環 通過所有的seq的元素。如果你正在做這個搜索多次， 你可能會更好建設的字典：

import collections 
seq = [[1, 2], [2, 2], [3, 5], [4, 1], [5, 2]] 
dct = collections.defaultdict(list) 
for item in seq: 
    key = item[1] 
    dct[key].append(item) 

然後你可以訪問的項目是這樣的：

In [22]: dct[1] 
Out[22]: [[4, 1]] 

In [23]: dct[2] 
Out[23]: [[1, 2], [2, 2], [5, 2]] 

---

列表內涵

[pair for pair in seq if pair[1] == 1] 

大致相當於

result = list() 
for pair in seq: 
    if pair[1] == 1: 
     result.append(pair) 

從這個意義上說，result將然後等於列表理解。

列表理解只是一種語法上更漂亮的方式來表達相同的 的東西。

上面的列表理解有三個部分：

[expression for-loop conditional] 

表達是pair，for循環是for pair in seq，和條件是if pair[1] == 1。

大多數，but not all列表解析共享此語法。完整的列表理解語法是given here。

Answer 3

排序在第二元件上爲方含列表（或元組）的列表：

from operator import itemgetter 
mylist = [[1, 2], [2, 2], [3, 5], [4, 1], [5, 2]] 
sortedlist = sorted(mylist, key=itemgetter(1)) 

見Python sorting howto。