Combining for-loops in the same line python without changing the object type in the dict