diff options
Diffstat (limited to 'wk6/pset/dna')
| -rw-r--r-- | wk6/pset/dna/dna.py | 7 |
1 files changed, 5 insertions, 2 deletions
diff --git a/wk6/pset/dna/dna.py b/wk6/pset/dna/dna.py index 3110e4e..4c37abc 100644 --- a/wk6/pset/dna/dna.py +++ b/wk6/pset/dna/dna.py @@ -22,8 +22,11 @@ def main(): rows = [] with open(database) as file: reader = csv.DictReader(file) - for row in reader: - rows.append(row) + # Get the header (first line) + header = next(reader) + + # Get the rest of the data + rows = list(reader) # TODO: Read DNA sequence file into a variable text = 'sequences/' + sys.argv[2] |