[python] Extracting first n columns of a numpy matrix