文件IO與defaultdict

Question

我試圖：

• 負載詞典
• 更新/更改字典
• 保存
• （重複）

問題：我想僅使用1個字典（players_scores） 工作，但defaultdict表達式創建了一個完全獨立的字典。 如何加載，更新並保存到一本字典？

代碼：

from collections import defaultdict#for manipulating dict 
players_scores = defaultdict(dict) 

import ast #module for removing string from dict once it's called back 


a = {} 

open_file = open("scores", "w") 
open_file.write(str(a)) 
open_file.close() 

open_file2 = open("scores") 
open_file2.readlines() 
open_file2.seek(0) 



i = input("Enter new player's name: ").upper() 
players_scores[i]['GOLF'] = 0 
players_scores[i]['MON DEAL'] = 0 
print() 

scores_str = open_file2.read() 
players_scores = ast.literal_eval(scores_str) 
open_file2.close() 
print(players_scores) 

Answer 1

您擦更改;而不是寫出你的文件，你重新閱讀它，結果用於替換您的players_scores字典。您的defaultdict在此之前工作得很好，即使您在這裏不能真正使用defaultdict（ast.literal_eval()確實不支持支持collections.defaultdict，只有標準的python文字字典符號）。

您可以通過使用json module這裏簡化代碼：

import json 

try: 
    with open('scores', 'r') as f: 
     player_scores = json.load(f) 
except IOError: 
    # no such file, create an empty dictionary 
    player_scores = {} 

name = input("Enter new player's name: ").upper() 
# create a complete, new dictionary 
players_scores[name] = {'GOLF': 0, 'MON DEAL': 0} 


with open('scores', 'w') as f: 
    json.dump(player_scores, f) 

你不需要defaultdict這裏所有;無論如何，你只是爲每個球員名字創建新的字典。

Answer 2

我認爲一個問題是，按照你想要的方式索引數據結構，像defaultdict(defaultdict(dict))這樣的東西是真正需要的東西 - 但不幸的是，不可能直接指定一個。然而，要解決的是，所有你需要做的就是定義一個簡單的中介工廠函數傳遞到上層defaultdict：

from collections import defaultdict 

def defaultdict_factory(*args, **kwargs): 
    """ Create and return a defaultdict(dict). """ 
    return defaultdict(dict, *args, **kwargs) 

然後你可以用players_scores = defaultdict(defaultdict_factory)創建一個。

但是ast.literal_eval()不能用於已轉換爲字符串表示形式的字符串，因爲它不是函數支持的簡單字面數據類型之一。相反，我建議你考慮使用Python的老式pickle模塊，它可以處理大部分Python的內置數據類型以及像我所描述的自定義類。以下是將其應用於代碼的示例（與上面的代碼結合使用）：

import pickle 

try: 
    with open('scores', 'rb') as input_file: 
     players_scores = pickle.load(input_file) 
except FileNotFoundError: 
    print('new scores file will be created') 
    players_scores = defaultdict(defaultdict_factory) 

player_name = input("Enter new player's name: ").upper() 
players_scores[player_name]['GOLF'] = 0 
players_scores[player_name]['MON DEAL'] = 0 
# below is a shorter way to do the initialization for a new player 
# players_scores[player_name] = defaultdict_factory({'GOLF': 0, 'MON DEAL': 0}) 

# write new/updated data structure (back) to disk 
with open('scores', 'wb') as output_file: 
    pickle.dump(players_scores, output_file) 

print(players_scores)