Logo Questions Linux Laravel Mysql Ubuntu Git Menu
 

What is the simple way to merge named tuples in Python?

I want to merge two namedtuples without loosing the key names. If, I just do a merge with '+' operator I am getting a tuple as a result but without the names.

For instance:

n [1]: from collections import namedtuple

In [2]: A = namedtuple("A", "a b c")

In [4]: B = namedtuple("B", "d e")

In [5]: a = A(10, 20, 30)

In [6]: b = B(40, 50)

In [7]: a + b
Out[7]: (10, 20, 30, 40, 50)

As you can see in the above case, the result of a + b has no names associated with them.

But, I am able to achieve it by creating a third namedtuple, which has fields from both A and B.

In [8]: C = namedtuple("C", A._fields + B._fields)

In [9]: C(*(a + b))
Out[9]: C(a=10, b=20, c=30, d=40, e=50)

Is this the right way or is there a better way to do this?

like image 958
Senthil Babu Avatar asked Aug 28 '12 14:08

Senthil Babu


2 Answers

Some observations:

  • In general Python would not know what to do when you try to merge two namedtuples that happen to have fields with the same name. Perhaps this is why there is no operator or function for this.

  • The documentation of _fields says:

Tuple of strings listing the field names. Useful for introspection and for creating new named tuple types from existing named tuples.

This suggests your approach is fine and perhaps even hinted at by the authors of the namedtuple code.

like image 50
Simeon Visser Avatar answered Nov 15 '22 14:11

Simeon Visser


You've pretty much nailed it as far as vanilla Python is concerned, but there's an extra simplification you can do if you're using Python 3.5+.

>>> from collections import namedtuple
>>> A = namedtuple("A", "a b c")
>>> B = namedtuple("B", "d e")
>>> a = A(10, 20, 30)
>>> b = B(40, 50)
>>> C = namedtuple("C", A._fields + B._fields)
>>> C(*(a + b))
C(a=10, b=20, c=30, d=40, e=50)
>>> #Available in Python 3.5+
>>> C(*a, *b)
C(a=10, b=20, c=30, d=40, e=50)

Also, here's a function you can use to eliminate boilerplate code if you find yourself doing this frequently:

>>> from functools import reduce
>>> from itertools import chain
>>> from operator import add
>>> def namedtuplemerge(*args):
...     cls = namedtuple('_'.join(arg.__class__.__name__ for arg in args), reduce(add, (arg._fields for arg in args)))
...     return cls(*chain(*args))
...
>>> namedtuplemerge(a, b)
A_B(a=10, b=20, c=30, d=40, e=50)
like image 26
John Crawford Avatar answered Nov 15 '22 16:11

John Crawford