Linux下使用python统计某个pid的进程的CPU利用率和MEM大小差值

# -*- coding:utf-8 -*-
import subprocess
import time
import sys
#支持命令行传参。传入你想测的pid值
pid = sys.argv[1]
# 第一个时间点的pid的进程信息
proc_info_1 = subprocess.check_output("pidstat -r -u -p {}".format(pid), shell=True).decode('utf-8')
time.sleep(3)
# 第二个时间点的pid的进程信息
proc_info_2 = subprocess.check_output("pidstat -r -u -p {}".format(pid), shell=True).decode('utf-8')

# 解析进程信息
def parse_process_info(info):
    lines = info.strip().split('\n')
    headers1 = lines[2].split()
    values1 = lines[3].split()
    headers2 = lines[5].split()
    values2 = lines[6].split()
    proc_data1 = dict(zip(headers1, values1))
    proc_data2 = dict(zip(headers2, values2))
    #print(proc_data1)
    return float(proc_data1['%CPU']), int(proc_data2['RSS'])

# 解析第一个时间点和第二个时间点的进程信息
cpu_1, mem_1 = parse_process_info(proc_info_1)
cpu_2, mem_2 = parse_process_info(proc_info_2)

# 计算差值
cpu_diff = cpu_2 - cpu_1
mem_diff = mem_2 - mem_1

# 打印结果
print("CPU利用率差值: {}%".format(cpu_diff))
print("内存占用差值: {}".format(mem_diff))