Getting error “invalid value encountered in double_scalars” in simple function

Question

I\'m relatively new to python and I\'ve been trying to do some of my university work on it to get used to the language. I came up with this problem and I saw some other people a