Using Dictionaries with numba njit function

by

The support for Dictionary has now been added in Numba version 43.0. Although it quite limited (does not support list and set as key/values). You can however read the updated documentation here for more info.

Here is an example

import numpy as np
from numba import njit
from numba import types
from numba.typed import Dict

# First create a dictionary using Dict.empty()
# Specify the data types for both key and value pairs

# Dict with key as strings and values of type float array
dict_param1 = Dict.empty(
    key_type=types.unicode_type,
    value_type=types.float64[:],
)

# Dict with keys as string and values of type float
dict_param2 = Dict.empty(
    key_type=types.unicode_type,
    value_type=types.float64,
)

# Type-expressions are currently not supported inside jit functions.
float_array = types.float64[:]

@njit
def add_values(d_param1, d_param2):
    # Make a result dictionary to store results
    # Dict with keys as string and values of type float array
    result_dict = Dict.empty(
        key_type=types.unicode_type,
        value_type=float_array,
    )

    for key in d_param1.keys():
      result_dict[key] = d_param1[key] + d_param2[key]

    return result_dict

dict_param1["hello"]  = np.asarray([1.5, 2.5, 3.5], dtype="f8")
dict_param1["world"]  = np.asarray([10.5, 20.5, 30.5], dtype="f8")

dict_param2["hello"]  = 1.5
dict_param2["world"]  = 10

final_dict = add_values(dict_param1, dict_param2)

print(final_dict)
# Output : {hello: [3. 4. 5.], world: [20.5 30.5 40.5]}

Link to Google colab notebook.

References:
– https://github.com/numba/numba/issues/3644
– https://numba.pydata.org/numba-doc/dev/reference/pysupported.html#dict

More Related Contents:

  1. Remove string in python dictionary value
  2. python building coder for ceasar cipher
  3. I don’t know how to display step 3 for this assignment
  4. Python 2.7.10: Using LAMBDA and LOGICAL operators in DICT to choose a case
  5. How to make a dict from n’th indices of tuples in list
  6. Create a dictionary with comprehension
  7. Loop through all nested dictionary values?
  8. How to use a dot “.” to access members of dictionary?
  9. Alternative to dict comprehension prior to Python 2.7
  10. How do I convert this list of dictionaries to a csv file?
  11. Multi-level defaultdict with variable depth?
  12. How to index into a dictionary?
  13. Sorting a dictionary by value then key
  14. How can I initialize a dictionary of distinct empty lists in Python?
  15. How to add multiple values to a dictionary key?
  16. How do I sort a dictionary by key?
  17. Appending a dictionary to a list in a loop
  18. Calculate difference in keys contained in two Python dictionaries
  19. Python: Collections.Counter vs defaultdict(int)
  20. How do I delete items from a dictionary while iterating over it?
  21. Counting each letter’s frequency in a string
  22. Collections.defaultdict difference with normal dict
  23. Right way to initialize an OrderedDict using its constructor such that it retains order of initial data?
  24. Using Python class as a data container [closed]
  25. How can I convert string values from a dictionary, into int/float datatypes?
  26. Rename columns in Pandas based on a dictionary
  27. sorting list of nested dictionaries in python
  28. How do I send list of dictionary as Body parameter together with Files in FastAPI?
  29. How to convert only numbers in a mixed list into float
  30. In what case would I use a tuple as a dictionary key? [closed]

Leave a Comment