python zip() function for a matrix,zip(*A)