Skip to main content

A tool for rounding numbers in complex Python objects

Project description

rounder: Rounding of numbers in complex Python objects

rounder is a lightweight package for rounding numbers in complex Python objects, such as dictionaries, lists, tuples, and sets, and any complex object that combines any number of such objects in any nested structure; you can also use it for instances of classes whose attributes contain numbers. The code is organized as a Python (Python >= 3.6 is required) package that can be installed from PyPi (pip install rounder), but as it is a one-file package, you can simply download its main module (rounder.py) and use it directly in your project.

The package is useful mainly for presentation purposes, but in some cases, it can be useful in other situations as well.

rounder offers you four functions for rounding objects:

  • round_object(obj, digits=0, use_copy=False), which rounds all numbers in obj to digits decimal places
  • floor_object(obj, use_copy=False), which rounds all numbers in obj down to the nearest integer
  • ceil_object(obj, use_copy=False), which rounds all numbers in obj up to the nearest integer
  • signif_object(obj, digits, use_copy=False), which rounds all numbers in obj to digits significant digits

In addition, rounder comes with a generalized function:

  • map_obj(func, obj, use_copy=False), which runs callable func, which takes a number as an argument and returns a number, to all numbers across the object.

rounder also offers a function for rounding numbers to significant digits:

  • signif(x, digits), which rounds x (either an int or a float) to digits significant digits

You can use signif in a simple way:

>>> import rounder as r
>>> r.signif(1.1212, 3)
1.12
>>> r.signif(12.1239112, 5)
12.124
>>> r.signif(121212.12, 3)
121000.0

The package is simple to use, but you have to remember that when you're working with mutable objects, such as dicts or lists, rounding them will affect the original object; no such effect, of course, will occur for immutable types (e.g., tuples and sets). To overcome this effect, simply use use_copy=True in the above functions (not in signif). If you do so, the function will create a copy of the object and work on (and return) its deepcopy, not the original object.

You can use rounder functions for rounding floats, but do remember that their behavior is slightly different than that of their builtin and math counterparts, as they do not throw an exception when a non-number object is used.

You can round a list, a tuple, a set (including a frozenset), a double array.array, a dict, and even a class instance:

>>> r.round_object([1.122, 2.4434], 1)
[1.1, 2.4]
>>> r.ceil_object([1.122, 2.4434])
[2, 3]
>>> r.floor_object([1.122, 2.4434])
[1, 2]
>>> r.signif_object([1.1224, 222.4434], 4)
[1.122, 222.4]

>>> r.round_object((1.122, 2.4434), 1)
(1.1, 2.4)
>>> r.round_object({1.122, 2.4434}, 1)
{1.1, 2.4}
>>> r.round_object({"1": 1.122, "q":2.4434}, 1)
{'1': 1.1, 'q': 2.4}

>>> import array
>>> arr = array.array("d", (1.122, 2.4434))
>>> r.round_object(arr, 1)
array('d', [1.1, 2.4])

As mentioned above, you can use rounder functions also for class instances:

>>> class ClassWithNumbers:
...     def __init__(self, x, y):
...         self.x = x
...         self.y = y
>>> inst = ClassWithNumbers(
...     x = 20.22045,
...     y={"list": [34.554, 666.777],
...     "tuple": (.111210, 343.3333)}
... )

>>> inst_copy = r.round_object(inst, 1, True)
>>> inst_copy.x
20.2
>>> inst_copy.y
{'list': [34.6, 666.8], 'tuple': (0.1, 343.3)}
>>> id(inst) != id(inst_copy)
True

>>> inst.x
20.22045
>>> inst_no_copy = r.floor_object(inst, False)
>>> id(inst) == id(inst_no_copy)
True
>>> inst.x
20

You can of course round a particular attribute of the class instance:

>>> _ = r.round_object(inst_copy.y, 0, False)
>>> inst_copy.y
{'list': [35.0, 667.0], 'tuple': (0.0, 343.0)}

Note that you do not have to worry about having non-roundable objects in the object fed into the rounder functions. Your objects can contain objects of any type; numbers will be rounded while all other objects will be remain untouched:

>>> r.round_object([1.122, "string", 2.4434, 2.45454545-2j], 1)
[1.1, 'string', 2.4, (2.5-2j)]

In fact, you can round any object, and the function will simply return it if it cannot be rounded:

>>> r.round_object("string")
'string'
>>> r.round_object(lambda x: x**3)(2)
8
>>> r.round_object(range(10))
range(0, 10)

But most of all, you can apply rounding for any complex object, of any structure. Imagine you have a structure like this:

>>> x = {
...     "items": ["item 1", "item 2", "item 3",],
...     "quantities": {"item 1": 235, "item 2" : 300, "item 3": 17,},
...     "prices": {
...         "item 1": {"$": 32.22534554, "EURO": 41.783234567},
...         "item 2": {"$": 42.26625, "EURO": 51.333578},
...         "item 3": {"$": 2.223043225, "EURO": 2.78098721346}
...     },
...     "income": {
...         "2009": {"$": 3445342.324364, "EURO":   39080.332546},
...         "2010": {"$": 6765675.56665554, "EURO": 78980.34564546},
...     }
... }

To round all the values in this structure, you would need to build a dedicated script for that. With rounder, this is a piece of cake:

>>> rounded_x = r.round_object(x, digits=2, use_copy=True)

And you will get this:

>>> from pprint import pprint
>>> pprint(rounded_x)
{'income': {'2009': {'$': 3445342.32, 'EURO': 39080.33},
            '2010': {'$': 6765675.57, 'EURO': 78980.35}},
 'items': ['item 1', 'item 2', 'item 3'],
 'prices': {'item 1': {'$': 32.23, 'EURO': 41.78},
            'item 2': {'$': 42.27, 'EURO': 51.33},
            'item 3': {'$': 2.22, 'EURO': 2.78}},
 'quantities': {'item 1': 235, 'item 2': 300, 'item 3': 17}}

Note that we used use_copy=True, which means that rounded_x is a deepcopy of x, so the original dictionary has not been affected anyway.

map_object

In addition, rounder offers you a map_object() function, which enables you to run any function that takes a number and returns a number for all numbers in an object. This works like the following:

>>> xy = {
...     "x": [12, 33.3, 45.5, 3543.22],
...     "y": [.45, .3554, .55223, .9911],
...     "expl": "x and y values"
... }
>>> r.round_object(
...     r.map_object(
...         lambda x: x**3/(1 - 1/x),
...         xy,
...         use_copy=True),
...     4,
...     use_copy=True
... )
{'x': [1885.0909, 38069.258, 96313.1475, 44495587353.9829], 'y': [-0.0746, -0.0248, -0.2077, -108.4126], 'expl': 'x and y values'}

You would have achieved the same result had you used round inside the lambda body:

>>> r.map_object(lambda x: round(x**3/(1 - 1/x), 4), xy, use_copy=True)
{'x': [1885.0909, 38069.258, 96313.1475, 44495587353.9829], 'y': [-0.0746, -0.0248, -0.2077, -108.4126], 'expl': 'x and y values'}

The latter approach, actually, will be quicker, as the full recursion is used just once (by r.map_object()), not twice, as it was done in the former example (first, by r.map_object(), and then by r.round_object()).

If the function takes additional arguments, you can use a wrapper function to overcome this issue:

>>> def forget(something): pass
>>> def fun(x, to_forget):
...     forget(to_forget)
...     return x**2
>>> def wrapper(x):
...     return fun(x, "this can be forgotten")
>>> r.map_object(wrapper, [2, 2, [3, 3, ], {"a": 5}])
[4, 4, [9, 9], {'a': 25}]

Or even:

>>> r.map_object(
...     lambda x: fun(x, "this can be forgotten"),
...     [2, 2, [3, 3, ], {"a": 5}]
... )
[4, 4, [9, 9], {'a': 25}]

Examples

First of all, all these functions will work the very same way as their original counterparts (not for signif, which does not have one):

>>> import math
>>> x = 12345.12345678901234567890
>>> for d in range(10):
...     assert round(x, d) == r.round_object(x, d)
...     assert math.ceil(x) == r.ceil_object(x)
...     assert math.floor(x) == r.floor_object(x)

Immutable types

rounder does work with immutable types! It simply creates a new object, with rounded numbers:

>>> x = {1.12, 4.555}
>>> r.round_object(x)
{1, 5}
>>> r.round_object(frozenset(x))
frozenset({1, 5})
>>> r.round_object((1.12, 4.555))
(1, 5)
>>> r.round_object(({1.1, 1.2}, frozenset({1.444, 2.222})))
({1}, frozenset({1, 2}))

So, note that it makes no difference whether you use True or False for use_copy, as with immutable types rounder will create a copy anyway.

Remember, however, that in the case of sets, you can get a shorter set then the original one:

>>> x = {1.12, 1.99}
>>> r.ceil_object(x)
{2}

Generators and other unpickable objects

This should be an extremely rare situation to request to round an object that contains a generator, or any other unpickable object. But if you happen to be in such a situation, be aware of some limitations of rounder functions.

As a rule, mainly for safety, generators are returned unchanged. This is a safe approach for the simple reason that you often choose to use a generator instead of, say, a list when the data you're processing can be too large for your machine's memory to handle. So:

>>> gen = (i**2 for i in range(10))
>>> rounded_gen = r.round_object(gen)
>>> rounded_gen is gen
True
>>> next(gen)
0
>>> next(gen)
1
>>> next(rounded_gen)
4

You will get the same result for range:

>>> ran = range(10)
>>> r.round_object(ran) is ran
True

But you have to remember that when you request a deepcopy (with use_copy=True), all elements to be rounded need to be pickable:

>>> gen_2 = (i**2 for i in range(10))
>>> gen_2_copied_rounded = r.round_object(gen_2, use_copy=True)
Traceback (most recent call last):
    ...
rounder.rounder.UnpickableObjectError

range() is pickable, so you can request a deepcopy of it in rounder functions.

This is a rare situation, however, to include unpickable objects in an object to be rounded. Remember about the above limitations, and you can either work with the original object (not its copy, so with default use_copy=False), or change it so that all its elements can be pickled.

List of types handled by rounder

  • int
  • float
  • complex
  • set
  • frozenset
  • list
  • tuple
  • collections.namedtuple
  • typing.NamedTuple
  • dict
  • OrderedDict
  • array.array

NumPy and Pandas

rounder does not work with numpy and pandas: they have their own builtin methods for rounding, and using them will be much quicker.

Testing

The package is covered with unit pytests, located in the tests/ folder. In addition, the package uses doctests, which are collected here, in this README, and in the main module, rounder.py. These doctests serve mainly documentation purposes, and since they can be run any time during development and before each release, they help to check whether all the examples are correct and work fine.

OS

The package is OS-independent. Its releases are checked in local machines, on Windows 10 and Ubuntu 20.04 for Windows, and in Pythonista for iPad.

Project details


Download files

Download the file for your platform. If you're not sure which to choose, learn more about installing packages.

Source Distribution

rounder-0.3.2.tar.gz (12.8 kB view hashes)

Uploaded Source

Built Distribution

rounder-0.3.2-py3-none-any.whl (12.2 kB view hashes)

Uploaded Python 3

Supported by

AWS AWS Cloud computing and Security Sponsor Datadog Datadog Monitoring Fastly Fastly CDN Google Google Download Analytics Microsoft Microsoft PSF Sponsor Pingdom Pingdom Monitoring Sentry Sentry Error logging StatusPage StatusPage Status page