Skip to content

Instantly share code, notes, and snippets.

@craine
Created April 11, 2020 16:40
Show Gist options
  • Save craine/0d7ef86ac02347280be9fadd4e3f366c to your computer and use it in GitHub Desktop.
Save craine/0d7ef86ac02347280be9fadd4e3f366c to your computer and use it in GitHub Desktop.
Add a column in Pandas
Display the source blob
Display the rendered blob
Raw
{
"nbformat": 4,
"nbformat_minor": 0,
"metadata": {
"colab": {
"name": "Add a column in Pandas",
"provenance": [],
"authorship_tag": "ABX9TyN6hSbpO4scLN8BMEpoK7Do",
"include_colab_link": true
},
"kernelspec": {
"name": "python3",
"display_name": "Python 3"
}
},
"cells": [
{
"cell_type": "markdown",
"metadata": {
"id": "view-in-github",
"colab_type": "text"
},
"source": [
"<a href=\"https://colab.research.google.com/gist/craine/0d7ef86ac02347280be9fadd4e3f366c/add-a-column-in-pandas.ipynb\" target=\"_parent\"><img src=\"https://colab.research.google.com/assets/colab-badge.svg\" alt=\"Open In Colab\"/></a>"
]
},
{
"cell_type": "code",
"metadata": {
"id": "fLKSNC6oFdvF",
"colab_type": "code",
"colab": {}
},
"source": [
"import pandas as pd"
],
"execution_count": 0,
"outputs": []
},
{
"cell_type": "code",
"metadata": {
"id": "TCYiOZTEFhmj",
"colab_type": "code",
"colab": {
"base_uri": "https://localhost:8080/",
"height": 142
},
"outputId": "bdf75d5a-230c-4426-860a-746fb9b5db35"
},
"source": [
"# Create a dataframe in pandas\n",
"df = pd.DataFrame()\n",
"\n",
"# Create your first column\n",
"df['team'] = ['Manchester City', 'Liverpool', 'Manchester']\n",
"\n",
"# View dataframe\n",
"df"
],
"execution_count": 2,
"outputs": [
{
"output_type": "execute_result",
"data": {
"text/html": [
"<div>\n",
"<style scoped>\n",
" .dataframe tbody tr th:only-of-type {\n",
" vertical-align: middle;\n",
" }\n",
"\n",
" .dataframe tbody tr th {\n",
" vertical-align: top;\n",
" }\n",
"\n",
" .dataframe thead th {\n",
" text-align: right;\n",
" }\n",
"</style>\n",
"<table border=\"1\" class=\"dataframe\">\n",
" <thead>\n",
" <tr style=\"text-align: right;\">\n",
" <th></th>\n",
" <th>team</th>\n",
" </tr>\n",
" </thead>\n",
" <tbody>\n",
" <tr>\n",
" <th>0</th>\n",
" <td>Manchester City</td>\n",
" </tr>\n",
" <tr>\n",
" <th>1</th>\n",
" <td>Liverpool</td>\n",
" </tr>\n",
" <tr>\n",
" <th>2</th>\n",
" <td>Manchester</td>\n",
" </tr>\n",
" </tbody>\n",
"</table>\n",
"</div>"
],
"text/plain": [
" team\n",
"0 Manchester City\n",
"1 Liverpool\n",
"2 Manchester"
]
},
"metadata": {
"tags": []
},
"execution_count": 2
}
]
},
{
"cell_type": "markdown",
"metadata": {
"id": "L39P4I8XGZUq",
"colab_type": "text"
},
"source": [
"Add data to your column"
]
},
{
"cell_type": "code",
"metadata": {
"id": "LZ-2FbY0GWzl",
"colab_type": "code",
"colab": {
"base_uri": "https://localhost:8080/",
"height": 142
},
"outputId": "0ded96a6-1ca4-4b9b-b38c-bd70d23f9279"
},
"source": [
"# Add a new column to your dataframe called 'wins'\n",
"df.assign(age = [41, 40, 21])"
],
"execution_count": 3,
"outputs": [
{
"output_type": "execute_result",
"data": {
"text/html": [
"<div>\n",
"<style scoped>\n",
" .dataframe tbody tr th:only-of-type {\n",
" vertical-align: middle;\n",
" }\n",
"\n",
" .dataframe tbody tr th {\n",
" vertical-align: top;\n",
" }\n",
"\n",
" .dataframe thead th {\n",
" text-align: right;\n",
" }\n",
"</style>\n",
"<table border=\"1\" class=\"dataframe\">\n",
" <thead>\n",
" <tr style=\"text-align: right;\">\n",
" <th></th>\n",
" <th>team</th>\n",
" <th>age</th>\n",
" </tr>\n",
" </thead>\n",
" <tbody>\n",
" <tr>\n",
" <th>0</th>\n",
" <td>Manchester City</td>\n",
" <td>41</td>\n",
" </tr>\n",
" <tr>\n",
" <th>1</th>\n",
" <td>Liverpool</td>\n",
" <td>40</td>\n",
" </tr>\n",
" <tr>\n",
" <th>2</th>\n",
" <td>Manchester</td>\n",
" <td>21</td>\n",
" </tr>\n",
" </tbody>\n",
"</table>\n",
"</div>"
],
"text/plain": [
" team age\n",
"0 Manchester City 41\n",
"1 Liverpool 40\n",
"2 Manchester 21"
]
},
"metadata": {
"tags": []
},
"execution_count": 3
}
]
}
]
}
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment