anachronism.py

import random
from collections import Counter

NUM_DISTRICTS = 500

# Doesn't matter, think of each of these as representing
# thousands of voters if you like.
NUM_VOTERS = 1000

# % of voters that vote D (remainder vote R)
DEM_PERCENT = .51

NUM_D = int(NUM_DISTRICTS * NUM_VOTERS * DEM_PERCENT)
NUM_R = int(NUM_DISTRICTS * NUM_VOTERS * (1 - DEM_PERCENT))

VOTERS = (['D'] * NUM_D) + (['R'] * NUM_R)

# No gerrymandering, so shuffle voters at random.
random.shuffle(VOTERS)

district_winners = Counter()

for i in range(0, NUM_VOTERS * NUM_DISTRICTS, NUM_VOTERS):
    # Count the votes in district i
    votes = Counter(VOTERS[i:i+NUM_VOTERS])

    # print(votes)

    # Find the winner
    winner = votes.most_common()[0][0]
    district_winners[winner] += 1

print(district_winners)