关于工伤离职后公司是否需要补偿4个月工资的问题，不能一概而论，需要分情况讨论：
如果工伤离职是基于劳动合同到期或双方协商一致解除劳动合同，且劳动者被鉴定为五级至十级伤残的情况：
- 根据《工伤保险条例》的相关规定，劳动者有权获得一次性工伤医疗补助金和一次性伤残就业补助金。这两项补助金的具体标准由省、自治区、直辖市人民政府规定，并非直接等同于4个月的工资。
- 举例说明：假设某劳动者因工伤被鉴定为七级伤残，在劳动合同到期时离职，他/她将获得一次性工伤医疗补助金和一次性伤残就业补助金，但这两项补助金的总额并不一定是其4个月的工资，而是根据当地政府规定的标准来计算。
如果工伤离职是因为用人单位存在违法行为，如未提供劳动保护、未支付劳动报酬等，导致劳动者被迫离职的情况：
- 在这种情况下，劳动者除了可以享受工伤保险待遇外，还有权根据《劳动合同法》的规定，要求用人单位支付经济补偿。经济补偿的标准根据劳动者在用人单位的工作年限和月工资来确定，但同样并不直接等同于4个月的工资。
- 计算公式：经济补偿按劳动者在本单位工作的年限，每满一年支付一个月工资的标准向劳动者支付。六个月以上不满一年的，按一年计算；不满六个月的，向劳动者支付半个月工资的经济补偿。举例说明：如果劳动者在用人单位工作了3年，月工资为5000元，那么他/她因用人单位违法行为被迫离职时，应获得的经济补偿为15000元（3年×5000元/年），而非固定的4个月工资。
如果工伤离职是劳动者主动提出，且不存在用人单位违法行为的情况：
- 在这种情况下，劳动者通常只能享受工伤保险待遇，而无法获得额外的经济补偿或补助金。当然，如果双方在劳动合同或离职协议中有特别约定，那么应按照约定来执行。
总结与建议：
- 工伤离职后公司是否需要补偿4个月工资，取决于离职的原因、劳动者的伤残等级以及当地政府的规定。
- 如果您面临工伤离职的情况，建议首先了解自己的伤残等级和当地的工伤保险政策，以便准确计算自己应获得的补助金或经济补偿。
- 同时，如果您认为用人单位存在违法行为或双方在离职补偿方面存在争议，可以及时向我咨询，我会根据您的具体情况提供专业的法律建议和帮助。