Looking for a more elegant and sophisticated solution when multiple if and for-loop are used

Question

I am beginner/intermediate user working with python and when I write elaborate code (at least for me), I always try to rewrite it looking for reducing the number of lines wh