Converting a list of lists to a single list
Terry Reedy
tjreedy at udel.edu
Tue Jul 23 19:02:44 EDT 2013
On 7/23/2013 5:52 PM, steve at divillo.com wrote:
> I think that itertools may be able to do what I want but I have not
> been able to figure out how.
A recursive generator suffices.
> I want to convert an arbitrary number of lists with an arbitrary
> number of elements in each list into a single list as follows.
>
> Say I have three lists:
>
> [[A0,A1,A2], [B0,B1,B2] [C0,C1,C2]]
>
> I would like to convert those to a single list that looks like this:
>
> [A0,B0,C0,C1,C2,B1,C0,C1,C2,B2,C0,C1,C2,
> A1,B0,C0,C1,C2,B1,C0,C1,C2,B2,C0,C1,C2,
> A2,B0,C0,C1,C2,B1,C0,C1,C2,B2,C0,C1,C2]
def crossflat(lofl):
if lofl:
first = lofl.pop(0)
for o in first:
yield o
yield from crossflat(lofl.copy())
A0, A1, A2 = 100, 101, 102
B0, B1, B2 = 10, 11, 12
C0, C1, C2 = 0, 1, 2
LL = [[A0, A1, A2], [B0, B1, B2], [C0, C1, C2]]
cfLL = list(crossflat(LL))
print(cfLL)
assert cfLL == [
A0, B0, C0, C1, C2, B1, C0, C1, C2, B2, C0, C1, C2,
A1, B0, C0, C1, C2, B1, C0, C1, C2, B2, C0, C1, C2,
A2, B0, C0, C1, C2, B1, C0, C1, C2, B2, C0, C1, C2]
passes
--
Terry Jan Reedy
More information about the Python-list
mailing list