python sort 2d array by first column