Python - 比較嵌套列表並將匹配追加到新列表中？

Question

我想比較不等長的嵌套列表。我只對每個子列表的第一個元素之間的匹配感興趣。如果存在匹配，我希望將該匹配添加到另一個列表中，以便隨後轉換爲製表符分隔的文件。以下是我與工作的一個例子：

x = [['1', 'a', 'b'], ['2', 'c', 'd']] 

y = [['1', 'z', 'x'], ['4', 'z', 'x']] 

match = [] 

def find_match(): 
    for i in x: 
     for j in y: 
      if i[0] == j[0]: 
       match.append(j) 
      return match 

這將返回：

[['1', 'x'], ['1', 'y'], ['1', 'x'], ['1', 'y'], ['1', 'z', 'x']] 

難道是重新處理列表中刪除重複很好的做法或可在此更簡單的方式完成？

而且，是它更好地使用元組的元組和/或元組進行比較的目的是什麼？

任何幫助，非常感謝。

問候， Seafoid。

Answer 1

• 使用裝置獲得的不重複的集合。

  • 您必須使用元組而不是列表作爲項目，因爲設置的項目必須是可哈希的。

• 您發佈的代碼似乎沒有生成您發佈的輸出。我不知道你應該如何從該輸入中生成輸出。例如，輸出有'y'，輸入不是。

• 我認爲你的功能的設計可以大大改善。目前，您將x，y和match定義爲模塊級別，並明確讀取和變更它們。這不是你想要設計函數的方式 - 作爲一般規則，函數不應該在全局級別上改變某些東西。它應該明確地傳遞它所需的一切，並返回一個結果，而不是隱式地接收信息並改變它自身之外的東西。

  我會改變

  x = some list 
  y = some list 
  match = [] 
  def find_match(): 
      for i in x: 
       for j in y: 
        if i[0] == j[0]: 
         match.append(j) 
      return match # This is the only line I changed. I think you meant 
         # your return to be over here? 
  find_match() 
  

  到

  x = some list 
  y = some list 
  
  def find_match(x, y): 
      match = [] 
      for i in x: 
       for j in y: 
        if i[0] == j[0]: 
         match.append(j) 
      return match 
  match = find_match(x, y) 
  

• 採取一個新的水平，去年的變化，我通常與同類發電機

  更換模式

  def f(...): 
      return_value = [] 
      for... 
       return_value.append(foo) 
      return return_value 
  

  def f(...): 
      for... 
       yield foo 
  

  這將使上述功能

  def find_match(x, y): 
      for i in x: 
       for j in y: 
        if i[0] == j[0]: 
         yield j 
  

  另一種方式來表達這種發電機的作用是與發電機表達(j for i in x for j in y if i[0] == j[0])。

Answer 2

if i[1] == j[1] 

檢查是否陣列的秒元件是相同的。你想要if i[0] == j[0]。

否則，我覺得你的代碼很可讀，不一定會改變它。

Answer 3

我不知道我是否正確理解你的問題，但鑑於您的例子看來，你可能使用了錯誤的指標：

變化

if i[1] == j[1]: 

到

if i[0] == j[0]: 

Answer 4

一個simplier表達應該在這裏工作過：

list_of_lists = filter(lambda l: l[0][0] == l[1][0], zip(x, y)) 
map(lambda l: l[1], list_of_lists) 

Answer 5

您可以通過使用集合更簡單地做到這一點。

set_x = set([i[0] for i in x]) 
set_y = set([i[0] for i in y]) 
matches = list(set_x & set_y)