這是我的 sql 表（2700 萬行）的導出示例，如下所示：test.json{ "table": "table", "rows": [ { "state": "UNKNOWN" }, { "state": "IE" } ]}這是我試圖用來在 elasticsearch 中加載表的生成器函數filename='test.json'def load_json(filename): with open(filename,'r') as open_file: yield json.load(open_file)['rows']我收到以下錯誤：TypeError: pop() takes at most 1 argument (2 given)在我的生成器函數上運行next(load_json(filename))時，這是輸出。[{'state': 'UNKNOWN'}, {'state': 'IE'}]如您所見，它正在返回一個list，這是導致錯誤的原因。有人可以解釋我如何正確訪問我的 json 以及為什么我的方法不正確。
2 回答
婷婷同學_
TA貢獻1844條經驗 獲得超8個贊
你產生了一個列表，所以你不能使用dict.pop. 在您的情況下，您應該這樣編寫代碼：
def load_json(file_name):
with open(filename,'r') as open_file:
for row in json.load(open_file)['rows']:
yield row
這意味著，一旦你加載了一個新行，你將跳轉到其他函數來處理該值。然后你可以在下面彈出。
for row in load_json(file_name):
print(row.pop('state'))
但是，還是要告訴你，這種方式并不能優化你的代碼，因為json.load會一次性讀取整個文件，所以還是會消耗很大的內存...
你說你用的是 elasticsearch，那你最好用 pagination 本身，而不是 generator。
揚帆大魚
TA貢獻1799條經驗 獲得超9個贊
您很可能希望遍歷行并像這樣一一輸出它們：
filename='test.json'
def load_json(filename):
with open(filename,'r') as open_file:
for r in json.load(open_file)['rows']:
yield r
目前你只產生一件事——整個列表。因此，由您的原始函數創建的迭代器將返回一個完整列表作為第一個元素，然后停止。
添加回答
舉報
0/150
提交
取消