diff --git a/s8.py b/s8.py index c87ed1e..7be379d 100644 --- a/s8.py +++ b/s8.py @@ -72,12 +72,20 @@ def middleman(ls, rs): def main(): + header = [ + 'lo_id', '', + ] + ['' for i in xrange(0, 19)] for m in MODEL_LIST: l_list, lo_data = get_test_data(m) + m_header = header[:] + m_header[1] = m for i in '1234': ros = get_ro_data(m, i) - result = middleman(lo_data, ros) + for r in ros: + m_header.append(r[0]) + result = middleman(lo_data, ros) output_name = '%s-%s.csv' % (m, i) + result = [m_header] + result create_csv(output_name, result, directory='part8') main()