python get number of columns in array