any way to manipulate 3 dim array with taylor series in numpy?

Question

I tried to manipulate 2 dim numpy array with taylor series and it works fine. Now I have trouble manipulating the same things that worked for 2 dim array to 3 dim arrays, but it