How can I loop through this dictionary instead of hardcoding the keys

Question

So far, I have this code (from cs50/pset6/DNA):

import csv

data_dict = {}
with open(argv[1]) as data_file:
    reader = csv.DictReader(data_file)
    for record          


        

Answer 1

You are on the right track using csv.DictReader.

import csv
from pprint import pprint

data_dict = {}

with open('fasta.csv', 'r') as f:
    reader = csv.DictReader(f)

    for record in reader:
        name = record.pop('name')
        data_dict[name] = record

pprint(data_dict)

Prints

{'Alice': {'AATG': '8', 'AGATC': '2', 'TATC': '3'},
 'Bob': {'AATG': '1', 'AGATC': '4', 'TATC': '5'},
 'Charlie': {'AATG': '2', 'AGATC': '3', 'TATC': '5'}}