If you need extreme focus to work with your code, that code is garbage.
Well written and well factored code is easy to get work done in, and you don't need to keep a zillion things in mind to navigate it.
Of course, reality is that the code most people work in is some level of garbage, and maybe there isn't much you can do about that. Either because you're not in charge, or you don't have the skills to improve the code.