On Linux, the command ps aux outputs a list of processes with multiple columns for each stat. e.g.
USER PID %CPU %MEM VSZ RSS TTY STAT START
Why don't you use PSI instead? PSI provides process information on Linux and other Unix variants.
import psi.process for p in psi.process.ProcessTable().values(): …