Skip to content

Instantly share code, notes, and snippets.

@Jim-Holmstroem
Created July 18, 2012 10:34
Show Gist options
  • Save Jim-Holmstroem/3135464 to your computer and use it in GitHub Desktop.
Save Jim-Holmstroem/3135464 to your computer and use it in GitHub Desktop.
itertools.foo(*A)
from itertools import *
def matrix_range(N,M):
return map(lambda i: range(i,i+M),range(0,N*M,M)) #Create a MxN matrix (for testing)
A=matrix_range(5,3)
At=zip(*A) #transpose
At2=map(list,zip(*A)) #transpose without the tuple problems (slower)
At3=map(list,izip(*A)) #could be faster for big lists
a=chain(*A) #flatten to vector, lazyversion: http://docs.python.org/library/itertools.html#itertools.chain.from_iterable
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment