2_process_asterism_data.ipynb

{
 "cells": [
  {
   "cell_type": "code",
   "execution_count": 1,
   "metadata": {},
   "outputs": [],
   "source": [
    "import pandas as pd\n",
    "import numpy as np"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 2,
   "metadata": {},
   "outputs": [
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "watermark 1.8.1\n",
      "numpy     1.15.4\n",
      "pandas    0.23.4\n",
      "ELEANOR LUTZ 2019-07-11 \n",
      "\n",
      "CPython 3.7.1\n",
      "IPython 7.2.0\n",
      "\n",
      "compiler   : MSC v.1900 64 bit (AMD64)\n",
      "system     : Windows\n",
      "release    : 10\n",
      "machine    : AMD64\n",
      "processor  : Intel64 Family 6 Model 63 Stepping 2, GenuineIntel\n",
      "CPU cores  : 12\n",
      "interpreter: 64bit\n"
     ]
    }
   ],
   "source": [
    "# Watermark is not required for this code, but is included for information. \n",
    "import watermark\n",
    "%load_ext watermark\n",
    "%watermark -a \"ELEANOR LUTZ\" -d -v -iv -m"
   ]
  },
  {
   "cell_type": "markdown",
   "metadata": {},
   "source": [
    "## Data Source\n",
    "The asterism data used in this Jupyter Notebook is from the open-source [Stellarium Astronomy Software](https://stellarium.org/). "
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 3,
   "metadata": {},
   "outputs": [
    {
     "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>constellation</th>\n",
       "      <th>num_pairs</th>\n",
       "      <th>stars</th>\n",
       "    </tr>\n",
       "  </thead>\n",
       "  <tbody>\n",
       "    <tr>\n",
       "      <th>0</th>\n",
       "      <td>Aql</td>\n",
       "      <td>8</td>\n",
       "      <td>[98036, 97649, 97649, 97278, 97649, 95501, 955...</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>1</th>\n",
       "      <td>And</td>\n",
       "      <td>5</td>\n",
       "      <td>[677, 3092, 3092, 5447, 9640, 5447, 5447, 4436...</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>2</th>\n",
       "      <td>Scl</td>\n",
       "      <td>3</td>\n",
       "      <td>[116231, 4577, 4577, 115102, 115102, 116231]</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>3</th>\n",
       "      <td>Ara</td>\n",
       "      <td>7</td>\n",
       "      <td>[88714, 85792, 85792, 83081, 83081, 82363, 823...</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>4</th>\n",
       "      <td>Lib</td>\n",
       "      <td>5</td>\n",
       "      <td>[77853, 76333, 76333, 74785, 74785, 72622, 726...</td>\n",
       "    </tr>\n",
       "  </tbody>\n",
       "</table>\n",
       "</div>"
      ],
      "text/plain": [
       "  constellation num_pairs                                              stars\n",
       "0           Aql         8  [98036, 97649, 97649, 97278, 97649, 95501, 955...\n",
       "1           And         5  [677, 3092, 3092, 5447, 9640, 5447, 5447, 4436...\n",
       "2           Scl         3       [116231, 4577, 4577, 115102, 115102, 116231]\n",
       "3           Ara         7  [88714, 85792, 85792, 83081, 83081, 82363, 823...\n",
       "4           Lib         5  [77853, 76333, 76333, 74785, 74785, 72622, 726..."
      ]
     },
     "metadata": {},
     "output_type": "display_data"
    },
    {
     "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>constellation</th>\n",
       "      <th>name</th>\n",
       "    </tr>\n",
       "  </thead>\n",
       "  <tbody>\n",
       "    <tr>\n",
       "      <th>0</th>\n",
       "      <td>Aql</td>\n",
       "      <td>Aquila</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>1</th>\n",
       "      <td>And</td>\n",
       "      <td>Andromeda</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>2</th>\n",
       "      <td>Scl</td>\n",
       "      <td>Sculptor</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>3</th>\n",
       "      <td>Ara</td>\n",
       "      <td>Ara</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>4</th>\n",
       "      <td>Lib</td>\n",
       "      <td>Libra</td>\n",
       "    </tr>\n",
       "  </tbody>\n",
       "</table>\n",
       "</div>"
      ],
      "text/plain": [
       "  constellation       name\n",
       "0           Aql     Aquila\n",
       "1           And  Andromeda\n",
       "2           Scl   Sculptor\n",
       "3           Ara        Ara\n",
       "4           Lib      Libra"
      ]
     },
     "metadata": {},
     "output_type": "display_data"
    },
    {
     "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>constellation</th>\n",
       "      <th>num_pairs</th>\n",
       "      <th>stars</th>\n",
       "      <th>name</th>\n",
       "    </tr>\n",
       "  </thead>\n",
       "  <tbody>\n",
       "    <tr>\n",
       "      <th>0</th>\n",
       "      <td>Aql</td>\n",
       "      <td>8</td>\n",
       "      <td>[98036, 97649, 97649, 97278, 97649, 95501, 955...</td>\n",
       "      <td>Aquila</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>1</th>\n",
       "      <td>And</td>\n",
       "      <td>5</td>\n",
       "      <td>[677, 3092, 3092, 5447, 9640, 5447, 5447, 4436...</td>\n",
       "      <td>Andromeda</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>2</th>\n",
       "      <td>Scl</td>\n",
       "      <td>3</td>\n",
       "      <td>[116231, 4577, 4577, 115102, 115102, 116231]</td>\n",
       "      <td>Sculptor</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>3</th>\n",
       "      <td>Ara</td>\n",
       "      <td>7</td>\n",
       "      <td>[88714, 85792, 85792, 83081, 83081, 82363, 823...</td>\n",
       "      <td>Ara</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>4</th>\n",
       "      <td>Lib</td>\n",
       "      <td>5</td>\n",
       "      <td>[77853, 76333, 76333, 74785, 74785, 72622, 726...</td>\n",
       "      <td>Libra</td>\n",
       "    </tr>\n",
       "  </tbody>\n",
       "</table>\n",
       "</div>"
      ],
      "text/plain": [
       "  constellation num_pairs                                              stars  \\\n",
       "0           Aql         8  [98036, 97649, 97649, 97278, 97649, 95501, 955...   \n",
       "1           And         5  [677, 3092, 3092, 5447, 9640, 5447, 5447, 4436...   \n",
       "2           Scl         3       [116231, 4577, 4577, 115102, 115102, 116231]   \n",
       "3           Ara         7  [88714, 85792, 85792, 83081, 83081, 82363, 823...   \n",
       "4           Lib         5  [77853, 76333, 76333, 74785, 74785, 72622, 726...   \n",
       "\n",
       "        name  \n",
       "0     Aquila  \n",
       "1  Andromeda  \n",
       "2   Sculptor  \n",
       "3        Ara  \n",
       "4      Libra  "
      ]
     },
     "metadata": {},
     "output_type": "display_data"
    }
   ],
   "source": [
    "df = pd.read_csv('./data/stellarium_western_asterisms/constellationship.fab', header=None)\n",
    "df['constellation'] = df[0].str.split().str.get(0)\n",
    "df['num_pairs'] = df[0].str.split().str.get(1)\n",
    "df['stars'] = df[0].str.split().str[2:]\n",
    "df.drop(0, axis=1, inplace=True)\n",
    "display(df.head())\n",
    "\n",
    "df_names = pd.read_csv('./data/stellarium_western_asterisms/constellation_names.eng.fab', header=None)\n",
    "df_names = df_names[0].str.replace('\\t', '').str.split('\"', expand=True)\n",
    "df_names.drop([2, 3, 4], axis=1, inplace=True)\n",
    "df_names.columns = ['constellation', 'name']\n",
    "display(df_names.head())\n",
    "\n",
    "assert len(df) == len(df_names)\n",
    "df = pd.merge(df, df_names, on=\"constellation\")\n",
    "display(df.head())"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 4,
   "metadata": {},
   "outputs": [
    {
     "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>star_ID</th>\n",
       "      <th>ra</th>\n",
       "      <th>dec</th>\n",
       "    </tr>\n",
       "  </thead>\n",
       "  <tbody>\n",
       "    <tr>\n",
       "      <th>0</th>\n",
       "      <td>677.0</td>\n",
       "      <td>0.139791</td>\n",
       "      <td>29.090432</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>1</th>\n",
       "      <td>746.0</td>\n",
       "      <td>0.152887</td>\n",
       "      <td>59.149780</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>2</th>\n",
       "      <td>765.0</td>\n",
       "      <td>0.156836</td>\n",
       "      <td>-45.747426</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>3</th>\n",
       "      <td>1067.0</td>\n",
       "      <td>0.220598</td>\n",
       "      <td>15.183596</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>4</th>\n",
       "      <td>1562.0</td>\n",
       "      <td>0.323799</td>\n",
       "      <td>-8.823921</td>\n",
       "    </tr>\n",
       "  </tbody>\n",
       "</table>\n",
       "</div>"
      ],
      "text/plain": [
       "   star_ID        ra        dec\n",
       "0    677.0  0.139791  29.090432\n",
       "1    746.0  0.152887  59.149780\n",
       "2    765.0  0.156836 -45.747426\n",
       "3   1067.0  0.220598  15.183596\n",
       "4   1562.0  0.323799  -8.823921"
      ]
     },
     "metadata": {},
     "output_type": "display_data"
    }
   ],
   "source": [
    "stars = [float(y) for x in df['stars'].tolist() for y in x]\n",
    "stars = sorted(set(stars))\n",
    "\n",
    "hip_df = pd.read_csv('./data/processed/hygdata_processed.csv', low_memory=False)\n",
    "ras, decs, = [], []\n",
    "for star in stars: \n",
    "    temp = hip_df[hip_df['hip']==star]\n",
    "    assert len(temp) == 1\n",
    "    ras.append(temp['ra'].tolist()[0])\n",
    "    decs.append(temp['dec'].tolist()[0])\n",
    "\n",
    "star_df = pd.DataFrame(data={'star_ID':stars, 'ra':ras, 'dec':decs})\n",
    "display(star_df.head())"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 5,
   "metadata": {
    "scrolled": false
   },
   "outputs": [
    {
     "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>constellation</th>\n",
       "      <th>num_pairs</th>\n",
       "      <th>stars</th>\n",
       "      <th>name</th>\n",
       "      <th>ra</th>\n",
       "      <th>dec</th>\n",
       "    </tr>\n",
       "  </thead>\n",
       "  <tbody>\n",
       "    <tr>\n",
       "      <th>0</th>\n",
       "      <td>Aql</td>\n",
       "      <td>8</td>\n",
       "      <td>[98036, 97649, 97649, 97278, 97649, 95501, 955...</td>\n",
       "      <td>Aquila</td>\n",
       "      <td>[19.921887, 19.846388, 19.846388, 19.770994, 1...</td>\n",
       "      <td>[6.406763, 8.868322000000003, 8.86832200000000...</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>1</th>\n",
       "      <td>And</td>\n",
       "      <td>5</td>\n",
       "      <td>[677, 3092, 3092, 5447, 9640, 5447, 5447, 4436...</td>\n",
       "      <td>Andromeda</td>\n",
       "      <td>[0.139791, 0.655462, 0.655462, 1.162194, 2.064...</td>\n",
       "      <td>[29.090432, 30.861024, 30.861024, 35.620558, 4...</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>2</th>\n",
       "      <td>Scl</td>\n",
       "      <td>3</td>\n",
       "      <td>[116231, 4577, 4577, 115102, 115102, 116231]</td>\n",
       "      <td>Sculptor</td>\n",
       "      <td>[23.549512, 0.976766, 0.976766, 23.313733, 23....</td>\n",
       "      <td>[-37.818268, -29.357449, -29.357449, -32.53202...</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>3</th>\n",
       "      <td>Ara</td>\n",
       "      <td>7</td>\n",
       "      <td>[88714, 85792, 85792, 83081, 83081, 82363, 823...</td>\n",
       "      <td>Ara</td>\n",
       "      <td>[18.11052, 17.530695, 17.530695, 16.977006, 16...</td>\n",
       "      <td>[-50.09147700000001, -49.876145, -49.876145, -...</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>4</th>\n",
       "      <td>Lib</td>\n",
       "      <td>5</td>\n",
       "      <td>[77853, 76333, 76333, 74785, 74785, 72622, 726...</td>\n",
       "      <td>Libra</td>\n",
       "      <td>[15.897093, 15.592104999999998, 15.59210499999...</td>\n",
       "      <td>[-16.729293, -14.789537, -14.789537, -9.382917...</td>\n",
       "    </tr>\n",
       "  </tbody>\n",
       "</table>\n",
       "</div>"
      ],
      "text/plain": [
       "  constellation num_pairs                                              stars  \\\n",
       "0           Aql         8  [98036, 97649, 97649, 97278, 97649, 95501, 955...   \n",
       "1           And         5  [677, 3092, 3092, 5447, 9640, 5447, 5447, 4436...   \n",
       "2           Scl         3       [116231, 4577, 4577, 115102, 115102, 116231]   \n",
       "3           Ara         7  [88714, 85792, 85792, 83081, 83081, 82363, 823...   \n",
       "4           Lib         5  [77853, 76333, 76333, 74785, 74785, 72622, 726...   \n",
       "\n",
       "        name                                                 ra  \\\n",
       "0     Aquila  [19.921887, 19.846388, 19.846388, 19.770994, 1...   \n",
       "1  Andromeda  [0.139791, 0.655462, 0.655462, 1.162194, 2.064...   \n",
       "2   Sculptor  [23.549512, 0.976766, 0.976766, 23.313733, 23....   \n",
       "3        Ara  [18.11052, 17.530695, 17.530695, 16.977006, 16...   \n",
       "4      Libra  [15.897093, 15.592104999999998, 15.59210499999...   \n",
       "\n",
       "                                                 dec  \n",
       "0  [6.406763, 8.868322000000003, 8.86832200000000...  \n",
       "1  [29.090432, 30.861024, 30.861024, 35.620558, 4...  \n",
       "2  [-37.818268, -29.357449, -29.357449, -32.53202...  \n",
       "3  [-50.09147700000001, -49.876145, -49.876145, -...  \n",
       "4  [-16.729293, -14.789537, -14.789537, -9.382917...  "
      ]
     },
     "metadata": {},
     "output_type": "display_data"
    }
   ],
   "source": [
    "df['ra'] = ''\n",
    "df['dec'] = ''\n",
    "\n",
    "for index, row in df.iterrows(): \n",
    "    ras, decs = [], []\n",
    "    for star in row['stars']: \n",
    "        temp = hip_df[hip_df['hip']==float(star)]\n",
    "        assert len(temp) == 1\n",
    "        ras.append(temp['ra'].tolist()[0])\n",
    "        decs.append(temp['dec'].tolist()[0])\n",
    "    df.at[index, 'ra'] = ras\n",
    "    df.at[index, 'dec'] = decs\n",
    "    \n",
    "display(df.head())"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 6,
   "metadata": {},
   "outputs": [
    {
     "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>constellation</th>\n",
       "      <th>num_pairs</th>\n",
       "      <th>stars</th>\n",
       "      <th>name</th>\n",
       "      <th>ra</th>\n",
       "      <th>dec</th>\n",
       "      <th>zodiac</th>\n",
       "    </tr>\n",
       "  </thead>\n",
       "  <tbody>\n",
       "    <tr>\n",
       "      <th>0</th>\n",
       "      <td>Aql</td>\n",
       "      <td>8</td>\n",
       "      <td>[98036, 97649, 97649, 97278, 97649, 95501, 955...</td>\n",
       "      <td>Aquila</td>\n",
       "      <td>[19.921887, 19.846388, 19.846388, 19.770994, 1...</td>\n",
       "      <td>[6.406763, 8.868322000000003, 8.86832200000000...</td>\n",
       "      <td>False</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>1</th>\n",
       "      <td>And</td>\n",
       "      <td>5</td>\n",
       "      <td>[677, 3092, 3092, 5447, 9640, 5447, 5447, 4436...</td>\n",
       "      <td>Andromeda</td>\n",
       "      <td>[0.139791, 0.655462, 0.655462, 1.162194, 2.064...</td>\n",
       "      <td>[29.090432, 30.861024, 30.861024, 35.620558, 4...</td>\n",
       "      <td>False</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>2</th>\n",
       "      <td>Scl</td>\n",
       "      <td>3</td>\n",
       "      <td>[116231, 4577, 4577, 115102, 115102, 116231]</td>\n",
       "      <td>Sculptor</td>\n",
       "      <td>[23.549512, 0.976766, 0.976766, 23.313733, 23....</td>\n",
       "      <td>[-37.818268, -29.357449, -29.357449, -32.53202...</td>\n",
       "      <td>False</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>3</th>\n",
       "      <td>Ara</td>\n",
       "      <td>7</td>\n",
       "      <td>[88714, 85792, 85792, 83081, 83081, 82363, 823...</td>\n",
       "      <td>Ara</td>\n",
       "      <td>[18.11052, 17.530695, 17.530695, 16.977006, 16...</td>\n",
       "      <td>[-50.09147700000001, -49.876145, -49.876145, -...</td>\n",
       "      <td>False</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>4</th>\n",
       "      <td>Lib</td>\n",
       "      <td>5</td>\n",
       "      <td>[77853, 76333, 76333, 74785, 74785, 72622, 726...</td>\n",
       "      <td>Libra</td>\n",
       "      <td>[15.897093, 15.592104999999998, 15.59210499999...</td>\n",
       "      <td>[-16.729293, -14.789537, -14.789537, -9.382917...</td>\n",
       "      <td>True</td>\n",
       "    </tr>\n",
       "  </tbody>\n",
       "</table>\n",
       "</div>"
      ],
      "text/plain": [
       "  constellation num_pairs                                              stars  \\\n",
       "0           Aql         8  [98036, 97649, 97649, 97278, 97649, 95501, 955...   \n",
       "1           And         5  [677, 3092, 3092, 5447, 9640, 5447, 5447, 4436...   \n",
       "2           Scl         3       [116231, 4577, 4577, 115102, 115102, 116231]   \n",
       "3           Ara         7  [88714, 85792, 85792, 83081, 83081, 82363, 823...   \n",
       "4           Lib         5  [77853, 76333, 76333, 74785, 74785, 72622, 726...   \n",
       "\n",
       "        name                                                 ra  \\\n",
       "0     Aquila  [19.921887, 19.846388, 19.846388, 19.770994, 1...   \n",
       "1  Andromeda  [0.139791, 0.655462, 0.655462, 1.162194, 2.064...   \n",
       "2   Sculptor  [23.549512, 0.976766, 0.976766, 23.313733, 23....   \n",
       "3        Ara  [18.11052, 17.530695, 17.530695, 16.977006, 16...   \n",
       "4      Libra  [15.897093, 15.592104999999998, 15.59210499999...   \n",
       "\n",
       "                                                 dec  zodiac  \n",
       "0  [6.406763, 8.868322000000003, 8.86832200000000...   False  \n",
       "1  [29.090432, 30.861024, 30.861024, 35.620558, 4...   False  \n",
       "2  [-37.818268, -29.357449, -29.357449, -32.53202...   False  \n",
       "3  [-50.09147700000001, -49.876145, -49.876145, -...   False  \n",
       "4  [-16.729293, -14.789537, -14.789537, -9.382917...    True  "
      ]
     },
     "metadata": {},
     "output_type": "display_data"
    }
   ],
   "source": [
    "zodiacs = ['Aquarius', 'Aries', 'Cancer', 'Capricornus', 'Gemini', 'Leo', 'Libra', \n",
    "           'Pisces', 'Sagittarius', 'Scorpius', 'Taurus', 'Virgo']\n",
    "df['zodiac'] = df['name'].isin(zodiacs)\n",
    "assert df['zodiac'].sum() == 12\n",
    "display(df.head())\n",
    "df.to_csv('./data/processed/asterisms.csv', index=False)"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": null,
   "metadata": {},
   "outputs": [],
   "source": []
  }
 ],
 "metadata": {
  "kernelspec": {
   "display_name": "Python 3",
   "language": "python",
   "name": "python3"
  },
  "language_info": {
   "codemirror_mode": {
    "name": "ipython",
    "version": 3
   },
   "file_extension": ".py",
   "mimetype": "text/x-python",
   "name": "python",
   "nbconvert_exporter": "python",
   "pygments_lexer": "ipython3",
   "version": "3.7.1"
  }
 },
 "nbformat": 4,
 "nbformat_minor": 2
}