avoiding 2d arrays in f2py... simpler indexing