number of columns in 2d array python