Curried

curried.py

from __future__ import division                                    
                                                                   
from funcy.curried import *                                        
from fn import _ as __                                             
from bookends import _

import survey


def total_probability((outcome, pmf)):
  return (_| pmf
           | select_keys(__ == outcome)
           | __.values
           | call
           | sum
           |_)
  

def prglength_pmf(records):
  return (_| records
           | count_by(__.prglength)
           | walk_values(__ / len(records))
           |_)


@curry
def born_on_or_after(records, week):
  return filter(__.prglength >= week,
                records)


def main():
  births = survey.read_records()
  outcomes = range(60)
  prglength_groups = map(born_on_or_after(births),
                         outcomes)
                         
  print (_| prglength_groups                                       
          | map(prglength_pmf)                                     
          | partial(zip, outcomes)                                 
          | map(total_probability)                                 
          |_)