python 2d array get first column