Babylonian Talmud, Tractate Qiddushin

בשביל חטא יחידי שעשה זה אבד ממנו ומכל העולם טובה הרבה ר"ש בן יוחי אומר אפילו צדיק גמור כל ימיו ומרד באחרונה איבד את הראשונות שנאמר צדקת הצדיק לא תצילנו ביום פשעו ואפילו רשע גמור כל ימיו ועשה תשובה באחרונה אין מזכירים לו שוב רשעו שנאמר ורשעת הרשע לא יכשל בה ביום שובו מרשעו וניהוי כמחצה עונות ומחצה זכיות אמר ריש לקיש בתוהא על הראשונות

גמ אמר רבי אלעזר ברבי צדוק למה צדיקים נמשלים בעולם הזה לאילן שכולו עומד במקום טהרה ונופו נוטה למקום טומאה נקצץ נופו כולו עומד במקום טהרה כך הקב"ה מביא יסורים על צדיקים בעולם הזה כדי שיירשו העולם הבא שנאמר והיה ראשיתך מצער ואחריתך ישגה מאד ולמה רשעים דומים בעולם הזה לאילן שכולו עומד במקום טומאה ונופו נוטה למקום טהרה נקצץ נופו כולו עומד במקום טומאה כך הקב"ה משפיע להן טובה לרשעים בעולם הזה כדי לטורדן ולהורישן למדריגה התחתונה שנאמר יש דרך ישר לפני איש ואחריתה דרכי מות