Mail stored in folders disappears using apple mail
Mail in apple mail folders disappears
Got to Settings > Passwords and Accounts > and see if the option for Mail Days to Sync can be set from No limit up to 1 Month. Choose the setting which best meets your need.