python--杂识--14--获取当前进程所占用的内存资源

import psutil

def get_process_memory_usage():
    """
    获取指定进程的内存使用情况（以MB为单位）
    """
    cmd = "python test.py".split()  # 需要根据所执行的命令修改
    for p in psutil.process_iter(['name', 'memory_info', 'cmdline']):
        if (p.info["cmdline"] is not None) and (len(p.info["cmdline"]) == 2):
            if p.info["cmdline"][0] == cmd[0] and p.info["cmdline"][1] == cmd[1]:
                mem_info = p.info['memory_info']
                return mem_info.rss / (1024 * 1024)


mem_usage = get_process_memory_usage()
print(mem_usage)