Golang Weighted Random Choice

This is a simple package for creating Weighted Random Choices.

Usage

To use WRC, import :

import "github.com/kontoulis/go-weighted-random-choice"

Instantiate a WeightedRandomChoice object by calling New(). New() takes optional argument precision which is by default 1000 and is used for the weight calibration in order to generate more precise results. The precision of 1000 will scale weights to a totalWeight = 1000, while keeping the weight ratios. e.g. weights 1, 2, 3 and 4 will become 100, 200, 300 and 400 respectively

wrc := WeightedRandomChoice.New()

Add elements one by one with name and weight.

wrc.AddElement("common", 59)
wrc.AddElement("epic", 1)
wrc.AddElement("rare", 10)
wrc.AddElement("green", 30)

You can also add multiple elements by passing a map[string]int in AddElements

wrc.AddElements(map[string]int{
    "common"    : 59,
    "epic"      : 1,
    "rare"      : 10,
    "green"     : 30,
})

Then you can call GetRandomChoice to retrieve a random weighted choice

choice := wrc.GetRandomChoice()