gdi-tlapip.ipynb

{
 "cells": [
  {
   "cell_type": "code",
   "execution_count": null,
   "metadata": {
    "collapsed": false
   },
   "outputs": [],
   "source": [
    "! head -5 ice-cream.csv"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": null,
   "metadata": {
    "collapsed": false
   },
   "outputs": [],
   "source": [
    "\"\"\"The simplest data reader.\"\"\"\n",
    "\n",
    "for line in open('ice-cream.csv'):\n",
    "    row = line.split(',')\n",
    "    print(row)"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": null,
   "metadata": {
    "collapsed": false
   },
   "outputs": [],
   "source": [
    "def read_data(path):\n",
    "    \"\"\"Reads comma separated data from path.\"\"\"\n",
    "    infile = open(path)\n",
    "    infile.readline()   # discard headers\n",
    "    for line in infile:\n",
    "        line = line.strip()\n",
    "        row = line.split(',')\n",
    "        print(row)\n",
    "\n",
    "read_data('ice-cream.csv')"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": null,
   "metadata": {
    "collapsed": false
   },
   "outputs": [],
   "source": [
    "'1011' / 10"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": null,
   "metadata": {
    "collapsed": false
   },
   "outputs": [],
   "source": [
    "row = ['1011', 'f', 't']\n",
    "assert row[1] in 'mf' and row[2] in 'ps'"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": null,
   "metadata": {
    "collapsed": false
   },
   "outputs": [],
   "source": [
    "line = '\"damon, matt\", m, p'\n",
    "line.split(',')"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": null,
   "metadata": {
    "collapsed": false
   },
   "outputs": [],
   "source": [
    "import csv\n",
    "\n",
    "def read_data(path):\n",
    "    \"\"\"Reads comma separated data from path.\"\"\"\n",
    "    with open(path) as csvfile:\n",
    "        reader = csv.DictReader(csvfile)\n",
    "        for row in reader:\n",
    "            assert row['gender'] in 'mf'\n",
    "            assert row['preference'] in 'ps'\n",
    "            print(row['gender'], row['preference'])\n",
    "\n",
    "read_data('ice-cream.csv')"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": null,
   "metadata": {
    "collapsed": false
   },
   "outputs": [],
   "source": [
    "counts = {\n",
    " 'm': \n",
    "    {'s': 8,\n",
    "     'p': 12},\n",
    " 'f':\n",
    "    {'s': 16,\n",
    "     'p': 4}\n",
    "}"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": null,
   "metadata": {
    "collapsed": false
   },
   "outputs": [],
   "source": [
    "counts['f']"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": null,
   "metadata": {
    "collapsed": false
   },
   "outputs": [],
   "source": [
    "counts['f']['p']"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": null,
   "metadata": {
    "collapsed": false
   },
   "outputs": [],
   "source": [
    "import csv\n",
    "from collections import defaultdict\n",
    "\n",
    "def ice_cream_counts(path):\n",
    "    \"\"\"Returns gender/preference counts from csv in path.\"\"\"\n",
    "    counts = {'m': defaultdict(int), 'f': defaultdict(int)}\n",
    "\n",
    "    with open(path) as csvfile:\n",
    "        reader = csv.DictReader(csvfile)\n",
    "        for row in reader:\n",
    "            counts[row['gender']][row['preference']] += 1\n",
    "\n",
    "    return counts\n",
    "\n",
    "counts = ice_cream_counts('ice-cream.csv')\n",
    "print(counts)"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": null,
   "metadata": {
    "collapsed": false
   },
   "outputs": [],
   "source": [
    "def write_html(counts):\n",
    "    flavors = {'p': 'pistachio', 's': 'strawberry'}\n",
    "\n",
    "    heading = \"\"\"<!DOCTYPE html>\n",
    "    <html><head><title>Ice Cream Preferences</title></head><body>\n",
    "    <table border=\"1\">\n",
    "    <tr> <th>preference</th> <th>female</th> <th>male</th> </tr>\"\"\"\n",
    "    table_row = '<tr> <td>{}</td> <td>{}</td> <td>{}</td> </tr>'\n",
    "\n",
    "    print(heading)\n",
    "    for f, flavor in flavors.items():\n",
    "        print(table_row.format(flavor, counts['f'][f], counts['m'][f]))\n",
    "    print('</table></body></html>')\n",
    "\n",
    "counts = ice_cream_counts('ice-cream.csv')\n",
    "write_html(counts)"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": null,
   "metadata": {
    "collapsed": true
   },
   "outputs": [],
   "source": []
  }
 ],
 "metadata": {
  "kernelspec": {
   "display_name": "IPython (Python 3)",
   "language": "python",
   "name": "python3"
  },
  "language_info": {
   "name": "python",
   "version": "3.4.3"
  }
 },
 "nbformat": 4,
 "nbformat_minor": 0
}