從excel數據循環到每個鍵有多個值的字典

Question

我對Python很新，我覺得這個問題相當簡單，但我無法弄清楚它...

我有一個數據表Excel中有B列字符串和C到I列作爲值。我想創建一個字典，其中對於列中的每個鍵值，我將C列的值分配給I.我計算出每行如何執行，每次執行一個，但我正在查找for循環語法在整個excel數據表中執行 。

這裏是我的代碼：

NYSE = {} 
NYSE.setdefault(sheet['B5'].value, []).append(sheet['C5'].value) 
NYSE.setdefault(sheet['B5'].value, []).append(sheet['D5'].value) 
NYSE.setdefault(sheet['B6'].value, []).append(sheet['C6'].value) 
NYSE.setdefault(sheet['B6'].value, []).append(sheet['D6'].value) 
print NYSE 

我可以手動不斷加入到這個... B7 C7，B7 D7等，但必須有辦法在功能和輸出字典迴路這一點。

Answer 1

你可以嘗試這樣的事：

from collections import defaultdict 

d = defaultdict(list) 
for line in open("pyex.csv").readlines(): 
    line = line.strip() 
    line = line.split(",") 
    key, value = line[0], line[1:] 
    d[key] += value 
print(d) 

所以，如果你有一個CSV文件看起來像這樣。其中第一列是字符串，每列第二和之後是值：

crow 19 13 
bird 16 32 

這個代碼將輸出：

defaultdict(<class 'list'>, {'crow ': ['19', '13'], 'bird': ['16', '32']}) 

[完成了0.1秒]

這可以讓你每個鍵都有多個值，因爲這些值包含在列表中。

UPDATE：

使用setdefault代替：

d = {} 
for line in open("pyex.csv").readlines(): 
    line = line.strip() 
    line = line.split(",") 
    key = line[0] 
    for value in line[1:]: 
     d.setdefault(key, []).append(value) 
print(d) 

輸出：

{'crow': ['19', '13'], 'bird': ['16', '32']} 

甚至與CSV庫

import csv 

csv_file = open("pyex.csv") 
csv_reader = csv.reader(csv_file) 

d = {} 
for line in csv_reader: 
    key = line[0] 
    for value in line[1:]: 
     d.setdefault(key, []).append(value) 
print(d) 

正如@martineu說，你不需要defaultdicts或setdefaults：

import csv 

csv_file = open("Book1.csv") 
csv_reader = csv.reader(csv_file) 

d = {} 
for line in csv_reader: 
    key = line[0] 
    d[key] = line[1:] 
print(d) 

Answer 2

你有沒有考慮過使用熊貓？我不確定你的數據目標是什麼，但它看起來像是你想要的最強大的方法。

如果您使用pandas.read_excel(path, sheetname=None)那麼它將默認創建一個字典，其中每個鍵是一個表，每個值是該表的一個數據框。然後你可以迭代你的字典來將它們合併在一起。用數據的僞樣本來獲得更具體的信息會更容易。