In my doctoral thesis, I looked at those who went back to work and those who stayed at home. That was mothers in the 1980s. It was mostly fathers who went back to work or who were working.
Those who went back to work soon functioned better than those who stayed at home, but that depends on the work environment. You need to have a work environment where you have the possibility to take breaks, and that you're well cared for in that system. If not, it's worse to go back to work. It's very individual.
My advice would be to return to work as soon as possible, but adjust the work. It depends on what kind of profession you have. There are a lot of things that go into that.