Skip to content

Instantly share code, notes, and snippets.

@devinbrady

devinbrady/pandas_pivot_table_demo.json

Created July 15, 2015 00:59
Show Gist options
  • Save devinbrady/f719ffdc4f4ea3eb379a to your computer and use it in GitHub Desktop.
Save devinbrady/f719ffdc4f4ea3eb379a to your computer and use it in GitHub Desktop.
Download ZIP
Raw
pandas_pivot_table_demo.json
{
"cells": [
{
"cell_type": "code",
"execution_count": 1,
"metadata": {
"collapsed": true
},
"outputs": [],
"source": [
"# This is a short example to take in a file and save out a pivot table. "
]
},
{
"cell_type": "code",
"execution_count": 50,
"metadata": {
"collapsed": true
},
"outputs": [],
"source": [
"import pandas as pd\n",
"import numpy as np"
]
},
{
"cell_type": "code",
"execution_count": 51,
"metadata": {
"collapsed": false
},
"outputs": [
{
"data": {
"text/html": [
"<div style=\"max-height:1000px;max-width:1500px;overflow:auto;\">\n",
"<table border=\"1\" class=\"dataframe\">\n",
" <thead>\n",
" <tr style=\"text-align: right;\">\n",
" <th></th>\n",
" <th>Voter ID</th>\n",
" <th>Election</th>\n",
" </tr>\n",
" </thead>\n",
" <tbody>\n",
" <tr>\n",
" <th>0</th>\n",
" <td> 1</td>\n",
" <td> 09/13/2012</td>\n",
" </tr>\n",
" <tr>\n",
" <th>1</th>\n",
" <td> 1</td>\n",
" <td> 11/06/2012</td>\n",
" </tr>\n",
" <tr>\n",
" <th>2</th>\n",
" <td> 2</td>\n",
" <td> 09/13/2012</td>\n",
" </tr>\n",
" <tr>\n",
" <th>3</th>\n",
" <td> 3</td>\n",
" <td> 09/13/2012</td>\n",
" </tr>\n",
" <tr>\n",
" <th>4</th>\n",
" <td> 3</td>\n",
" <td> 11/06/2012</td>\n",
" </tr>\n",
" </tbody>\n",
"</table>\n",
"</div>"
],
"text/plain": [
" Voter ID Election\n",
"0 1 09/13/2012\n",
"1 1 11/06/2012\n",
"2 2 09/13/2012\n",
"3 3 09/13/2012\n",
"4 3 11/06/2012"
]
},
"execution_count": 51,
"metadata": {},
"output_type": "execute_result"
}
],
"source": [
"voter_file = pd.read_csv('voter_file_n5.csv')\n",
"voter_file"
]
},
{
"cell_type": "code",
"execution_count": 52,
"metadata": {
"collapsed": false
},
"outputs": [
{
"data": {
"text/html": [
"<div style=\"max-height:1000px;max-width:1500px;overflow:auto;\">\n",
"<table border=\"1\" class=\"dataframe\">\n",
" <thead>\n",
" <tr style=\"text-align: right;\">\n",
" <th></th>\n",
" <th>Voter ID</th>\n",
" <th>Election</th>\n",
" <th>Voted</th>\n",
" </tr>\n",
" </thead>\n",
" <tbody>\n",
" <tr>\n",
" <th>0</th>\n",
" <td> 1</td>\n",
" <td> 09/13/2012</td>\n",
" <td> Yes</td>\n",
" </tr>\n",
" <tr>\n",
" <th>1</th>\n",
" <td> 1</td>\n",
" <td> 11/06/2012</td>\n",
" <td> Yes</td>\n",
" </tr>\n",
" <tr>\n",
" <th>2</th>\n",
" <td> 2</td>\n",
" <td> 09/13/2012</td>\n",
" <td> Yes</td>\n",
" </tr>\n",
" <tr>\n",
" <th>3</th>\n",
" <td> 3</td>\n",
" <td> 09/13/2012</td>\n",
" <td> Yes</td>\n",
" </tr>\n",
" <tr>\n",
" <th>4</th>\n",
" <td> 3</td>\n",
" <td> 11/06/2012</td>\n",
" <td> Yes</td>\n",
" </tr>\n",
" </tbody>\n",
"</table>\n",
"</div>"
],
"text/plain": [
" Voter ID Election Voted\n",
"0 1 09/13/2012 Yes\n",
"1 1 11/06/2012 Yes\n",
"2 2 09/13/2012 Yes\n",
"3 3 09/13/2012 Yes\n",
"4 3 11/06/2012 Yes"
]
},
"execution_count": 52,
"metadata": {},
"output_type": "execute_result"
}
],
"source": [
"voter_file['Voted'] = 'Yes'\n",
"voter_file"
]
},
{
"cell_type": "code",
"execution_count": 53,
"metadata": {
"collapsed": false
},
"outputs": [
{
"data": {
"text/html": [
"<div style=\"max-height:1000px;max-width:1500px;overflow:auto;\">\n",
"<table border=\"1\" class=\"dataframe\">\n",
" <thead>\n",
" <tr>\n",
" <th></th>\n",
" <th colspan=\"2\" halign=\"left\">Voted</th>\n",
" </tr>\n",
" <tr>\n",
" <th>Election</th>\n",
" <th>09/13/2012</th>\n",
" <th>11/06/2012</th>\n",
" </tr>\n",
" <tr>\n",
" <th>Voter ID</th>\n",
" <th></th>\n",
" <th></th>\n",
" </tr>\n",
" </thead>\n",
" <tbody>\n",
" <tr>\n",
" <th>1</th>\n",
" <td> Yes</td>\n",
" <td> Yes</td>\n",
" </tr>\n",
" <tr>\n",
" <th>2</th>\n",
" <td> Yes</td>\n",
" <td> NaN</td>\n",
" </tr>\n",
" <tr>\n",
" <th>3</th>\n",
" <td> Yes</td>\n",
" <td> Yes</td>\n",
" </tr>\n",
" </tbody>\n",
"</table>\n",
"</div>"
],
"text/plain": [
" Voted \n",
"Election 09/13/2012 11/06/2012\n",
"Voter ID \n",
"1 Yes Yes\n",
"2 Yes NaN\n",
"3 Yes Yes"
]
},
"execution_count": 53,
"metadata": {},
"output_type": "execute_result"
}
],
"source": [
"voter_pivot = voter_file.pivot('Voter ID', 'Election')\n",
"voter_pivot"
]
},
{
"cell_type": "code",
"execution_count": 54,
"metadata": {
"collapsed": false
},
"outputs": [
{
"data": {
"text/html": [
"<div style=\"max-height:1000px;max-width:1500px;overflow:auto;\">\n",
"<table border=\"1\" class=\"dataframe\">\n",
" <thead>\n",
" <tr>\n",
" <th></th>\n",
" <th colspan=\"2\" halign=\"left\">Voted</th>\n",
" </tr>\n",
" <tr>\n",
" <th>Election</th>\n",
" <th>09/13/2012</th>\n",
" <th>11/06/2012</th>\n",
" </tr>\n",
" <tr>\n",
" <th>Voter ID</th>\n",
" <th></th>\n",
" <th></th>\n",
" </tr>\n",
" </thead>\n",
" <tbody>\n",
" <tr>\n",
" <th>1</th>\n",
" <td> Yes</td>\n",
" <td> Yes</td>\n",
" </tr>\n",
" <tr>\n",
" <th>2</th>\n",
" <td> Yes</td>\n",
" <td> No</td>\n",
" </tr>\n",
" <tr>\n",
" <th>3</th>\n",
" <td> Yes</td>\n",
" <td> Yes</td>\n",
" </tr>\n",
" </tbody>\n",
"</table>\n",
"</div>"
],
"text/plain": [
" Voted \n",
"Election 09/13/2012 11/06/2012\n",
"Voter ID \n",
"1 Yes Yes\n",
"2 Yes No\n",
"3 Yes Yes"
]
},
"execution_count": 54,
"metadata": {},
"output_type": "execute_result"
}
],
"source": [
"voter_pivot = voter_pivot.fillna('No')\n",
"voter_pivot"
]
},
{
"cell_type": "code",
"execution_count": 55,
"metadata": {
"collapsed": true
},
"outputs": [],
"source": [
"voter_pivot.to_csv('voter_pivot.csv')"
]
}
],
"metadata": {
"kernelspec": {
"display_name": "Python 2",
"language": "python",
"name": "python2"
},
"language_info": {
"codemirror_mode": {
"name": "ipython",
"version": 2
},
"file_extension": ".py",
"mimetype": "text/x-python",
"name": "python",
"nbconvert_exporter": "python",
"pygments_lexer": "ipython2",
"version": "2.7.6"
}
},
"nbformat": 4,
"nbformat_minor": 0
}
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment