get first column of 2d array python