<Trading Game> Mechanical Turk Graphs Generator

graphs.py

def graphs(results):
  for user,res in results.items():
    clf()
    axes([.075,.05, .725,.9])
    xlabel('Time')
    ylabel('Cash Value')
    title('Results for %s' % user)
    plot(res[0], res[1], label='Cash in Wallet')
    plot(res[0], res[2], label='Inventory Value')
    plot(res[0], res[3], label='Net Worth')
    legend(loc=[1.0,0.6])
    savefig('graphs/user_%s.png' % user)
  
  for label,idx in (('Cash in Wallet', 1), ('Inventory Value', 2), ('Net Worth', 3)):
    clf()
    axes([.075,.05, .725,.9])
    xlabel('Time')
    ylabel('Cash Value')
    title('Combined %s' % label)
    for user,res in results.items():
      plot(res[0], res[idx], label=user)
    legend(loc=[1.0,0.6])
    savefig('graphs/Combined %s.png' % label)