python get columns of 2d list