diff options
| -rw-r--r-- | gn3/computations/qtlreaper.py | 4 |
1 files changed, 3 insertions, 1 deletions
diff --git a/gn3/computations/qtlreaper.py b/gn3/computations/qtlreaper.py index 49d363b..a88659e 100644 --- a/gn3/computations/qtlreaper.py +++ b/gn3/computations/qtlreaper.py @@ -27,7 +27,9 @@ def generate_traits_file(strains, trait_values, traits_filename): header = "Traits\t{}\n".format("\t".join(strains)) data = [header] + [ "T{}\t{}\n".format(i+1, "\t".join([str(i) for i in t])) - for i, t in enumerate(trait_values)] + for i, t in enumerate(trait_values[:-1])] + [ + "T{}\t{}".format(len(trait_values), "\t".join([str(i) for i in t])) + for t in trait_values[-1:]] with open(traits_filename, "w") as outfile: outfile.writelines(data) |