The general consensus is it is all to do with rhythm and comfort. Flavio Cobolli said at Wimbledon was "normal" for players to keep tennis balls in their shorts, adding it felt more "comfortable".