This script takes the Radio 2 Top 2000 and analyses artist performance

top2000.py

''' This script takes the Radio 2 Top 2000 (http://www.nporadio2.nl/top2000)
    and analyses artist performance:
    - count: number of occurences of artist in list
    - score: number of points (#1 song gets 2000 point, #2 song gets 1999 etc.)

    store input file as CSV with ;-delimiter in 'TOP-2000-2015.csv'
'''

import csv
import sys

def main(csvfile, score):
    print >>sys.stderr, "Reading %d lines from %s" % (score, csvfile)

    # read and analyze
    with open('TOP-2000-2015.csv') as csvfile:
        artists = dict()
        reader = csv.DictReader(csvfile, delimiter=';')
        score = 2000
        for row in reader:
            artist = row['artiest']
            if artist in artists:
                artist = artists[artist]
                artist['count'] += 1
                artist['score'] += score
            else:
                artists[artist] = dict(name=artist)
                artist = artists[artist]
                artist['count'] = 1
                artist['score'] = score
            score -= 1

    # convert to list
    artist_list = []
    for name, data in artists.iteritems():
        artist_list.append(data)

    # write as sorted CSV to stdout
    writer = csv.DictWriter(sys.stdout, fieldnames='name count score'.split(), lineterminator='\n')
    writer.writeheader()
    for artist in sorted(artist_list, reverse=True, key=lambda artist: artist['score']):
        writer.writerow(artist)

if __name__ == "__main__":
    sys.exit(main('TOP-2000-2015.csv', 2000))