Lua中是否有Python的defaultdict功能

Question

Lua中是否有類似於Python中可用的collections.defaultdict的功能，它可以自動處理不存在的關聯數組鍵的默認值？

我希望下面的代碼將nil設置爲v而不是錯誤。所以基本上辦法a[2]（不存在的鍵）是table默認：

a = {} 
v = a[2][3] 

>>> PANIC: unprotected error in call to Lua API (main.lua:603: attempt to index field '?' (a nil value)) 

在Python這是可以做到這樣的：

>>> import collections 
>>> a = collections.defaultdict(dict) 
>>> print a[2] 
{} 

Answer 1

有一個Lua標準函數來做到這一點？不，但你可以用metatables輕鬆完成。你甚至可以編寫一個函數來創建這樣的表：

function CreateTableWithDefaultElement(default) 
    local tbl = {} 
    local mtbl = {} 
    mtbl.__index = function(tbl, key) 
    local val = rawget(tbl, key) 
    return val or default 
    end 
    setmetatable(tbl, mtbl) 
    return tbl 
end 

注意，每個元素將得到相同默認值。因此，如果您將默認值設置爲表格，則返回表格中的每個「空」元素都將有效地引用同一個表格。如果這不是你想要的，你將不得不修改該功能。

Answer 2

只是以爲我會分享我的這段代碼的版本，如果有人發現這個需要與表，對象等等兼容的版本。而且，與Nicol的解決方案相反，這實際上在表格中創建了請求的條目。

function defaultdict(default_value_factory) 
    local t = {} 
    local metatable = {} 
    metatable.__index = function(t, key) 
     if not rawget(t, key) then 
      rawset(t, key, default_value_factory(key)) 
     end 
     return rawget(t, key) 
    end 
    return setmetatable(t, metatable) 
end 

用法示例：

d = defaultidct(function() return {} end) 
table.insert(d["people"], {"Bob", "The Builder"}) 

names = defaultdict(function(key) return key end) 
print(names["bob"]) -- bob 
names["bob"] = "bob the builder" 
names["ashley"] = "ashley the fire princess"