工作中需要根據(jù)某個(gè)應(yīng)用程序具體吃了多少內(nèi)存來(lái)決定執(zhí)行某些操作，所以需要寫(xiě)個(gè)小工具來(lái)模擬應(yīng)用程序使用內(nèi)存情況，下面是我寫(xiě)的一個(gè)Python腳本的實(shí)現(xiàn)。

#!/usr/bin/python# -*- coding: utf-8 -*-import sysimport reimport timedef print_help():  print 'Usage: '  print ' python mem.py 100MB'  print ' python mem.py 1GB'if __name__ == "__main__":  if len(sys.argv) == 2:    pattern = re.compile('^(/d*)([M|G]B)$')    match = pattern.match(sys.argv[1].upper())    if match:      num = int(match.group(1))      unit = match.group(2)      if unit == 'MB':        s = ' ' * (num * 1024 * 1024)      else:        s = ' ' * (num * 1024 * 1024 * 1024)      time.sleep(10000)    else:      print_help()  else:    print_help()

使用方法如下：

python mem.py 100Mpython mem.py 1G

以上這篇Python小工具之消耗系統(tǒng)指定大小內(nèi)存的方法就是小編分享給大家的全部?jī)?nèi)容了，希望能給大家一個(gè)參考，也希望大家多多支持武林站長(zhǎng)站。