從一個檔案夾中的多個檔案加載json并將它們放入一個字典（或串列）中

我有多個文本檔案，里面有這個（具有不同的值）：

{"abandon": {"R1F2V0CQAYJOZZ": 2, "R3NUFWQ9SPGVJO": 1}, "abduct": {"R1F2V0CQAYJOZZ": 1, "R3376OQSHCTV1A": 1, "R14BW4EQZNVKKG": 1, "R233CMES8RCOCU": 1},

如果我在線格式化它會變成這樣：

   "abandon":{
      "R1F2V0CQAYJOZZ":2,
      "R3NUFWQ9SPGVJO":1
   },
   "abduct":{
      "R1F2V0CQAYJOZZ":1,
      "R3376OQSHCTV1A":1,
      "R14BW4EQZNVKKG":1,
      "R233CMES8RCOCU":1
   },

這個 JSON 的意思是：

"word":{
   "Document name":"value"
},

但是在不同的檔案中有重復的單詞。我想要做的是：讀取所有檔案并將所有內容存盤在一本字典中，但是：

1. 如果字典中存在“單詞”，則檢查“檔案”是否存在；
2. 如果“檔案存在”，則增加“值”，否則將檔案放在那里并且“值= 1”
3. 如果“word”不存在，則存盤“word”、“document”和“value = 1”

編輯：

所以想象一下這兩個檔案：

File1.txt = {"abandon": {"doc1": 2, "doc2": 1}, "abduct": {"doc1": 1, "doc2": 1, "doc8": 1},

File1.txt = {"abandon": {"doc1": 1, "doc3": 1}, "abduct": {"doc5": 1, "doc8": 1},

我希望我的字典以這樣的方式結束：

{"abandon": {"doc1": 3, "doc2": 1, "doc3": 1}, "abduct": {"doc1": 1, "doc2": 1, "doc5": 1, "doc8": 2},

EDIT2：它也可以是嵌套串列

uj5u.com熱心網友回復：

IIUC，嘗試：

import os
import json

files = [f for f in os.listdir() if f.endswith(".txt")]
result = dict()

for file in files:
    d = json.load(open(file))
    for word in d:
        if word not in result:
            result[word] = dict()
        for doc in d[word]:
            if doc not in result[word]:
                result[word][doc] = d[word][doc]
            else:
                result[word][doc]  = d[word][doc]

>>> result
{'abandon': {'doc1': 3, 'doc2': 1, 'doc3': 1},
 'abduct': {'doc1': 1, 'doc2': 1, 'doc8': 2, 'doc5': 1}}

輸入檔案：

檔案1.txt：

{"abandon": {"doc1": 2, "doc2": 1}, "abduct": {"doc1": 1, "doc2": 1, "doc8": 1}}

檔案2.txt：

{"abandon": {"doc1": 1, "doc3": 1}, "abduct": {"doc5": 1, "doc8": 1}}

uj5u.com熱心網友回復：

import json

files = ["input", "list", "of", "files"]
outDict = {}
for file in files:  # iterate over the files
    with open(file) as fn:
        newDict = json.load(fn)
    for word in newDict:  # iterate over each word from the file
        inWord = newDict[word]
        outWord = outDict.get(word, {})  # returns an empty dict if word isn't already in the output dictionary
        for docName in inWord:  # iterate over each document name from the file
            value = outWord.get(docName, 0)  # returns 0 if the document name isn't already in the output dictionary
            value  = 1  # increment the value
            outWord[docName] = value  # update the output dictionary

uj5u.com熱心網友回復：

合并使用很簡單.setdefault：

import json
import glob

merged = {}

for file in glob.glob('*.txt'):  # match *.txt files in current directory

    with open(file) as f:
        in_dict = json.load(f)

        for word, docs in in_dict.items():
            for doc, value in docs.items():
                merged.setdefault(word,{})       # create word with empty dict value if it doesn't exist
                merged[word].setdefault(doc, 0)  # create value of 0 for document if it doesn't exist
                merged[word][doc]  = value       # add the doc's value.

print(json.dumps(merged,indent=2))

或使用defaultdict. defaultdict 的引數必須是回傳默認值的函式，因此 lambda 回傳默認整數字典：

import json
import glob
from collections import defaultdict

merged = defaultdict(lambda: defaultdict(int))

for file in glob.glob('*.txt'):  # match *.txt files in current directory

    with open(file) as f:
        in_dict = json.load(f)

        for word, docs in in_dict.items():
            for doc,value in docs.items():
                merged[word][doc]  = value       # add the doc's value.

print(json.dumps(merged,indent=2))

標籤：Python json

上一篇：如何使用Flutter決議動態JSON？

下一篇：獲取字串MSAccessSQL查詢中第一個字母后的子字串