Zip（）已經初始化的列表

Question

我正在嘗試編寫一個將NumPy數組寫入文件的函數。我想用Pythonic方法使用zip()。以下代碼不起作用;我試圖將命名參數（其中的值是1D NumPy數組，鍵是標籤）**data_lists轉換爲zip函數所需的數組列表。

def write_data_file(filename,**data_lists): 
    with open("{}.dat".format(filename), "w") as f: 
     header = "#"   
     for lst in data_lists: 
      header = "{} {} \t".format(header,lst) 
     header = "{} \n".format(header) 
     f.write(header) 
     # The following line of code does not work 
     for row in zip(data_lists[lst] for lst in data_lists): 
      f.write("\t".join([str(item) for item in row])+ "\n") 

我不能分配到列表第一如下：

trial = [data_lists[lst] for lst in data_lists] 
zip(trial) 

這並沒有什麼作爲審判被視爲一個參數。我如何獲得zip()以單獨處理trial中的項目或以其他方式完成此項目？

Answer 1

變化：

zip(data_lists[lst] for lst in data_lists) 

到：

zip(*data_lists.values()) 

解包的data_listsdict值放入順序位置參數。 *被稱爲「splat」或可迭代解包運算符。

雖然有點奇怪，因爲values的順序沒有定義;你可以按照Python版本的不同順序輸入zip（並且在一個版本中，通常zip的運行順序不同）。您正在使用迭代順序來定義標題，因此列每次都會匹配標題，但列的順序會因運行而異。你可能會想嘗試接收頭作爲單獨的參數，然後在位置值，例如：

def write_data_file(filename, headers, *data_lists): 
    if len(headers) != len(data_lists): ... raise Exception for mismatched lengths ... 

那麼你只是做zip(*data_lists)（無.values()調用），並獲得一致的標題排序。