51 |
52 |
Source code for clintrials.coll
53 | __author__ = 'Kristian Brock'
54 | __contact__ = 'kristian.brock@gmail.com'
55 |
56 | """ Functions and classes for manipulating collections. """
57 |
58 |
59 | [docs]def to_1d_list_gen(x):
60 |
""" Generator function to reduce lists of lists of arbitrary depth (and scalars) to single depth-1 list.
61 |
62 |
.. note:: this function is recursive.
63 |
64 |
"""
65 |
66 |
if isinstance(x, list):
67 |
for y in x:
68 |
for z in to_1d_list_gen(y):
69 |
yield z
70 |
else:
71 |
yield x
72 |
73 |
74 | [docs]def to_1d_list(x):
75 |
""" Reshape scalars, lists and lists of lists of arbitrary depth as a single flat list, i.e. list of depth 1.
76 |
77 |
.. note:: this function basically offloads all its work to a generator function because **we like yield**!
78 |
79 |
E.g.
80 |
81 |
>>> to_1d_list(0)
82 |
[0]
83 |
>>> to_1d_list([1])
84 |
[1]
85 |
>>> to_1d_list([[1,2],3,[4,5]])
86 |
[1, 2, 3, 4, 5]
87 |
>>> to_1d_list([[1,2],3,[4,5,[6,[7,8,[9]]]]])
88 |
[1, 2, 3, 4, 5, 6, 7, 8, 9]
89 |
90 |
"""
91 |
return list(to_1d_list_gen(x))
92 |
93 |
94 |
