Python cProfile to CSV function

README.md

Convert Python cProfile output to CSV

Based on this post: https://qxf2.com/blog/saving-cprofile-stats-to-a-csv-file/

Usage:

if __name__ == "__main__":
    pr = cProfile.Profile()
    pr.enable()
    main()
    pr.disable()
    csv = prof_to_csv(pr)
    with open("prof.csv", 'w+') as f:
        f.write(csv)

prof_to_csv.py

def prof_to_csv(prof: cProfile.Profile):
    out_stream = io.StringIO()
    pstats.Stats(prof, stream=out_stream).print_stats()
    result = out_stream.getvalue()
    # chop off header lines
    result = 'ncalls' + result.split('ncalls')[-1]
    lines = [','.join(line.rstrip().split(None, 5)) for line in result.split('\n')]
    return '\n'.join(lines)