{ "metadata": { "name": "", "signature": "sha256:798fda98bebba5b506ed6281d9e76b081ce3088ef2aff43ccafaea9870e2b6b2" }, "nbformat": 3, "nbformat_minor": 0, "worksheets": [ { "cells": [ { "cell_type": "markdown", "metadata": {}, "source": [ "# Create a Column Based on a Conditional in pandas\n", "\n", "- **Author:** [Chris Albon](http://www.chrisalbon.com/), [@ChrisAlbon](https://twitter.com/chrisalbon)\n", "- **Date:** -\n", "- **Repo:** [Python 3 code snippets for data science](https://github.com/chrisalbon/code_py)\n", "- **Note:**" ] }, { "cell_type": "markdown", "metadata": {}, "source": [ "### Preliminaries" ] }, { "cell_type": "code", "collapsed": false, "input": [ "# Import required modules\n", "import pandas as pd\n", "import numpy as np" ], "language": "python", "metadata": {}, "outputs": [], "prompt_number": 4 }, { "cell_type": "markdown", "metadata": {}, "source": [ "### Make a dataframe" ] }, { "cell_type": "code", "collapsed": false, "input": [ "data = {'name': ['Jason', 'Molly', 'Tina', 'Jake', 'Amy'], \n", " 'age': [42, 52, 36, 24, 73], \n", " 'preTestScore': [4, 24, 31, 2, 3],\n", " 'postTestScore': [25, 94, 57, 62, 70]}\n", "df = pd.DataFrame(data, columns = ['name', 'age', 'preTestScore', 'postTestScore'])\n", "df" ], "language": "python", "metadata": {}, "outputs": [ { "html": [ "
\n", " | name | \n", "age | \n", "preTestScore | \n", "postTestScore | \n", "
---|---|---|---|---|
0 | \n", "Jason | \n", "42 | \n", "4 | \n", "25 | \n", "
1 | \n", "Molly | \n", "52 | \n", "24 | \n", "94 | \n", "
2 | \n", "Tina | \n", "36 | \n", "31 | \n", "57 | \n", "
3 | \n", "Jake | \n", "24 | \n", "2 | \n", "62 | \n", "
4 | \n", "Amy | \n", "73 | \n", "3 | \n", "70 | \n", "
5 rows \u00d7 4 columns
\n", "\n", " | name | \n", "age | \n", "preTestScore | \n", "postTestScore | \n", "elderly | \n", "
---|---|---|---|---|---|
0 | \n", "Jason | \n", "42 | \n", "4 | \n", "25 | \n", "no | \n", "
1 | \n", "Molly | \n", "52 | \n", "24 | \n", "94 | \n", "yes | \n", "
2 | \n", "Tina | \n", "36 | \n", "31 | \n", "57 | \n", "no | \n", "
3 | \n", "Jake | \n", "24 | \n", "2 | \n", "62 | \n", "no | \n", "
4 | \n", "Amy | \n", "73 | \n", "3 | \n", "70 | \n", "yes | \n", "
5 rows \u00d7 5 columns
\n", "