我正在使用 Python 的热点分析器：http ://docs.python.org/2/library/hotshot.html
它显示了如何打印统计信息：
stats.print_stats(20)
但是我如何将其放入文件中？我不确定如何获取这些信息，因此我可以使用 write() 将其写入文件。
编辑：
我想要以这种方式打印的同样易于阅读的结果：
stats = hotshot.stats.load("stones.prof")
stats.strip_dirs()
stats.sort_stats('time', 'calls')
stats.print_stats(20)
所以它看起来像这样：
ncalls tottime percall cumtime percall filename:lineno(function)
1 3.295 3.295 10.090 10.090 pystone.py:79(Proc0)
（所以当我打开stones.prof时，它看起来不像）
Stats 采用可选的“流”参数。只需打开一个文件并将打开的文件对象传递给 Stats 构造函数，如下所示。从那时起，任何对 print_stats() 的调用都将输出到您传递给构造函数的流中。希望这可以帮助。:)
with open('path/to/output', 'w') as stream:
stats = pstats.Stats('path/to/input', stream=stream)
stats.print_stats()
输出重定向怎么样？
import sys
import pstats
sys.stdout = open('readable.profile', 'w')
p = pstats.Stats('input.profile')
p.print_stats()
我最终重写了 print_stats() 函数，首先从 pstats.py 复制它。它返回一个字符串，然后可以将其写入文件。我没有测试每个 if-else 循环，只是它在我需要它的示例中有效。我把原来的行注释掉了。尽管该函数不再使用真正的“自我”，但我已将变量名称保持不变。
stats = hotshot.stats.load("stones.prof")
stats.strip_dirs()
stats.sort_stats('time', 'calls')
readable_str = xprint_stats(stats, 20)
import pstats
def xprint_stats(self, *amount):
x = ""
for filename in self.files:
x += " " + filename
#if self.files: print >> self.stream
# ?
indent = ' ' * 8
for func in self.top_level:
#print >> self.stream, indent, xfunc_get_function_name(func)
x += indent + pstats.func_get_function_name(func)
#print >> self.stream, indent, self.total_calls, "function calls",
x += indent + str(self.total_calls) + " function calls" + " "
if self.total_calls != self.prim_calls:
#print >> self.stream, "(%d primitive calls)" % self.prim_calls,
x += "(%d primitive calls)" % self.prim_calls + " "
#print >> self.stream, "in %.3f seconds" % self.total_tt
#print >> self.stream
x += "in %.3f seconds" % self.total_tt + "\n"
#width, list = stats.get_print_list(amount)
msg, width, list = xget_print_list(stats, amount)
x += msg
if list:
#self.print_title()
x += "\n" + ' ncalls tottime percall cumtime percall filename:lineno(function)'
x += "\n"
for func in list:
#self.print_line(func)
x += xprint_line(self, func) + "\n"
# print >> self.stream
# print >> self.stream
#return self
return x
def xprint_line(self, func):
x = ""
cc, nc, tt, ct, callers = self.stats[func]
c = str(nc)
if nc != cc:
c = c + '/' + str(cc)
# print >> self.stream, c.rjust(9),
# print >> self.stream, f8(tt),
x += c.rjust(9) + " "
x += pstats.f8(tt) + " "
if nc == 0:
#print >> self.stream, ' '*8,
x += ' '*8
else:
#print >> self.stream, f8(float(tt)/nc),
x += pstats.f8(float(tt)/nc) + " "
#print >> self.stream, f8(ct),
x += pstats.f8(ct) + " "
if cc == 0:
#print >> self.stream, ' '*8,
x += ' '*8
else:
#print >> self.stream, f8(float(ct)/cc),
x += pstats.f8(float(ct)/cc) + " "
#print >> self.stream, func_std_string(func)
x += pstats.func_std_string(func) + " "
return x
def xget_print_list(self, sel_list):
width = self.max_name_len
if self.fcn_list:
stat_list = self.fcn_list[:]
msg = " Ordered by: " + self.sort_type + '\n'
else:
stat_list = self.stats.keys()
msg = " Random listing order was used\n"
for selection in sel_list:
stat_list, msg = self.eval_print_amount(selection, stat_list, msg)
count = len(stat_list)
if not stat_list:
return 0, stat_list
#print >> self.stream, msg
if count < len(self.stats):
width = 0
for func in stat_list:
if len(pstats.func_std_string(func)) > width:
width = len(pstats.func_std_string(func))
#return width+2, stat_list
return msg, width+2, stat_list
您可以使用该库：pstats_print2list https://pypi.python.org/pypi/pstats_print2list
pip install pstats_print2list
和用法：
from pstats_print2list import get_pstats_print2list, print_pstats_list
fname_stats = 'my_profiling_out.stats'
pstats_list = get_pstats_print2list(
os.path.expanduser(fname_stats),
filter_fnames=['myfile1.py', 'myfile2.py', 'root_path1'],
exclude_fnames=['dontshow.py', 'path_dont_show'],
sort='cumulative',
limit=5,
)
print_pstats_list(pstats_list)
对于那些寻找另一种解决方案的人，我们可以将字符串数据放入 txt 文件。下面的代码将使用两个函数将两个数字相加，然后制作 Profile 数据，最后将其写入 out.txt 文件。
import cProfile
import time
import pstats
from io import StringIO
def add_slow(a, b):
time.sleep(0.5)
return a+b
def add_fast(a, b):
return a+b
prof = cProfile.Profile()
def main_func():
arr = []
prof.enable()
for i in range(10):
if i%2==0:
arr.append(add_slow(i,i))
else:
arr.append(add_fast(i,i))
prof.disable()
#prof.print_stats(sort='time')
prof.dump_stats("main_funcs.prof")
return arr
main_func()
stream = StringIO();
stats = pstats.Stats("main_funcs.prof", stream=stream);
stats.print_stats()
stream.seek(0)
print(16*'=',"RESULTS",16*'=')
data = stream.read()
print(data)
myText = open('out.txt', 'w')
myText.write(data)
myText.close()