python - 如何通过进程名获取PID？

Question

有什么方法可以在 Python 中通过进程名获取 PID？

  PID USER      PR  NI  VIRT  RES  SHR S  %CPU %MEM    TIME+  COMMAND                                                                                        
 3110 meysam    20   0  971m 286m  63m S  14.0  7.9  14:24.50 chrome 

例如我需要通过chrome获取3110。

Answer 1

您可以通过 subprocess.check_output 使用 pidof 按名称获取进程的 pid :

from subprocess import check_output
def get_pid(name):
    return check_output(["pidof",name])


In [5]: get_pid("java")
Out[5]: '23366\n'

check_output(["pidof",name]) 将作为 "pidof process_name" 运行命令，如果返回码不为零，则会引发一个 CalledProcessError。

处理多个条目并转换为整数:

from subprocess import check_output
def get_pid(name):
    return map(int,check_output(["pidof",name]).split())

在 [21] 中:get_pid("chrome")

Out[21]: 
[27698, 27678, 27665, 27649, 27540, 27530, 27517, 14884, 14719, 13849, 13708, 7713, 7310, 7291, 7217, 7208, 7204, 7189, 7180, 7175, 7166, 7151, 7138, 7127, 7117, 7114, 7107, 7095, 7091, 7087, 7083, 7073, 7065, 7056, 7048, 7028, 7011, 6997]

或者通过 -s 标志来获取单个 pid:

def get_pid(name):
    return int(check_output(["pidof","-s",name]))

In [25]: get_pid("chrome")
Out[25]: 27698