诚信请教版上python大神一道题

本人本科是经济专业的 最近在自学data science 这一道题不知道做的对不对，感觉有更直接的做法，请大神们指点，谢谢
题目：

Write a function to calculate the marginal product of labor (MPL) numerically 我的做法：