mirror/batteries
1
0
Fork 0
mirror of https://github.com/1bardesign/batteries.git synced 2024-11-22 14:14:36 +00:00
Code Issues Packages Projects Releases Wiki Activity

Add a weighted random pick for tablex

Browse Source
This commit is contained in:
Jesse Viikari 2021-11-17 20:20:04 +02:00
parent 63a1cf5438
commit 7eb6bcc576
1 changed files with 20 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

20
tablex.lua
View File

@ -202,6 +202,26 @@ function tablex.take_random(t, r)
return table.remove(t, tablex.random_index(t, r)) return table.remove(t, tablex.random_index(t, r))
end end
--return a random index based on weights provided
--Example: { 0.3, 1, 6, 0.5 } (3rd index most likely)
function tablex.weighted_random(t, r)
if #t == 0 then
return nil
end
local sum = 0
for _, weight in ipairs (t) do
sum = sum + weight
end
local rnd = _random(nil, nil, r) * sum
sum = 0
for i, weight in ipairs (t) do
sum = sum + weight
if rnd <= sum then
return i, weight
end
end
end
--shuffle the order of a table --shuffle the order of a table
function tablex.shuffle(t, r) function tablex.shuffle(t, r)
for i = 1, #t do for i = 1, #t do