Dicts

This F# library provides:

• A dedicated Dict<'K,'V> type. It is a thin wrapper around Dictionary<'K,'V> with more functionality and nicer Error messages.

• A DefaultDict<'K,'V> type. It works like Python's defaultdict.
  By providing a default function in the constructor it will always return a value for any key.

• Extension methods for working with the IDictionary<'K,'V> interface.

It also works in JS and TS with Fable.

This library was designed for use with F# scripting. It follows the semantics of System.Collections.Generic.Dictionary, but a missing key never gives you a null or default value silently. Functions starting with try... return an F# Option (except TryGetValue, which keeps the .NET bool * value shape). Otherwise when a function fails on invalid input it will throw a descriptive exception.

I was always annoyed that a KeyNotFoundException does not include the actual bad key nor a pretty printed dictionary. This library fixes that in Dict.Get, the Dict indexer, Pop, the IDictionary extension GetValue and other item access functions.

Examples

All examples assume:

#r "nuget: Dicts"
open Dicts

Dict — Better error messages

Dict<'K,'V> is a thin wrapper around Dictionary<'K,'V> that gives descriptive exceptions when a key is missing, including the key, item count, and a pretty-printed dictionary.

// Create from key-value pairs (throws on duplicate keys, like the Dictionary constructor)
let d = Dict.create [ "a", 1; "b", 2; "c", 3 ]

d.["a"]          // 1
d.Get "b"        // 2
d.Set "d" 4      // adds or updates key "d"
d.Count          // 4

// Add works like Dictionary.Add: it throws if the key already exists.
d.Add("e", 5)    // adds key "e"
d.Add("e", 6)    // throws ArgumentException, use d.Set or d.["e"] <- 6 to replace a value

// Nicer error messages than System.Collections.Generic.Dictionary:
d.["z"]          // throws KeyNotFoundException:
                 // "Dict.get failed to find key "z" in seq [[a, 1]; [b, 2]; [c, 3]; [d, 4]; ...] of 5 items"

// Check for keys
d.ContainsKey "a"       // true
d.DoesNotContainKey "z" // true

// Use IsEmpty / IsNotEmpty
d.IsEmpty       // false
d.IsNotEmpty    // true

Dict — Pop and TryPop (Python-like)

let d = Dict.create [ "x", 10; "y", 20 ]

d.Pop "x"       // 10  (key "x" is removed from d)
d.Count          // 1

d.TryPop "y"     // Some 20  (key "y" is removed)
d.TryPop "y"     // None     (already removed, no exception)

Dict — Conditional set and defaults

let d = Dict.create [ "a", 1 ]

// Only sets if key is absent, returns true if it was set
d.SetIfKeyAbsent "a" 99  // false  (key "a" exists, value stays 1)
d.SetIfKeyAbsent "b" 42  // true   (key "b" is now 42)

// Get existing value, or create and store a default
d.GetOrSetDefaultValue 0 "c"   // 0  (key "c" is now set to 0)
d.GetOrSetDefaultValue 0 "a"   // 1  (key "a" already exists)

// Default from a function that receives the key
d.GetOrSetDefault (fun k -> k.Length) "hello"  // 5

DefaultDict — Auto-creating missing keys

DefaultDict<'K,'V> calls a default function whenever a missing key is accessed. This is inspired by Python's defaultdict.

// Count word occurrences using a mutable ref cell
let counter = DefaultDict<string, int ref>(fun _ -> ref 0)
for word in ["hi"; "world"; "hi"; "hi"] do
    incr counter.[word]

counter.["hi"].Value     // 3
counter.["world"].Value  // 1

// Group items by key
let groups = DefaultDict<string, ResizeArray<int>>(fun _ -> ResizeArray())
groups.["evens"].Add 2
groups.["odds"].Add  1
groups.["evens"].Add 4

groups.["evens"] |> Seq.toList  // [2; 4]
groups.["odds"]  |> Seq.toList  // [1]

Important: Accessing a missing key with Get or the indexer .[key] creates it. Use TryGetValue or ContainsKey to check without creating:

let dd = DefaultDict<string, int>(fun _ -> 0)
dd.ContainsKey "x"         // false  (does not create "x")
let ok, _ = dd.TryGetValue "x"  // ok = false  (does not create "x")
dd.["x"]                   // 0  (now "x" IS created with the default)
dd.ContainsKey "x"         // true

DefaultDict implements IEnumerable, ICollection and IReadOnlyCollection of KeyValuePair, but not IDictionary, because a TryGetValue that finds nothing while Get always returns a value would break the IDictionary contract. So the Dict module functions below don't take a DefaultDict. Use dd.InternalDictionary if you need an IDictionary.

Dict module — Functional-style operations

The Dict module provides functions that work on any IDictionary<'K,'V>, including plain Dictionary and Dict.

let d = Dict.create [ "a", 1; "b", 2; "c", 3 ]

// Functional get / set / add / tryGet
Dict.get "a" d            // 1
Dict.set "d" 4 d          // sets key "d" to 4 (adds or replaces)
Dict.add "d" 5 d          // throws ArgumentException, key "d" exists already (like Dictionary.Add)
Dict.tryGet "z" d         // None
Dict.tryGet "a" d         // Some 1

// Pop and tryPop
Dict.pop "d" d             // 4  (removes key "d")
Dict.tryPop "d" d          // None  (already removed)

// Conditional set
Dict.setIfKeyAbsent "a" 99 d  // false  (key "a" exists)
Dict.setIfKeyAbsent "e" 5 d   // true   (key "e" is now 5)

// Get or create a default
Dict.getOrSetDefaultValue 0 "f" d   // 0  (key "f" is now 0)

// Iteration
Dict.keys   d |> Seq.toList   // ["a"; "b"; "c"; "e"; "f"]
Dict.values d |> Seq.toList   // [1; 2; 3; 5; 0]
Dict.items  d |> Seq.toList   // [("a",1); ("b",2); ("c",3); ("e",5); ("f",0)]

Dict.iter (fun k v -> printfn "%s = %d" k v) d
Dict.map  (fun k v -> $"{k}:{v}") d |> Seq.toList

Dict.memoize — Cache function results

let expensiveComputation = Dict.memoize (fun n ->
    printfn "computing %d..." n
    n * n
)

expensiveComputation 5   // prints "computing 5...", returns 25
expensiveComputation 5   // returns 25 immediately, no print

The cache is a plain Dictionary, so don't call a memoized function from several threads at the same time.

IDictionary extensions

Extension methods available on any IDictionary<'K,'V> (including Dictionary<'K,'V>):

open Dicts.ExtensionsIDictionary

let d = System.Collections.Generic.Dictionary<string,int>()
d.["x"] <- 10; d.["y"] <- 20; d.["z"] <- 30

d.GetValue "x"           // 10  (with descriptive error on missing key)
d.SetValue "w" 40        // adds key "w"
d.Pop "w"                // 40  (removes key "w")
d.TryPop "w"             // None
d.Items      |> Seq.toList  // seq of (key, value) tuples
d.KeysSeq    |> Seq.toList  // seq of keys
d.ValuesSeq  |> Seq.toList  // seq of values
d.DoesNotContainKey "w"  // true

Pretty printing

All types provide readable string representations, in the same format as ResizeArrayT:

let d = Dict.create [ "name", "Alice"; "city", "Zurich"; "lang", "F#" ]

d.ToString()    // "Dict<String,String> with 3 items"
d.AsString      // "Dict<String,String> with 3 items:\n  name: Alice\n  city: Zurich\n  lang: F#\n"
d.ToString(1)   // "Dict<String,String> with 3 items:\n  name: Alice\n  ...\n  lang: F#\n"

AsString shows up to 5 entries, ToString(n) up to n entries. If entries are left out, ... and the last entry follow. Each key and value is shown on one line and cut off after 200 characters. In Fable ToString() shows the generic parameters as Dict<'K,'V>, while AsString and ToString(n) show the actual type names.

Full API Documentation

goswinr.github.io/Dicts

Tests

All Tests run in both javascript and dotnet. They are written with Scriptorium, so the same tests run unchanged on both. Successful Fable compilation to typescript is verified too. Go to the tests folder:

cd Tests

For testing with .NET:

dotnet run

for JS testing via Fable and TS verification:

dotnet tool restore
npm ci
npm test

License

MIT

Changelog

see CHANGELOG.md