How to kill a process using process name in python script

Question

My requirement is to kill a process. I have the process name. Below is my code:

def kill_process(name):
  os.system(f"TASKKILL /F /IM {name}")