Tabla de funciones en python

[gaby_287]

Hola a todos, soy nueva en python, anteriormente trabajaba con Mathematica y estoy tratando de cambiar algunos códigos a python. Tengo varias dudas que espero ir resolviendo poco a poco. La primera es que si tengo una tabla de datos y luego quiero hacer operaciones sobre ella de tal modo que se "defina" una función, cual es la forma correcta de hacerlo?, lo que quiero hacer es algo del estilo:

Código Python:

import numpy as np
from numpy import pi
 
tabla1_np = np.genfromtxt('Data/tabla', usecols=0) 
tabla2_np = np.genfromtxt('Data/tabla', usecols=1)
 
### Esta es la primera función (de dos variables que quiero definir)
 
def  velocidad(r0, rs):
        return (4*pi*r0*rs**3)*(-tabla1_np)/(tabla2_np + rs)
 
###Después ocuparía la función anterior para definir una nueva
 
def  suma2(r0, rs):
        return tabla2_np - velocidad(r0, rs)