how to get memory usage of specific function in python code example

Example: evaluate how much a python program memory

ps -m -o %cpu,%mem,command