Cleaning_Song-dataset.ipynb

{
 "cells": [
  {
   "cell_type": "markdown",
   "metadata": {},
   "source": [
    "# Pre-processing Songs dataset"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 1,
   "metadata": {
    "collapsed": true
   },
   "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": [
      "Raw data size:  (36733, 16)\n"
     ]
    },
    {
     "data": {
      "text/html": [
       "<div>\n",
       "<style>\n",
       "    .dataframe thead tr:only-child th {\n",
       "        text-align: right;\n",
       "    }\n",
       "\n",
       "    .dataframe thead th {\n",
       "        text-align: left;\n",
       "    }\n",
       "\n",
       "    .dataframe tbody tr th {\n",
       "        vertical-align: top;\n",
       "    }\n",
       "</style>\n",
       "<table border=\"1\" class=\"dataframe\">\n",
       "  <thead>\n",
       "    <tr style=\"text-align: right;\">\n",
       "      <th></th>\n",
       "      <th>_id</th>\n",
       "      <th>album</th>\n",
       "      <th>artist</th>\n",
       "      <th>audio_features</th>\n",
       "      <th>context</th>\n",
       "      <th>decades</th>\n",
       "      <th>genres</th>\n",
       "      <th>lyrics_features</th>\n",
       "      <th>moods</th>\n",
       "      <th>name</th>\n",
       "      <th>new_context</th>\n",
       "      <th>picture</th>\n",
       "      <th>recording_id</th>\n",
       "      <th>sub_context</th>\n",
       "      <th>yt_id</th>\n",
       "      <th>yt_views</th>\n",
       "    </tr>\n",
       "  </thead>\n",
       "  <tbody>\n",
       "    <tr>\n",
       "      <th>0</th>\n",
       "      <td>{'$oid': '52fdfb440b9398049f3d7a8c'}</td>\n",
       "      <td>Gangnam Style (강남스타일)</td>\n",
       "      <td>PSY</td>\n",
       "      <td>[11, 0.912744, 0.083704, 132.069, 0.293137, 0....</td>\n",
       "      <td>[work out]</td>\n",
       "      <td>[]</td>\n",
       "      <td>[pop]</td>\n",
       "      <td>[oppa, gangnam, style, gangnam, style, najeneu...</td>\n",
       "      <td>[energetic, motivational]</td>\n",
       "      <td>Gangnam Style (강남스타일)</td>\n",
       "      <td>work out</td>\n",
       "      <td>http://images.musicnet.com/albums/073/463/405/...</td>\n",
       "      <td>50232.0</td>\n",
       "      <td>[working out: cardio]</td>\n",
       "      <td>9bZkp7q19f0</td>\n",
       "      <td>2450112089</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>1</th>\n",
       "      <td>{'$oid': '52fdfb3d0b9398049f3cbc8e'}</td>\n",
       "      <td>Native</td>\n",
       "      <td>OneRepublic</td>\n",
       "      <td>[6, 0.7457039999999999, 0.11995499999999999, 1...</td>\n",
       "      <td>[energetic]</td>\n",
       "      <td>[2012]</td>\n",
       "      <td>[pop]</td>\n",
       "      <td>[lately, i, ve, been, i, ve, been, losing, sle...</td>\n",
       "      <td>[happy]</td>\n",
       "      <td>Counting Stars</td>\n",
       "      <td>energetic</td>\n",
       "      <td>http://images.musicnet.com/albums/081/851/887/...</td>\n",
       "      <td>5839.0</td>\n",
       "      <td>[energy boost]</td>\n",
       "      <td>hT_nvWreIhg</td>\n",
       "      <td>1020297206</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>2</th>\n",
       "      <td>{'$oid': '52fdfb420b9398049f3d3ea5'}</td>\n",
       "      <td>Party Rock Anthem</td>\n",
       "      <td>LMFAO</td>\n",
       "      <td>[5, 0.709932, 0.231455, 130.03, 0.121740999999...</td>\n",
       "      <td>[energetic, energetic, energetic, energetic]</td>\n",
       "      <td>[]</td>\n",
       "      <td>[]</td>\n",
       "      <td>[party, rock, yeah, woo, let, s, go, party, ro...</td>\n",
       "      <td>[happy, celebratory, rowdy]</td>\n",
       "      <td>Party Rock Anthem</td>\n",
       "      <td>housework</td>\n",
       "      <td>http://images.musicnet.com/albums/049/414/127/...</td>\n",
       "      <td>52379.0</td>\n",
       "      <td>[energy boost, pleasing a crowd, housework, dr...</td>\n",
       "      <td>KQ6zr6kCPj8</td>\n",
       "      <td>971128436</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>3</th>\n",
       "      <td>{'$oid': '52fdfb410b9398049f3d1eac'}</td>\n",
       "      <td>Gentleman</td>\n",
       "      <td>PSY</td>\n",
       "      <td>[3, 0.705822, 0.053292, 126.009, 0.126016, 0.0...</td>\n",
       "      <td>[party, party, party, party, party, party]</td>\n",
       "      <td>[2010s]</td>\n",
       "      <td>[dance]</td>\n",
       "      <td>[alagamun, lan, weh, wakun, heya, hanun, gon, ...</td>\n",
       "      <td>[happy, energetic, celebratory]</td>\n",
       "      <td>Gentleman</td>\n",
       "      <td>energetic</td>\n",
       "      <td>http://images.musicnet.com/albums/082/950/461/...</td>\n",
       "      <td>12353.0</td>\n",
       "      <td>[driving in the left lane, energy boost, girls...</td>\n",
       "      <td>ASO_zypdnsQ</td>\n",
       "      <td>892096527</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>4</th>\n",
       "      <td>{'$oid': '52fdfb400b9398049f3d0b19'}</td>\n",
       "      <td>On The Floor</td>\n",
       "      <td>Jennifer Lopez</td>\n",
       "      <td>[3, 0.741757, 0.07277399999999999, 129.985, 0....</td>\n",
       "      <td>[party, party]</td>\n",
       "      <td>[2000s]</td>\n",
       "      <td>[reggaeton]</td>\n",
       "      <td>[j, lo, the, other, side, out, my, mine, it, s...</td>\n",
       "      <td>[energetic]</td>\n",
       "      <td>On The Floor</td>\n",
       "      <td>work out</td>\n",
       "      <td>http://images.musicnet.com/albums/050/131/765/...</td>\n",
       "      <td>29502.0</td>\n",
       "      <td>[working out: cardio, dance party: sweaty]</td>\n",
       "      <td>t4H_Zoh7G5A</td>\n",
       "      <td>873285189</td>\n",
       "    </tr>\n",
       "  </tbody>\n",
       "</table>\n",
       "</div>"
      ],
      "text/plain": [
       "                                    _id                  album  \\\n",
       "0  {'$oid': '52fdfb440b9398049f3d7a8c'}  Gangnam Style (강남스타일)   \n",
       "1  {'$oid': '52fdfb3d0b9398049f3cbc8e'}                 Native   \n",
       "2  {'$oid': '52fdfb420b9398049f3d3ea5'}      Party Rock Anthem   \n",
       "3  {'$oid': '52fdfb410b9398049f3d1eac'}              Gentleman   \n",
       "4  {'$oid': '52fdfb400b9398049f3d0b19'}           On The Floor   \n",
       "\n",
       "           artist                                     audio_features  \\\n",
       "0             PSY  [11, 0.912744, 0.083704, 132.069, 0.293137, 0....   \n",
       "1     OneRepublic  [6, 0.7457039999999999, 0.11995499999999999, 1...   \n",
       "2           LMFAO  [5, 0.709932, 0.231455, 130.03, 0.121740999999...   \n",
       "3             PSY  [3, 0.705822, 0.053292, 126.009, 0.126016, 0.0...   \n",
       "4  Jennifer Lopez  [3, 0.741757, 0.07277399999999999, 129.985, 0....   \n",
       "\n",
       "                                        context  decades       genres  \\\n",
       "0                                    [work out]       []        [pop]   \n",
       "1                                   [energetic]   [2012]        [pop]   \n",
       "2  [energetic, energetic, energetic, energetic]       []           []   \n",
       "3    [party, party, party, party, party, party]  [2010s]      [dance]   \n",
       "4                                [party, party]  [2000s]  [reggaeton]   \n",
       "\n",
       "                                     lyrics_features  \\\n",
       "0  [oppa, gangnam, style, gangnam, style, najeneu...   \n",
       "1  [lately, i, ve, been, i, ve, been, losing, sle...   \n",
       "2  [party, rock, yeah, woo, let, s, go, party, ro...   \n",
       "3  [alagamun, lan, weh, wakun, heya, hanun, gon, ...   \n",
       "4  [j, lo, the, other, side, out, my, mine, it, s...   \n",
       "\n",
       "                             moods                   name new_context  \\\n",
       "0        [energetic, motivational]  Gangnam Style (강남스타일)    work out   \n",
       "1                          [happy]         Counting Stars   energetic   \n",
       "2      [happy, celebratory, rowdy]      Party Rock Anthem   housework   \n",
       "3  [happy, energetic, celebratory]              Gentleman   energetic   \n",
       "4                      [energetic]           On The Floor    work out   \n",
       "\n",
       "                                             picture  recording_id  \\\n",
       "0  http://images.musicnet.com/albums/073/463/405/...       50232.0   \n",
       "1  http://images.musicnet.com/albums/081/851/887/...        5839.0   \n",
       "2  http://images.musicnet.com/albums/049/414/127/...       52379.0   \n",
       "3  http://images.musicnet.com/albums/082/950/461/...       12353.0   \n",
       "4  http://images.musicnet.com/albums/050/131/765/...       29502.0   \n",
       "\n",
       "                                         sub_context        yt_id    yt_views  \n",
       "0                              [working out: cardio]  9bZkp7q19f0  2450112089  \n",
       "1                                     [energy boost]  hT_nvWreIhg  1020297206  \n",
       "2  [energy boost, pleasing a crowd, housework, dr...  KQ6zr6kCPj8   971128436  \n",
       "3  [driving in the left lane, energy boost, girls...  ASO_zypdnsQ   892096527  \n",
       "4         [working out: cardio, dance party: sweaty]  t4H_Zoh7G5A   873285189  "
      ]
     },
     "execution_count": 2,
     "metadata": {},
     "output_type": "execute_result"
    }
   ],
   "source": [
    "song_dataset = pd.read_json('MasterSongList.json')\n",
    "print('Raw data size: ', song_dataset.shape)\n",
    "song_dataset.head()"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 3,
   "metadata": {
    "collapsed": true
   },
   "outputs": [],
   "source": [
    "def consolidateGenres (genre):\n",
    "    if (len(genre) > 0):\n",
    "        gen = genre[0]\n",
    "        return gen.split(':')[0]\n",
    "    else:\n",
    "        return np.nan\n",
    "    \n",
    "song_dataset.loc[:, 'genres']= song_dataset['genres'].apply(consolidateGenres)"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 4,
   "metadata": {
    "scrolled": true
   },
   "outputs": [
    {
     "data": {
      "text/plain": [
       "_id                             {'$oid': '52fdfb400b9398049f3d0db7'}\n",
       "album                                           Radio Disney Jams 12\n",
       "artist                                  Hannah Montana (Miley Cyrus)\n",
       "audio_features                                                    []\n",
       "context            [studying / working, studying / working, study...\n",
       "decades                                                      [2010s]\n",
       "genres                                                           pop\n",
       "lyrics_features    [i, hopped, off, the, plane, at, lax, with, my...\n",
       "moods                                           [happy, celebratory]\n",
       "name                                             Party In The U.S.A.\n",
       "new_context                                                housework\n",
       "picture            http://images.musicnet.com/albums/038/672/411/...\n",
       "recording_id                                                   20459\n",
       "sub_context        [energy boost, pleasing a crowd, housework, wo...\n",
       "yt_id                                                    M11SvDtPBhA\n",
       "yt_views                                                   484967367\n",
       "Name: 23, dtype: object"
      ]
     },
     "execution_count": 4,
     "metadata": {},
     "output_type": "execute_result"
    }
   ],
   "source": [
    "song_dataset.iloc[23, :]"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 5,
   "metadata": {
    "scrolled": true
   },
   "outputs": [
    {
     "data": {
      "text/plain": [
       "0                       energetic,motivational\n",
       "1                                        happy\n",
       "2                      happy,celebratory,rowdy\n",
       "3                  happy,energetic,celebratory\n",
       "4                                    energetic\n",
       "5                              happy,sprightly\n",
       "6                                         warm\n",
       "7                       energetic,motivational\n",
       "8                          seductive,nocturnal\n",
       "9                            happy,celebratory\n",
       "10                                   seductive\n",
       "11                       energetic,celebratory\n",
       "12                     happy,celebratory,campy\n",
       "13                      energetic,motivational\n",
       "14                                   energetic\n",
       "15                           happy,celebratory\n",
       "16                                       happy\n",
       "17                                    soothing\n",
       "18                      energetic,motivational\n",
       "19                           happy,celebratory\n",
       "20                 happy,energetic,celebratory\n",
       "21                       energetic,celebratory\n",
       "22                               sprightly,raw\n",
       "23                           happy,celebratory\n",
       "24                     happy,celebratory,campy\n",
       "25                                      trippy\n",
       "26                                       happy\n",
       "27                           happy,celebratory\n",
       "28                                   energetic\n",
       "29                     happy,celebratory,campy\n",
       "                         ...                  \n",
       "36703              happy,energetic,celebratory\n",
       "36704                       atmospheric,spacey\n",
       "36705                            visceral,lush\n",
       "36706                                   mellow\n",
       "36707                                   classy\n",
       "36708                         earthy,sprightly\n",
       "36709                         earthy,sprightly\n",
       "36710                                    funky\n",
       "36711                                     lush\n",
       "36712                                    rowdy\n",
       "36713                                sprightly\n",
       "36714                             classy,campy\n",
       "36715                                     lush\n",
       "36716          sexual,seductive,nocturnal,lush\n",
       "36717                                sprightly\n",
       "36718                              campy,sweet\n",
       "36719                                 soothing\n",
       "36720                       energetic,visceral\n",
       "36721                           energetic,cold\n",
       "36722    seductive,sad,classy,nocturnal,mellow\n",
       "36723                                    cocky\n",
       "36724                                   trippy\n",
       "36725                              cocky,rowdy\n",
       "36726                     cold,atmospheric,sad\n",
       "36727                            trippy,mellow\n",
       "36728                                   classy\n",
       "36729                       atmospheric,spacey\n",
       "36730                               aggressive\n",
       "36731                                nocturnal\n",
       "36732                               aggressive\n",
       "Name: moods, Length: 36733, dtype: object"
      ]
     },
     "execution_count": 5,
     "metadata": {},
     "output_type": "execute_result"
    }
   ],
   "source": [
    "song_dataset.loc[:, 'moods'] = song_dataset['moods'].apply(lambda x: ','.join(x))\n",
    "song_dataset['moods']"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 6,
   "metadata": {},
   "outputs": [
    {
     "data": {
      "text/html": [
       "<div>\n",
       "<style>\n",
       "    .dataframe thead tr:only-child th {\n",
       "        text-align: right;\n",
       "    }\n",
       "\n",
       "    .dataframe thead th {\n",
       "        text-align: left;\n",
       "    }\n",
       "\n",
       "    .dataframe tbody tr th {\n",
       "        vertical-align: top;\n",
       "    }\n",
       "</style>\n",
       "<table border=\"1\" class=\"dataframe\">\n",
       "  <thead>\n",
       "    <tr style=\"text-align: right;\">\n",
       "      <th></th>\n",
       "      <th>_id</th>\n",
       "      <th>album</th>\n",
       "      <th>artist</th>\n",
       "      <th>audio_features</th>\n",
       "      <th>context</th>\n",
       "      <th>decades</th>\n",
       "      <th>genres</th>\n",
       "      <th>lyrics_features</th>\n",
       "      <th>moods</th>\n",
       "      <th>name</th>\n",
       "      <th>new_context</th>\n",
       "      <th>picture</th>\n",
       "      <th>recording_id</th>\n",
       "      <th>sub_context</th>\n",
       "      <th>yt_id</th>\n",
       "      <th>yt_views</th>\n",
       "    </tr>\n",
       "  </thead>\n",
       "  <tbody>\n",
       "  </tbody>\n",
       "</table>\n",
       "</div>"
      ],
      "text/plain": [
       "Empty DataFrame\n",
       "Columns: [_id, album, artist, audio_features, context, decades, genres, lyrics_features, moods, name, new_context, picture, recording_id, sub_context, yt_id, yt_views]\n",
       "Index: []"
      ]
     },
     "execution_count": 6,
     "metadata": {},
     "output_type": "execute_result"
    }
   ],
   "source": [
    "song_dataset[song_dataset['moods'].isnull()]"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 7,
   "metadata": {
    "collapsed": true
   },
   "outputs": [],
   "source": [
    "audioFeatures_names = ['key', 'energy', 'liveliness', 'tempo', 'speechiness', 'acousticness', 'instrumentalness', \n",
    "                  'time_signature', 'duration', 'loudness', 'valence', 'danceability', 'mode', \n",
    "                  'time_signature_confidence', 'tempo_confidence', 'key_confidence', 'mode_confidence']"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 8,
   "metadata": {
    "collapsed": true
   },
   "outputs": [],
   "source": [
    "def audioFeatureFrame (features_val, features_names):\n",
    "    values = [val for val in features_val]\n",
    "    frame = pd.DataFrame(values, columns = features_names)\n",
    "    return frame\n",
    "    "
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 9,
   "metadata": {
    "scrolled": true
   },
   "outputs": [
    {
     "data": {
      "text/html": [
       "<div>\n",
       "<style>\n",
       "    .dataframe thead tr:only-child th {\n",
       "        text-align: right;\n",
       "    }\n",
       "\n",
       "    .dataframe thead th {\n",
       "        text-align: left;\n",
       "    }\n",
       "\n",
       "    .dataframe tbody tr th {\n",
       "        vertical-align: top;\n",
       "    }\n",
       "</style>\n",
       "<table border=\"1\" class=\"dataframe\">\n",
       "  <thead>\n",
       "    <tr style=\"text-align: right;\">\n",
       "      <th></th>\n",
       "      <th>key</th>\n",
       "      <th>energy</th>\n",
       "      <th>liveliness</th>\n",
       "      <th>tempo</th>\n",
       "      <th>speechiness</th>\n",
       "      <th>acousticness</th>\n",
       "      <th>instrumentalness</th>\n",
       "      <th>time_signature</th>\n",
       "      <th>duration</th>\n",
       "      <th>loudness</th>\n",
       "      <th>valence</th>\n",
       "      <th>danceability</th>\n",
       "      <th>mode</th>\n",
       "      <th>time_signature_confidence</th>\n",
       "      <th>tempo_confidence</th>\n",
       "      <th>key_confidence</th>\n",
       "      <th>mode_confidence</th>\n",
       "    </tr>\n",
       "  </thead>\n",
       "  <tbody>\n",
       "    <tr>\n",
       "      <th>0</th>\n",
       "      <td>11.0</td>\n",
       "      <td>0.912744</td>\n",
       "      <td>0.083704</td>\n",
       "      <td>132.069</td>\n",
       "      <td>0.293137</td>\n",
       "      <td>0.005423</td>\n",
       "      <td>0.000001</td>\n",
       "      <td>0.0</td>\n",
       "      <td>4.0</td>\n",
       "      <td>218.30667</td>\n",
       "      <td>-3.890</td>\n",
       "      <td>0.752186</td>\n",
       "      <td>0.726920</td>\n",
       "      <td>0.552</td>\n",
       "      <td>0.541</td>\n",
       "      <td>1.000</td>\n",
       "      <td>1.000</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>1</th>\n",
       "      <td>6.0</td>\n",
       "      <td>0.745704</td>\n",
       "      <td>0.119955</td>\n",
       "      <td>100.008</td>\n",
       "      <td>0.046255</td>\n",
       "      <td>0.026230</td>\n",
       "      <td>0.012727</td>\n",
       "      <td>1.0</td>\n",
       "      <td>4.0</td>\n",
       "      <td>235.06086</td>\n",
       "      <td>-7.687</td>\n",
       "      <td>0.351282</td>\n",
       "      <td>0.691817</td>\n",
       "      <td>0.737</td>\n",
       "      <td>0.634</td>\n",
       "      <td>0.796</td>\n",
       "      <td>1.000</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>2</th>\n",
       "      <td>5.0</td>\n",
       "      <td>0.709932</td>\n",
       "      <td>0.231455</td>\n",
       "      <td>130.030</td>\n",
       "      <td>0.121741</td>\n",
       "      <td>0.036662</td>\n",
       "      <td>0.000000</td>\n",
       "      <td>0.0</td>\n",
       "      <td>4.0</td>\n",
       "      <td>232.46104</td>\n",
       "      <td>-5.150</td>\n",
       "      <td>0.374390</td>\n",
       "      <td>0.704729</td>\n",
       "      <td>0.565</td>\n",
       "      <td>0.565</td>\n",
       "      <td>0.743</td>\n",
       "      <td>1.000</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>3</th>\n",
       "      <td>3.0</td>\n",
       "      <td>0.705822</td>\n",
       "      <td>0.053292</td>\n",
       "      <td>126.009</td>\n",
       "      <td>0.126016</td>\n",
       "      <td>0.001966</td>\n",
       "      <td>0.000000</td>\n",
       "      <td>0.0</td>\n",
       "      <td>4.0</td>\n",
       "      <td>194.09333</td>\n",
       "      <td>-3.898</td>\n",
       "      <td>0.592798</td>\n",
       "      <td>0.875137</td>\n",
       "      <td>0.004</td>\n",
       "      <td>0.114</td>\n",
       "      <td>1.000</td>\n",
       "      <td>0.742</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>4</th>\n",
       "      <td>3.0</td>\n",
       "      <td>0.741757</td>\n",
       "      <td>0.072774</td>\n",
       "      <td>129.985</td>\n",
       "      <td>0.051255</td>\n",
       "      <td>0.096732</td>\n",
       "      <td>0.000474</td>\n",
       "      <td>0.0</td>\n",
       "      <td>4.0</td>\n",
       "      <td>285.42667</td>\n",
       "      <td>-5.860</td>\n",
       "      <td>0.585630</td>\n",
       "      <td>0.730711</td>\n",
       "      <td>0.271</td>\n",
       "      <td>0.324</td>\n",
       "      <td>0.822</td>\n",
       "      <td>1.000</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>5</th>\n",
       "      <td>8.0</td>\n",
       "      <td>0.733856</td>\n",
       "      <td>0.093043</td>\n",
       "      <td>174.952</td>\n",
       "      <td>0.058137</td>\n",
       "      <td>0.293325</td>\n",
       "      <td>0.000000</td>\n",
       "      <td>0.0</td>\n",
       "      <td>4.0</td>\n",
       "      <td>188.94281</td>\n",
       "      <td>-4.312</td>\n",
       "      <td>0.953424</td>\n",
       "      <td>0.785102</td>\n",
       "      <td>0.177</td>\n",
       "      <td>0.404</td>\n",
       "      <td>0.826</td>\n",
       "      <td>1.000</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>6</th>\n",
       "      <td>8.0</td>\n",
       "      <td>0.777375</td>\n",
       "      <td>0.054104</td>\n",
       "      <td>104.946</td>\n",
       "      <td>0.029302</td>\n",
       "      <td>0.130350</td>\n",
       "      <td>0.000000</td>\n",
       "      <td>1.0</td>\n",
       "      <td>4.0</td>\n",
       "      <td>228.29333</td>\n",
       "      <td>-5.112</td>\n",
       "      <td>0.525632</td>\n",
       "      <td>0.729051</td>\n",
       "      <td>0.817</td>\n",
       "      <td>0.672</td>\n",
       "      <td>0.394</td>\n",
       "      <td>0.963</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>7</th>\n",
       "      <td>7.0</td>\n",
       "      <td>0.585564</td>\n",
       "      <td>0.108297</td>\n",
       "      <td>120.014</td>\n",
       "      <td>0.038924</td>\n",
       "      <td>0.011707</td>\n",
       "      <td>0.000005</td>\n",
       "      <td>1.0</td>\n",
       "      <td>4.0</td>\n",
       "      <td>193.57333</td>\n",
       "      <td>-6.583</td>\n",
       "      <td>0.622176</td>\n",
       "      <td>0.781822</td>\n",
       "      <td>0.970</td>\n",
       "      <td>0.861</td>\n",
       "      <td>0.792</td>\n",
       "      <td>1.000</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>8</th>\n",
       "      <td>0.0</td>\n",
       "      <td>0.418212</td>\n",
       "      <td>0.105322</td>\n",
       "      <td>129.054</td>\n",
       "      <td>0.045461</td>\n",
       "      <td>0.548385</td>\n",
       "      <td>0.000120</td>\n",
       "      <td>0.0</td>\n",
       "      <td>4.0</td>\n",
       "      <td>248.28000</td>\n",
       "      <td>-7.284</td>\n",
       "      <td>0.440790</td>\n",
       "      <td>0.841824</td>\n",
       "      <td>0.011</td>\n",
       "      <td>0.303</td>\n",
       "      <td>0.843</td>\n",
       "      <td>1.000</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>9</th>\n",
       "      <td>4.0</td>\n",
       "      <td>0.814030</td>\n",
       "      <td>0.079196</td>\n",
       "      <td>124.991</td>\n",
       "      <td>0.072440</td>\n",
       "      <td>0.005355</td>\n",
       "      <td>0.000000</td>\n",
       "      <td>1.0</td>\n",
       "      <td>4.0</td>\n",
       "      <td>198.25333</td>\n",
       "      <td>-1.985</td>\n",
       "      <td>0.838512</td>\n",
       "      <td>0.700860</td>\n",
       "      <td>0.429</td>\n",
       "      <td>0.375</td>\n",
       "      <td>0.740</td>\n",
       "      <td>1.000</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>10</th>\n",
       "      <td>11.0</td>\n",
       "      <td>0.903055</td>\n",
       "      <td>0.629349</td>\n",
       "      <td>96.082</td>\n",
       "      <td>0.074866</td>\n",
       "      <td>0.306908</td>\n",
       "      <td>0.000000</td>\n",
       "      <td>1.0</td>\n",
       "      <td>4.0</td>\n",
       "      <td>167.98667</td>\n",
       "      <td>-6.240</td>\n",
       "      <td>0.898559</td>\n",
       "      <td>0.634209</td>\n",
       "      <td>0.502</td>\n",
       "      <td>0.518</td>\n",
       "      <td>0.437</td>\n",
       "      <td>0.992</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>11</th>\n",
       "      <td>11.0</td>\n",
       "      <td>0.689804</td>\n",
       "      <td>0.112703</td>\n",
       "      <td>92.032</td>\n",
       "      <td>0.042006</td>\n",
       "      <td>0.001365</td>\n",
       "      <td>0.000000</td>\n",
       "      <td>0.0</td>\n",
       "      <td>4.0</td>\n",
       "      <td>225.02667</td>\n",
       "      <td>-5.502</td>\n",
       "      <td>0.422368</td>\n",
       "      <td>0.565887</td>\n",
       "      <td>0.662</td>\n",
       "      <td>0.567</td>\n",
       "      <td>0.270</td>\n",
       "      <td>0.991</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>12</th>\n",
       "      <td>8.0</td>\n",
       "      <td>0.869554</td>\n",
       "      <td>0.107945</td>\n",
       "      <td>124.065</td>\n",
       "      <td>0.051083</td>\n",
       "      <td>0.132712</td>\n",
       "      <td>0.000000</td>\n",
       "      <td>1.0</td>\n",
       "      <td>4.0</td>\n",
       "      <td>227.97333</td>\n",
       "      <td>-3.739</td>\n",
       "      <td>0.639499</td>\n",
       "      <td>0.636239</td>\n",
       "      <td>0.239</td>\n",
       "      <td>0.416</td>\n",
       "      <td>0.669</td>\n",
       "      <td>1.000</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>13</th>\n",
       "      <td>0.0</td>\n",
       "      <td>0.878755</td>\n",
       "      <td>0.057114</td>\n",
       "      <td>119.000</td>\n",
       "      <td>0.040855</td>\n",
       "      <td>0.000579</td>\n",
       "      <td>0.000000</td>\n",
       "      <td>1.0</td>\n",
       "      <td>4.0</td>\n",
       "      <td>295.90667</td>\n",
       "      <td>-7.769</td>\n",
       "      <td>0.726486</td>\n",
       "      <td>0.634661</td>\n",
       "      <td>0.391</td>\n",
       "      <td>0.334</td>\n",
       "      <td>0.763</td>\n",
       "      <td>0.927</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>14</th>\n",
       "      <td>5.0</td>\n",
       "      <td>0.908538</td>\n",
       "      <td>0.824374</td>\n",
       "      <td>127.914</td>\n",
       "      <td>0.063644</td>\n",
       "      <td>0.004802</td>\n",
       "      <td>0.000000</td>\n",
       "      <td>1.0</td>\n",
       "      <td>4.0</td>\n",
       "      <td>231.57333</td>\n",
       "      <td>-2.106</td>\n",
       "      <td>0.273037</td>\n",
       "      <td>0.611672</td>\n",
       "      <td>0.135</td>\n",
       "      <td>0.218</td>\n",
       "      <td>0.832</td>\n",
       "      <td>1.000</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>15</th>\n",
       "      <td>5.0</td>\n",
       "      <td>0.838853</td>\n",
       "      <td>0.161256</td>\n",
       "      <td>109.007</td>\n",
       "      <td>0.043218</td>\n",
       "      <td>0.014582</td>\n",
       "      <td>0.000000</td>\n",
       "      <td>1.0</td>\n",
       "      <td>4.0</td>\n",
       "      <td>220.28000</td>\n",
       "      <td>-5.426</td>\n",
       "      <td>0.456284</td>\n",
       "      <td>0.636132</td>\n",
       "      <td>0.459</td>\n",
       "      <td>0.454</td>\n",
       "      <td>0.577</td>\n",
       "      <td>0.995</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>16</th>\n",
       "      <td>1.0</td>\n",
       "      <td>0.625657</td>\n",
       "      <td>0.418317</td>\n",
       "      <td>80.082</td>\n",
       "      <td>0.034819</td>\n",
       "      <td>0.003484</td>\n",
       "      <td>0.000000</td>\n",
       "      <td>0.0</td>\n",
       "      <td>4.0</td>\n",
       "      <td>230.25333</td>\n",
       "      <td>-7.264</td>\n",
       "      <td>0.486710</td>\n",
       "      <td>0.585064</td>\n",
       "      <td>0.440</td>\n",
       "      <td>0.613</td>\n",
       "      <td>0.000</td>\n",
       "      <td>1.000</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>17</th>\n",
       "      <td>9.0</td>\n",
       "      <td>0.323877</td>\n",
       "      <td>0.109264</td>\n",
       "      <td>135.116</td>\n",
       "      <td>0.027887</td>\n",
       "      <td>0.893351</td>\n",
       "      <td>0.000000</td>\n",
       "      <td>1.0</td>\n",
       "      <td>4.0</td>\n",
       "      <td>285.26667</td>\n",
       "      <td>-8.237</td>\n",
       "      <td>0.297851</td>\n",
       "      <td>0.561547</td>\n",
       "      <td>0.358</td>\n",
       "      <td>0.526</td>\n",
       "      <td>0.465</td>\n",
       "      <td>0.720</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>18</th>\n",
       "      <td>3.0</td>\n",
       "      <td>0.640167</td>\n",
       "      <td>0.051898</td>\n",
       "      <td>127.963</td>\n",
       "      <td>0.049790</td>\n",
       "      <td>0.029364</td>\n",
       "      <td>0.008626</td>\n",
       "      <td>0.0</td>\n",
       "      <td>4.0</td>\n",
       "      <td>214.14735</td>\n",
       "      <td>-7.490</td>\n",
       "      <td>0.684089</td>\n",
       "      <td>0.846949</td>\n",
       "      <td>0.000</td>\n",
       "      <td>0.245</td>\n",
       "      <td>0.858</td>\n",
       "      <td>1.000</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>19</th>\n",
       "      <td>11.0</td>\n",
       "      <td>0.821101</td>\n",
       "      <td>0.499412</td>\n",
       "      <td>127.016</td>\n",
       "      <td>0.218697</td>\n",
       "      <td>0.249091</td>\n",
       "      <td>0.000004</td>\n",
       "      <td>1.0</td>\n",
       "      <td>4.0</td>\n",
       "      <td>200.45333</td>\n",
       "      <td>-5.589</td>\n",
       "      <td>0.663748</td>\n",
       "      <td>0.717298</td>\n",
       "      <td>0.648</td>\n",
       "      <td>0.514</td>\n",
       "      <td>0.501</td>\n",
       "      <td>1.000</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>20</th>\n",
       "      <td>0.0</td>\n",
       "      <td>0.616411</td>\n",
       "      <td>0.171423</td>\n",
       "      <td>130.009</td>\n",
       "      <td>0.059577</td>\n",
       "      <td>0.058936</td>\n",
       "      <td>0.000065</td>\n",
       "      <td>1.0</td>\n",
       "      <td>4.0</td>\n",
       "      <td>284.41333</td>\n",
       "      <td>-7.443</td>\n",
       "      <td>0.476111</td>\n",
       "      <td>0.789760</td>\n",
       "      <td>0.499</td>\n",
       "      <td>0.489</td>\n",
       "      <td>0.708</td>\n",
       "      <td>1.000</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>21</th>\n",
       "      <td>8.0</td>\n",
       "      <td>0.932226</td>\n",
       "      <td>0.303175</td>\n",
       "      <td>129.025</td>\n",
       "      <td>0.164729</td>\n",
       "      <td>0.180590</td>\n",
       "      <td>0.000002</td>\n",
       "      <td>1.0</td>\n",
       "      <td>4.0</td>\n",
       "      <td>256.28000</td>\n",
       "      <td>-3.257</td>\n",
       "      <td>0.529215</td>\n",
       "      <td>0.673568</td>\n",
       "      <td>0.294</td>\n",
       "      <td>0.401</td>\n",
       "      <td>0.555</td>\n",
       "      <td>1.000</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>22</th>\n",
       "      <td>10.0</td>\n",
       "      <td>0.839959</td>\n",
       "      <td>0.372695</td>\n",
       "      <td>157.625</td>\n",
       "      <td>0.101830</td>\n",
       "      <td>0.018690</td>\n",
       "      <td>0.000000</td>\n",
       "      <td>0.0</td>\n",
       "      <td>4.0</td>\n",
       "      <td>240.04000</td>\n",
       "      <td>-10.859</td>\n",
       "      <td>0.854567</td>\n",
       "      <td>0.477058</td>\n",
       "      <td>0.000</td>\n",
       "      <td>0.116</td>\n",
       "      <td>0.544</td>\n",
       "      <td>0.973</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>23</th>\n",
       "      <td>NaN</td>\n",
       "      <td>NaN</td>\n",
       "      <td>NaN</td>\n",
       "      <td>NaN</td>\n",
       "      <td>NaN</td>\n",
       "      <td>NaN</td>\n",
       "      <td>NaN</td>\n",
       "      <td>NaN</td>\n",
       "      <td>NaN</td>\n",
       "      <td>NaN</td>\n",
       "      <td>NaN</td>\n",
       "      <td>NaN</td>\n",
       "      <td>NaN</td>\n",
       "      <td>NaN</td>\n",
       "      <td>NaN</td>\n",
       "      <td>NaN</td>\n",
       "      <td>NaN</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>24</th>\n",
       "      <td>2.0</td>\n",
       "      <td>0.777544</td>\n",
       "      <td>0.063793</td>\n",
       "      <td>100.023</td>\n",
       "      <td>0.073157</td>\n",
       "      <td>0.223289</td>\n",
       "      <td>0.000000</td>\n",
       "      <td>1.0</td>\n",
       "      <td>4.0</td>\n",
       "      <td>263.02667</td>\n",
       "      <td>-3.070</td>\n",
       "      <td>0.591259</td>\n",
       "      <td>0.691084</td>\n",
       "      <td>0.474</td>\n",
       "      <td>0.518</td>\n",
       "      <td>0.595</td>\n",
       "      <td>1.000</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>25</th>\n",
       "      <td>11.0</td>\n",
       "      <td>0.398953</td>\n",
       "      <td>0.170640</td>\n",
       "      <td>77.821</td>\n",
       "      <td>0.033343</td>\n",
       "      <td>0.007855</td>\n",
       "      <td>0.457147</td>\n",
       "      <td>1.0</td>\n",
       "      <td>4.0</td>\n",
       "      <td>548.03156</td>\n",
       "      <td>-19.753</td>\n",
       "      <td>0.243535</td>\n",
       "      <td>0.198917</td>\n",
       "      <td>0.901</td>\n",
       "      <td>0.676</td>\n",
       "      <td>0.362</td>\n",
       "      <td>0.996</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>26</th>\n",
       "      <td>2.0</td>\n",
       "      <td>0.555380</td>\n",
       "      <td>0.132021</td>\n",
       "      <td>95.000</td>\n",
       "      <td>0.050934</td>\n",
       "      <td>0.397652</td>\n",
       "      <td>0.000278</td>\n",
       "      <td>1.0</td>\n",
       "      <td>4.0</td>\n",
       "      <td>242.73333</td>\n",
       "      <td>-7.275</td>\n",
       "      <td>0.454030</td>\n",
       "      <td>0.779140</td>\n",
       "      <td>0.565</td>\n",
       "      <td>0.640</td>\n",
       "      <td>0.414</td>\n",
       "      <td>1.000</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>27</th>\n",
       "      <td>5.0</td>\n",
       "      <td>0.785597</td>\n",
       "      <td>0.186539</td>\n",
       "      <td>175.041</td>\n",
       "      <td>0.147627</td>\n",
       "      <td>0.021278</td>\n",
       "      <td>0.000001</td>\n",
       "      <td>1.0</td>\n",
       "      <td>4.0</td>\n",
       "      <td>188.50667</td>\n",
       "      <td>-4.841</td>\n",
       "      <td>0.639066</td>\n",
       "      <td>0.634571</td>\n",
       "      <td>0.220</td>\n",
       "      <td>0.303</td>\n",
       "      <td>0.417</td>\n",
       "      <td>1.000</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>28</th>\n",
       "      <td>7.0</td>\n",
       "      <td>0.877359</td>\n",
       "      <td>0.061576</td>\n",
       "      <td>125.013</td>\n",
       "      <td>0.081238</td>\n",
       "      <td>0.085122</td>\n",
       "      <td>0.001956</td>\n",
       "      <td>1.0</td>\n",
       "      <td>4.0</td>\n",
       "      <td>224.69333</td>\n",
       "      <td>-3.813</td>\n",
       "      <td>0.924909</td>\n",
       "      <td>0.804825</td>\n",
       "      <td>0.430</td>\n",
       "      <td>0.536</td>\n",
       "      <td>0.826</td>\n",
       "      <td>1.000</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>29</th>\n",
       "      <td>7.0</td>\n",
       "      <td>0.736727</td>\n",
       "      <td>0.069281</td>\n",
       "      <td>125.928</td>\n",
       "      <td>0.044913</td>\n",
       "      <td>0.054524</td>\n",
       "      <td>0.000014</td>\n",
       "      <td>1.0</td>\n",
       "      <td>4.0</td>\n",
       "      <td>234.62667</td>\n",
       "      <td>-3.416</td>\n",
       "      <td>0.687324</td>\n",
       "      <td>0.763795</td>\n",
       "      <td>0.490</td>\n",
       "      <td>0.708</td>\n",
       "      <td>0.957</td>\n",
       "      <td>1.000</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>...</th>\n",
       "      <td>...</td>\n",
       "      <td>...</td>\n",
       "      <td>...</td>\n",
       "      <td>...</td>\n",
       "      <td>...</td>\n",
       "      <td>...</td>\n",
       "      <td>...</td>\n",
       "      <td>...</td>\n",
       "      <td>...</td>\n",
       "      <td>...</td>\n",
       "      <td>...</td>\n",
       "      <td>...</td>\n",
       "      <td>...</td>\n",
       "      <td>...</td>\n",
       "      <td>...</td>\n",
       "      <td>...</td>\n",
       "      <td>...</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>36703</th>\n",
       "      <td>NaN</td>\n",
       "      <td>NaN</td>\n",
       "      <td>NaN</td>\n",
       "      <td>NaN</td>\n",
       "      <td>NaN</td>\n",
       "      <td>NaN</td>\n",
       "      <td>NaN</td>\n",
       "      <td>NaN</td>\n",
       "      <td>NaN</td>\n",
       "      <td>NaN</td>\n",
       "      <td>NaN</td>\n",
       "      <td>NaN</td>\n",
       "      <td>NaN</td>\n",
       "      <td>NaN</td>\n",
       "      <td>NaN</td>\n",
       "      <td>NaN</td>\n",
       "      <td>NaN</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>36704</th>\n",
       "      <td>NaN</td>\n",
       "      <td>NaN</td>\n",
       "      <td>NaN</td>\n",
       "      <td>NaN</td>\n",
       "      <td>NaN</td>\n",
       "      <td>NaN</td>\n",
       "      <td>NaN</td>\n",
       "      <td>NaN</td>\n",
       "      <td>NaN</td>\n",
       "      <td>NaN</td>\n",
       "      <td>NaN</td>\n",
       "      <td>NaN</td>\n",
       "      <td>NaN</td>\n",
       "      <td>NaN</td>\n",
       "      <td>NaN</td>\n",
       "      <td>NaN</td>\n",
       "      <td>NaN</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>36705</th>\n",
       "      <td>NaN</td>\n",
       "      <td>NaN</td>\n",
       "      <td>NaN</td>\n",
       "      <td>NaN</td>\n",
       "      <td>NaN</td>\n",
       "      <td>NaN</td>\n",
       "      <td>NaN</td>\n",
       "      <td>NaN</td>\n",
       "      <td>NaN</td>\n",
       "      <td>NaN</td>\n",
       "      <td>NaN</td>\n",
       "      <td>NaN</td>\n",
       "      <td>NaN</td>\n",
       "      <td>NaN</td>\n",
       "      <td>NaN</td>\n",
       "      <td>NaN</td>\n",
       "      <td>NaN</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>36706</th>\n",
       "      <td>NaN</td>\n",
       "      <td>NaN</td>\n",
       "      <td>NaN</td>\n",
       "      <td>NaN</td>\n",
       "      <td>NaN</td>\n",
       "      <td>NaN</td>\n",
       "      <td>NaN</td>\n",
       "      <td>NaN</td>\n",
       "      <td>NaN</td>\n",
       "      <td>NaN</td>\n",
       "      <td>NaN</td>\n",
       "      <td>NaN</td>\n",
       "      <td>NaN</td>\n",
       "      <td>NaN</td>\n",
       "      <td>NaN</td>\n",
       "      <td>NaN</td>\n",
       "      <td>NaN</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>36707</th>\n",
       "      <td>NaN</td>\n",
       "      <td>NaN</td>\n",
       "      <td>NaN</td>\n",
       "      <td>NaN</td>\n",
       "      <td>NaN</td>\n",
       "      <td>NaN</td>\n",
       "      <td>NaN</td>\n",
       "      <td>NaN</td>\n",
       "      <td>NaN</td>\n",
       "      <td>NaN</td>\n",
       "      <td>NaN</td>\n",
       "      <td>NaN</td>\n",
       "      <td>NaN</td>\n",
       "      <td>NaN</td>\n",
       "      <td>NaN</td>\n",
       "      <td>NaN</td>\n",
       "      <td>NaN</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>36708</th>\n",
       "      <td>NaN</td>\n",
       "      <td>NaN</td>\n",
       "      <td>NaN</td>\n",
       "      <td>NaN</td>\n",
       "      <td>NaN</td>\n",
       "      <td>NaN</td>\n",
       "      <td>NaN</td>\n",
       "      <td>NaN</td>\n",
       "      <td>NaN</td>\n",
       "      <td>NaN</td>\n",
       "      <td>NaN</td>\n",
       "      <td>NaN</td>\n",
       "      <td>NaN</td>\n",
       "      <td>NaN</td>\n",
       "      <td>NaN</td>\n",
       "      <td>NaN</td>\n",
       "      <td>NaN</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>36709</th>\n",
       "      <td>NaN</td>\n",
       "      <td>NaN</td>\n",
       "      <td>NaN</td>\n",
       "      <td>NaN</td>\n",
       "      <td>NaN</td>\n",
       "      <td>NaN</td>\n",
       "      <td>NaN</td>\n",
       "      <td>NaN</td>\n",
       "      <td>NaN</td>\n",
       "      <td>NaN</td>\n",
       "      <td>NaN</td>\n",
       "      <td>NaN</td>\n",
       "      <td>NaN</td>\n",
       "      <td>NaN</td>\n",
       "      <td>NaN</td>\n",
       "      <td>NaN</td>\n",
       "      <td>NaN</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>36710</th>\n",
       "      <td>NaN</td>\n",
       "      <td>NaN</td>\n",
       "      <td>NaN</td>\n",
       "      <td>NaN</td>\n",
       "      <td>NaN</td>\n",
       "      <td>NaN</td>\n",
       "      <td>NaN</td>\n",
       "      <td>NaN</td>\n",
       "      <td>NaN</td>\n",
       "      <td>NaN</td>\n",
       "      <td>NaN</td>\n",
       "      <td>NaN</td>\n",
       "      <td>NaN</td>\n",
       "      <td>NaN</td>\n",
       "      <td>NaN</td>\n",
       "      <td>NaN</td>\n",
       "      <td>NaN</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>36711</th>\n",
       "      <td>NaN</td>\n",
       "      <td>NaN</td>\n",
       "      <td>NaN</td>\n",
       "      <td>NaN</td>\n",
       "      <td>NaN</td>\n",
       "      <td>NaN</td>\n",
       "      <td>NaN</td>\n",
       "      <td>NaN</td>\n",
       "      <td>NaN</td>\n",
       "      <td>NaN</td>\n",
       "      <td>NaN</td>\n",
       "      <td>NaN</td>\n",
       "      <td>NaN</td>\n",
       "      <td>NaN</td>\n",
       "      <td>NaN</td>\n",
       "      <td>NaN</td>\n",
       "      <td>NaN</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>36712</th>\n",
       "      <td>NaN</td>\n",
       "      <td>NaN</td>\n",
       "      <td>NaN</td>\n",
       "      <td>NaN</td>\n",
       "      <td>NaN</td>\n",
       "      <td>NaN</td>\n",
       "      <td>NaN</td>\n",
       "      <td>NaN</td>\n",
       "      <td>NaN</td>\n",
       "      <td>NaN</td>\n",
       "      <td>NaN</td>\n",
       "      <td>NaN</td>\n",
       "      <td>NaN</td>\n",
       "      <td>NaN</td>\n",
       "      <td>NaN</td>\n",
       "      <td>NaN</td>\n",
       "      <td>NaN</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>36713</th>\n",
       "      <td>NaN</td>\n",
       "      <td>NaN</td>\n",
       "      <td>NaN</td>\n",
       "      <td>NaN</td>\n",
       "      <td>NaN</td>\n",
       "      <td>NaN</td>\n",
       "      <td>NaN</td>\n",
       "      <td>NaN</td>\n",
       "      <td>NaN</td>\n",
       "      <td>NaN</td>\n",
       "      <td>NaN</td>\n",
       "      <td>NaN</td>\n",
       "      <td>NaN</td>\n",
       "      <td>NaN</td>\n",
       "      <td>NaN</td>\n",
       "      <td>NaN</td>\n",
       "      <td>NaN</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>36714</th>\n",
       "      <td>NaN</td>\n",
       "      <td>NaN</td>\n",
       "      <td>NaN</td>\n",
       "      <td>NaN</td>\n",
       "      <td>NaN</td>\n",
       "      <td>NaN</td>\n",
       "      <td>NaN</td>\n",
       "      <td>NaN</td>\n",
       "      <td>NaN</td>\n",
       "      <td>NaN</td>\n",
       "      <td>NaN</td>\n",
       "      <td>NaN</td>\n",
       "      <td>NaN</td>\n",
       "      <td>NaN</td>\n",
       "      <td>NaN</td>\n",
       "      <td>NaN</td>\n",
       "      <td>NaN</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>36715</th>\n",
       "      <td>NaN</td>\n",
       "      <td>NaN</td>\n",
       "      <td>NaN</td>\n",
       "      <td>NaN</td>\n",
       "      <td>NaN</td>\n",
       "      <td>NaN</td>\n",
       "      <td>NaN</td>\n",
       "      <td>NaN</td>\n",
       "      <td>NaN</td>\n",
       "      <td>NaN</td>\n",
       "      <td>NaN</td>\n",
       "      <td>NaN</td>\n",
       "      <td>NaN</td>\n",
       "      <td>NaN</td>\n",
       "      <td>NaN</td>\n",
       "      <td>NaN</td>\n",
       "      <td>NaN</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>36716</th>\n",
       "      <td>NaN</td>\n",
       "      <td>NaN</td>\n",
       "      <td>NaN</td>\n",
       "      <td>NaN</td>\n",
       "      <td>NaN</td>\n",
       "      <td>NaN</td>\n",
       "      <td>NaN</td>\n",
       "      <td>NaN</td>\n",
       "      <td>NaN</td>\n",
       "      <td>NaN</td>\n",
       "      <td>NaN</td>\n",
       "      <td>NaN</td>\n",
       "      <td>NaN</td>\n",
       "      <td>NaN</td>\n",
       "      <td>NaN</td>\n",
       "      <td>NaN</td>\n",
       "      <td>NaN</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>36717</th>\n",
       "      <td>NaN</td>\n",
       "      <td>NaN</td>\n",
       "      <td>NaN</td>\n",
       "      <td>NaN</td>\n",
       "      <td>NaN</td>\n",
       "      <td>NaN</td>\n",
       "      <td>NaN</td>\n",
       "      <td>NaN</td>\n",
       "      <td>NaN</td>\n",
       "      <td>NaN</td>\n",
       "      <td>NaN</td>\n",
       "      <td>NaN</td>\n",
       "      <td>NaN</td>\n",
       "      <td>NaN</td>\n",
       "      <td>NaN</td>\n",
       "      <td>NaN</td>\n",
       "      <td>NaN</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>36718</th>\n",
       "      <td>NaN</td>\n",
       "      <td>NaN</td>\n",
       "      <td>NaN</td>\n",
       "      <td>NaN</td>\n",
       "      <td>NaN</td>\n",
       "      <td>NaN</td>\n",
       "      <td>NaN</td>\n",
       "      <td>NaN</td>\n",
       "      <td>NaN</td>\n",
       "      <td>NaN</td>\n",
       "      <td>NaN</td>\n",
       "      <td>NaN</td>\n",
       "      <td>NaN</td>\n",
       "      <td>NaN</td>\n",
       "      <td>NaN</td>\n",
       "      <td>NaN</td>\n",
       "      <td>NaN</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>36719</th>\n",
       "      <td>NaN</td>\n",
       "      <td>NaN</td>\n",
       "      <td>NaN</td>\n",
       "      <td>NaN</td>\n",
       "      <td>NaN</td>\n",
       "      <td>NaN</td>\n",
       "      <td>NaN</td>\n",
       "      <td>NaN</td>\n",
       "      <td>NaN</td>\n",
       "      <td>NaN</td>\n",
       "      <td>NaN</td>\n",
       "      <td>NaN</td>\n",
       "      <td>NaN</td>\n",
       "      <td>NaN</td>\n",
       "      <td>NaN</td>\n",
       "      <td>NaN</td>\n",
       "      <td>NaN</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>36720</th>\n",
       "      <td>NaN</td>\n",
       "      <td>NaN</td>\n",
       "      <td>NaN</td>\n",
       "      <td>NaN</td>\n",
       "      <td>NaN</td>\n",
       "      <td>NaN</td>\n",
       "      <td>NaN</td>\n",
       "      <td>NaN</td>\n",
       "      <td>NaN</td>\n",
       "      <td>NaN</td>\n",
       "      <td>NaN</td>\n",
       "      <td>NaN</td>\n",
       "      <td>NaN</td>\n",
       "      <td>NaN</td>\n",
       "      <td>NaN</td>\n",
       "      <td>NaN</td>\n",
       "      <td>NaN</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>36721</th>\n",
       "      <td>NaN</td>\n",
       "      <td>NaN</td>\n",
       "      <td>NaN</td>\n",
       "      <td>NaN</td>\n",
       "      <td>NaN</td>\n",
       "      <td>NaN</td>\n",
       "      <td>NaN</td>\n",
       "      <td>NaN</td>\n",
       "      <td>NaN</td>\n",
       "      <td>NaN</td>\n",
       "      <td>NaN</td>\n",
       "      <td>NaN</td>\n",
       "      <td>NaN</td>\n",
       "      <td>NaN</td>\n",
       "      <td>NaN</td>\n",
       "      <td>NaN</td>\n",
       "      <td>NaN</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>36722</th>\n",
       "      <td>NaN</td>\n",
       "      <td>NaN</td>\n",
       "      <td>NaN</td>\n",
       "      <td>NaN</td>\n",
       "      <td>NaN</td>\n",
       "      <td>NaN</td>\n",
       "      <td>NaN</td>\n",
       "      <td>NaN</td>\n",
       "      <td>NaN</td>\n",
       "      <td>NaN</td>\n",
       "      <td>NaN</td>\n",
       "      <td>NaN</td>\n",
       "      <td>NaN</td>\n",
       "      <td>NaN</td>\n",
       "      <td>NaN</td>\n",
       "      <td>NaN</td>\n",
       "      <td>NaN</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>36723</th>\n",
       "      <td>NaN</td>\n",
       "      <td>NaN</td>\n",
       "      <td>NaN</td>\n",
       "      <td>NaN</td>\n",
       "      <td>NaN</td>\n",
       "      <td>NaN</td>\n",
       "      <td>NaN</td>\n",
       "      <td>NaN</td>\n",
       "      <td>NaN</td>\n",
       "      <td>NaN</td>\n",
       "      <td>NaN</td>\n",
       "      <td>NaN</td>\n",
       "      <td>NaN</td>\n",
       "      <td>NaN</td>\n",
       "      <td>NaN</td>\n",
       "      <td>NaN</td>\n",
       "      <td>NaN</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>36724</th>\n",
       "      <td>NaN</td>\n",
       "      <td>NaN</td>\n",
       "      <td>NaN</td>\n",
       "      <td>NaN</td>\n",
       "      <td>NaN</td>\n",
       "      <td>NaN</td>\n",
       "      <td>NaN</td>\n",
       "      <td>NaN</td>\n",
       "      <td>NaN</td>\n",
       "      <td>NaN</td>\n",
       "      <td>NaN</td>\n",
       "      <td>NaN</td>\n",
       "      <td>NaN</td>\n",
       "      <td>NaN</td>\n",
       "      <td>NaN</td>\n",
       "      <td>NaN</td>\n",
       "      <td>NaN</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>36725</th>\n",
       "      <td>NaN</td>\n",
       "      <td>NaN</td>\n",
       "      <td>NaN</td>\n",
       "      <td>NaN</td>\n",
       "      <td>NaN</td>\n",
       "      <td>NaN</td>\n",
       "      <td>NaN</td>\n",
       "      <td>NaN</td>\n",
       "      <td>NaN</td>\n",
       "      <td>NaN</td>\n",
       "      <td>NaN</td>\n",
       "      <td>NaN</td>\n",
       "      <td>NaN</td>\n",
       "      <td>NaN</td>\n",
       "      <td>NaN</td>\n",
       "      <td>NaN</td>\n",
       "      <td>NaN</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>36726</th>\n",
       "      <td>NaN</td>\n",
       "      <td>NaN</td>\n",
       "      <td>NaN</td>\n",
       "      <td>NaN</td>\n",
       "      <td>NaN</td>\n",
       "      <td>NaN</td>\n",
       "      <td>NaN</td>\n",
       "      <td>NaN</td>\n",
       "      <td>NaN</td>\n",
       "      <td>NaN</td>\n",
       "      <td>NaN</td>\n",
       "      <td>NaN</td>\n",
       "      <td>NaN</td>\n",
       "      <td>NaN</td>\n",
       "      <td>NaN</td>\n",
       "      <td>NaN</td>\n",
       "      <td>NaN</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>36727</th>\n",
       "      <td>NaN</td>\n",
       "      <td>NaN</td>\n",
       "      <td>NaN</td>\n",
       "      <td>NaN</td>\n",
       "      <td>NaN</td>\n",
       "      <td>NaN</td>\n",
       "      <td>NaN</td>\n",
       "      <td>NaN</td>\n",
       "      <td>NaN</td>\n",
       "      <td>NaN</td>\n",
       "      <td>NaN</td>\n",
       "      <td>NaN</td>\n",
       "      <td>NaN</td>\n",
       "      <td>NaN</td>\n",
       "      <td>NaN</td>\n",
       "      <td>NaN</td>\n",
       "      <td>NaN</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>36728</th>\n",
       "      <td>NaN</td>\n",
       "      <td>NaN</td>\n",
       "      <td>NaN</td>\n",
       "      <td>NaN</td>\n",
       "      <td>NaN</td>\n",
       "      <td>NaN</td>\n",
       "      <td>NaN</td>\n",
       "      <td>NaN</td>\n",
       "      <td>NaN</td>\n",
       "      <td>NaN</td>\n",
       "      <td>NaN</td>\n",
       "      <td>NaN</td>\n",
       "      <td>NaN</td>\n",
       "      <td>NaN</td>\n",
       "      <td>NaN</td>\n",
       "      <td>NaN</td>\n",
       "      <td>NaN</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>36729</th>\n",
       "      <td>NaN</td>\n",
       "      <td>NaN</td>\n",
       "      <td>NaN</td>\n",
       "      <td>NaN</td>\n",
       "      <td>NaN</td>\n",
       "      <td>NaN</td>\n",
       "      <td>NaN</td>\n",
       "      <td>NaN</td>\n",
       "      <td>NaN</td>\n",
       "      <td>NaN</td>\n",
       "      <td>NaN</td>\n",
       "      <td>NaN</td>\n",
       "      <td>NaN</td>\n",
       "      <td>NaN</td>\n",
       "      <td>NaN</td>\n",
       "      <td>NaN</td>\n",
       "      <td>NaN</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>36730</th>\n",
       "      <td>NaN</td>\n",
       "      <td>NaN</td>\n",
       "      <td>NaN</td>\n",
       "      <td>NaN</td>\n",
       "      <td>NaN</td>\n",
       "      <td>NaN</td>\n",
       "      <td>NaN</td>\n",
       "      <td>NaN</td>\n",
       "      <td>NaN</td>\n",
       "      <td>NaN</td>\n",
       "      <td>NaN</td>\n",
       "      <td>NaN</td>\n",
       "      <td>NaN</td>\n",
       "      <td>NaN</td>\n",
       "      <td>NaN</td>\n",
       "      <td>NaN</td>\n",
       "      <td>NaN</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>36731</th>\n",
       "      <td>NaN</td>\n",
       "      <td>NaN</td>\n",
       "      <td>NaN</td>\n",
       "      <td>NaN</td>\n",
       "      <td>NaN</td>\n",
       "      <td>NaN</td>\n",
       "      <td>NaN</td>\n",
       "      <td>NaN</td>\n",
       "      <td>NaN</td>\n",
       "      <td>NaN</td>\n",
       "      <td>NaN</td>\n",
       "      <td>NaN</td>\n",
       "      <td>NaN</td>\n",
       "      <td>NaN</td>\n",
       "      <td>NaN</td>\n",
       "      <td>NaN</td>\n",
       "      <td>NaN</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>36732</th>\n",
       "      <td>NaN</td>\n",
       "      <td>NaN</td>\n",
       "      <td>NaN</td>\n",
       "      <td>NaN</td>\n",
       "      <td>NaN</td>\n",
       "      <td>NaN</td>\n",
       "      <td>NaN</td>\n",
       "      <td>NaN</td>\n",
       "      <td>NaN</td>\n",
       "      <td>NaN</td>\n",
       "      <td>NaN</td>\n",
       "      <td>NaN</td>\n",
       "      <td>NaN</td>\n",
       "      <td>NaN</td>\n",
       "      <td>NaN</td>\n",
       "      <td>NaN</td>\n",
       "      <td>NaN</td>\n",
       "    </tr>\n",
       "  </tbody>\n",
       "</table>\n",
       "<p>36733 rows × 17 columns</p>\n",
       "</div>"
      ],
      "text/plain": [
       "        key    energy  liveliness    tempo  speechiness  acousticness  \\\n",
       "0      11.0  0.912744    0.083704  132.069     0.293137      0.005423   \n",
       "1       6.0  0.745704    0.119955  100.008     0.046255      0.026230   \n",
       "2       5.0  0.709932    0.231455  130.030     0.121741      0.036662   \n",
       "3       3.0  0.705822    0.053292  126.009     0.126016      0.001966   \n",
       "4       3.0  0.741757    0.072774  129.985     0.051255      0.096732   \n",
       "5       8.0  0.733856    0.093043  174.952     0.058137      0.293325   \n",
       "6       8.0  0.777375    0.054104  104.946     0.029302      0.130350   \n",
       "7       7.0  0.585564    0.108297  120.014     0.038924      0.011707   \n",
       "8       0.0  0.418212    0.105322  129.054     0.045461      0.548385   \n",
       "9       4.0  0.814030    0.079196  124.991     0.072440      0.005355   \n",
       "10     11.0  0.903055    0.629349   96.082     0.074866      0.306908   \n",
       "11     11.0  0.689804    0.112703   92.032     0.042006      0.001365   \n",
       "12      8.0  0.869554    0.107945  124.065     0.051083      0.132712   \n",
       "13      0.0  0.878755    0.057114  119.000     0.040855      0.000579   \n",
       "14      5.0  0.908538    0.824374  127.914     0.063644      0.004802   \n",
       "15      5.0  0.838853    0.161256  109.007     0.043218      0.014582   \n",
       "16      1.0  0.625657    0.418317   80.082     0.034819      0.003484   \n",
       "17      9.0  0.323877    0.109264  135.116     0.027887      0.893351   \n",
       "18      3.0  0.640167    0.051898  127.963     0.049790      0.029364   \n",
       "19     11.0  0.821101    0.499412  127.016     0.218697      0.249091   \n",
       "20      0.0  0.616411    0.171423  130.009     0.059577      0.058936   \n",
       "21      8.0  0.932226    0.303175  129.025     0.164729      0.180590   \n",
       "22     10.0  0.839959    0.372695  157.625     0.101830      0.018690   \n",
       "23      NaN       NaN         NaN      NaN          NaN           NaN   \n",
       "24      2.0  0.777544    0.063793  100.023     0.073157      0.223289   \n",
       "25     11.0  0.398953    0.170640   77.821     0.033343      0.007855   \n",
       "26      2.0  0.555380    0.132021   95.000     0.050934      0.397652   \n",
       "27      5.0  0.785597    0.186539  175.041     0.147627      0.021278   \n",
       "28      7.0  0.877359    0.061576  125.013     0.081238      0.085122   \n",
       "29      7.0  0.736727    0.069281  125.928     0.044913      0.054524   \n",
       "...     ...       ...         ...      ...          ...           ...   \n",
       "36703   NaN       NaN         NaN      NaN          NaN           NaN   \n",
       "36704   NaN       NaN         NaN      NaN          NaN           NaN   \n",
       "36705   NaN       NaN         NaN      NaN          NaN           NaN   \n",
       "36706   NaN       NaN         NaN      NaN          NaN           NaN   \n",
       "36707   NaN       NaN         NaN      NaN          NaN           NaN   \n",
       "36708   NaN       NaN         NaN      NaN          NaN           NaN   \n",
       "36709   NaN       NaN         NaN      NaN          NaN           NaN   \n",
       "36710   NaN       NaN         NaN      NaN          NaN           NaN   \n",
       "36711   NaN       NaN         NaN      NaN          NaN           NaN   \n",
       "36712   NaN       NaN         NaN      NaN          NaN           NaN   \n",
       "36713   NaN       NaN         NaN      NaN          NaN           NaN   \n",
       "36714   NaN       NaN         NaN      NaN          NaN           NaN   \n",
       "36715   NaN       NaN         NaN      NaN          NaN           NaN   \n",
       "36716   NaN       NaN         NaN      NaN          NaN           NaN   \n",
       "36717   NaN       NaN         NaN      NaN          NaN           NaN   \n",
       "36718   NaN       NaN         NaN      NaN          NaN           NaN   \n",
       "36719   NaN       NaN         NaN      NaN          NaN           NaN   \n",
       "36720   NaN       NaN         NaN      NaN          NaN           NaN   \n",
       "36721   NaN       NaN         NaN      NaN          NaN           NaN   \n",
       "36722   NaN       NaN         NaN      NaN          NaN           NaN   \n",
       "36723   NaN       NaN         NaN      NaN          NaN           NaN   \n",
       "36724   NaN       NaN         NaN      NaN          NaN           NaN   \n",
       "36725   NaN       NaN         NaN      NaN          NaN           NaN   \n",
       "36726   NaN       NaN         NaN      NaN          NaN           NaN   \n",
       "36727   NaN       NaN         NaN      NaN          NaN           NaN   \n",
       "36728   NaN       NaN         NaN      NaN          NaN           NaN   \n",
       "36729   NaN       NaN         NaN      NaN          NaN           NaN   \n",
       "36730   NaN       NaN         NaN      NaN          NaN           NaN   \n",
       "36731   NaN       NaN         NaN      NaN          NaN           NaN   \n",
       "36732   NaN       NaN         NaN      NaN          NaN           NaN   \n",
       "\n",
       "       instrumentalness  time_signature  duration   loudness  valence  \\\n",
       "0              0.000001             0.0       4.0  218.30667   -3.890   \n",
       "1              0.012727             1.0       4.0  235.06086   -7.687   \n",
       "2              0.000000             0.0       4.0  232.46104   -5.150   \n",
       "3              0.000000             0.0       4.0  194.09333   -3.898   \n",
       "4              0.000474             0.0       4.0  285.42667   -5.860   \n",
       "5              0.000000             0.0       4.0  188.94281   -4.312   \n",
       "6              0.000000             1.0       4.0  228.29333   -5.112   \n",
       "7              0.000005             1.0       4.0  193.57333   -6.583   \n",
       "8              0.000120             0.0       4.0  248.28000   -7.284   \n",
       "9              0.000000             1.0       4.0  198.25333   -1.985   \n",
       "10             0.000000             1.0       4.0  167.98667   -6.240   \n",
       "11             0.000000             0.0       4.0  225.02667   -5.502   \n",
       "12             0.000000             1.0       4.0  227.97333   -3.739   \n",
       "13             0.000000             1.0       4.0  295.90667   -7.769   \n",
       "14             0.000000             1.0       4.0  231.57333   -2.106   \n",
       "15             0.000000             1.0       4.0  220.28000   -5.426   \n",
       "16             0.000000             0.0       4.0  230.25333   -7.264   \n",
       "17             0.000000             1.0       4.0  285.26667   -8.237   \n",
       "18             0.008626             0.0       4.0  214.14735   -7.490   \n",
       "19             0.000004             1.0       4.0  200.45333   -5.589   \n",
       "20             0.000065             1.0       4.0  284.41333   -7.443   \n",
       "21             0.000002             1.0       4.0  256.28000   -3.257   \n",
       "22             0.000000             0.0       4.0  240.04000  -10.859   \n",
       "23                  NaN             NaN       NaN        NaN      NaN   \n",
       "24             0.000000             1.0       4.0  263.02667   -3.070   \n",
       "25             0.457147             1.0       4.0  548.03156  -19.753   \n",
       "26             0.000278             1.0       4.0  242.73333   -7.275   \n",
       "27             0.000001             1.0       4.0  188.50667   -4.841   \n",
       "28             0.001956             1.0       4.0  224.69333   -3.813   \n",
       "29             0.000014             1.0       4.0  234.62667   -3.416   \n",
       "...                 ...             ...       ...        ...      ...   \n",
       "36703               NaN             NaN       NaN        NaN      NaN   \n",
       "36704               NaN             NaN       NaN        NaN      NaN   \n",
       "36705               NaN             NaN       NaN        NaN      NaN   \n",
       "36706               NaN             NaN       NaN        NaN      NaN   \n",
       "36707               NaN             NaN       NaN        NaN      NaN   \n",
       "36708               NaN             NaN       NaN        NaN      NaN   \n",
       "36709               NaN             NaN       NaN        NaN      NaN   \n",
       "36710               NaN             NaN       NaN        NaN      NaN   \n",
       "36711               NaN             NaN       NaN        NaN      NaN   \n",
       "36712               NaN             NaN       NaN        NaN      NaN   \n",
       "36713               NaN             NaN       NaN        NaN      NaN   \n",
       "36714               NaN             NaN       NaN        NaN      NaN   \n",
       "36715               NaN             NaN       NaN        NaN      NaN   \n",
       "36716               NaN             NaN       NaN        NaN      NaN   \n",
       "36717               NaN             NaN       NaN        NaN      NaN   \n",
       "36718               NaN             NaN       NaN        NaN      NaN   \n",
       "36719               NaN             NaN       NaN        NaN      NaN   \n",
       "36720               NaN             NaN       NaN        NaN      NaN   \n",
       "36721               NaN             NaN       NaN        NaN      NaN   \n",
       "36722               NaN             NaN       NaN        NaN      NaN   \n",
       "36723               NaN             NaN       NaN        NaN      NaN   \n",
       "36724               NaN             NaN       NaN        NaN      NaN   \n",
       "36725               NaN             NaN       NaN        NaN      NaN   \n",
       "36726               NaN             NaN       NaN        NaN      NaN   \n",
       "36727               NaN             NaN       NaN        NaN      NaN   \n",
       "36728               NaN             NaN       NaN        NaN      NaN   \n",
       "36729               NaN             NaN       NaN        NaN      NaN   \n",
       "36730               NaN             NaN       NaN        NaN      NaN   \n",
       "36731               NaN             NaN       NaN        NaN      NaN   \n",
       "36732               NaN             NaN       NaN        NaN      NaN   \n",
       "\n",
       "       danceability      mode  time_signature_confidence  tempo_confidence  \\\n",
       "0          0.752186  0.726920                      0.552             0.541   \n",
       "1          0.351282  0.691817                      0.737             0.634   \n",
       "2          0.374390  0.704729                      0.565             0.565   \n",
       "3          0.592798  0.875137                      0.004             0.114   \n",
       "4          0.585630  0.730711                      0.271             0.324   \n",
       "5          0.953424  0.785102                      0.177             0.404   \n",
       "6          0.525632  0.729051                      0.817             0.672   \n",
       "7          0.622176  0.781822                      0.970             0.861   \n",
       "8          0.440790  0.841824                      0.011             0.303   \n",
       "9          0.838512  0.700860                      0.429             0.375   \n",
       "10         0.898559  0.634209                      0.502             0.518   \n",
       "11         0.422368  0.565887                      0.662             0.567   \n",
       "12         0.639499  0.636239                      0.239             0.416   \n",
       "13         0.726486  0.634661                      0.391             0.334   \n",
       "14         0.273037  0.611672                      0.135             0.218   \n",
       "15         0.456284  0.636132                      0.459             0.454   \n",
       "16         0.486710  0.585064                      0.440             0.613   \n",
       "17         0.297851  0.561547                      0.358             0.526   \n",
       "18         0.684089  0.846949                      0.000             0.245   \n",
       "19         0.663748  0.717298                      0.648             0.514   \n",
       "20         0.476111  0.789760                      0.499             0.489   \n",
       "21         0.529215  0.673568                      0.294             0.401   \n",
       "22         0.854567  0.477058                      0.000             0.116   \n",
       "23              NaN       NaN                        NaN               NaN   \n",
       "24         0.591259  0.691084                      0.474             0.518   \n",
       "25         0.243535  0.198917                      0.901             0.676   \n",
       "26         0.454030  0.779140                      0.565             0.640   \n",
       "27         0.639066  0.634571                      0.220             0.303   \n",
       "28         0.924909  0.804825                      0.430             0.536   \n",
       "29         0.687324  0.763795                      0.490             0.708   \n",
       "...             ...       ...                        ...               ...   \n",
       "36703           NaN       NaN                        NaN               NaN   \n",
       "36704           NaN       NaN                        NaN               NaN   \n",
       "36705           NaN       NaN                        NaN               NaN   \n",
       "36706           NaN       NaN                        NaN               NaN   \n",
       "36707           NaN       NaN                        NaN               NaN   \n",
       "36708           NaN       NaN                        NaN               NaN   \n",
       "36709           NaN       NaN                        NaN               NaN   \n",
       "36710           NaN       NaN                        NaN               NaN   \n",
       "36711           NaN       NaN                        NaN               NaN   \n",
       "36712           NaN       NaN                        NaN               NaN   \n",
       "36713           NaN       NaN                        NaN               NaN   \n",
       "36714           NaN       NaN                        NaN               NaN   \n",
       "36715           NaN       NaN                        NaN               NaN   \n",
       "36716           NaN       NaN                        NaN               NaN   \n",
       "36717           NaN       NaN                        NaN               NaN   \n",
       "36718           NaN       NaN                        NaN               NaN   \n",
       "36719           NaN       NaN                        NaN               NaN   \n",
       "36720           NaN       NaN                        NaN               NaN   \n",
       "36721           NaN       NaN                        NaN               NaN   \n",
       "36722           NaN       NaN                        NaN               NaN   \n",
       "36723           NaN       NaN                        NaN               NaN   \n",
       "36724           NaN       NaN                        NaN               NaN   \n",
       "36725           NaN       NaN                        NaN               NaN   \n",
       "36726           NaN       NaN                        NaN               NaN   \n",
       "36727           NaN       NaN                        NaN               NaN   \n",
       "36728           NaN       NaN                        NaN               NaN   \n",
       "36729           NaN       NaN                        NaN               NaN   \n",
       "36730           NaN       NaN                        NaN               NaN   \n",
       "36731           NaN       NaN                        NaN               NaN   \n",
       "36732           NaN       NaN                        NaN               NaN   \n",
       "\n",
       "       key_confidence  mode_confidence  \n",
       "0               1.000            1.000  \n",
       "1               0.796            1.000  \n",
       "2               0.743            1.000  \n",
       "3               1.000            0.742  \n",
       "4               0.822            1.000  \n",
       "5               0.826            1.000  \n",
       "6               0.394            0.963  \n",
       "7               0.792            1.000  \n",
       "8               0.843            1.000  \n",
       "9               0.740            1.000  \n",
       "10              0.437            0.992  \n",
       "11              0.270            0.991  \n",
       "12              0.669            1.000  \n",
       "13              0.763            0.927  \n",
       "14              0.832            1.000  \n",
       "15              0.577            0.995  \n",
       "16              0.000            1.000  \n",
       "17              0.465            0.720  \n",
       "18              0.858            1.000  \n",
       "19              0.501            1.000  \n",
       "20              0.708            1.000  \n",
       "21              0.555            1.000  \n",
       "22              0.544            0.973  \n",
       "23                NaN              NaN  \n",
       "24              0.595            1.000  \n",
       "25              0.362            0.996  \n",
       "26              0.414            1.000  \n",
       "27              0.417            1.000  \n",
       "28              0.826            1.000  \n",
       "29              0.957            1.000  \n",
       "...               ...              ...  \n",
       "36703             NaN              NaN  \n",
       "36704             NaN              NaN  \n",
       "36705             NaN              NaN  \n",
       "36706             NaN              NaN  \n",
       "36707             NaN              NaN  \n",
       "36708             NaN              NaN  \n",
       "36709             NaN              NaN  \n",
       "36710             NaN              NaN  \n",
       "36711             NaN              NaN  \n",
       "36712             NaN              NaN  \n",
       "36713             NaN              NaN  \n",
       "36714             NaN              NaN  \n",
       "36715             NaN              NaN  \n",
       "36716             NaN              NaN  \n",
       "36717             NaN              NaN  \n",
       "36718             NaN              NaN  \n",
       "36719             NaN              NaN  \n",
       "36720             NaN              NaN  \n",
       "36721             NaN              NaN  \n",
       "36722             NaN              NaN  \n",
       "36723             NaN              NaN  \n",
       "36724             NaN              NaN  \n",
       "36725             NaN              NaN  \n",
       "36726             NaN              NaN  \n",
       "36727             NaN              NaN  \n",
       "36728             NaN              NaN  \n",
       "36729             NaN              NaN  \n",
       "36730             NaN              NaN  \n",
       "36731             NaN              NaN  \n",
       "36732             NaN              NaN  \n",
       "\n",
       "[36733 rows x 17 columns]"
      ]
     },
     "execution_count": 9,
     "metadata": {},
     "output_type": "execute_result"
    }
   ],
   "source": [
    "audio_features = audioFeatureFrame(song_dataset.loc[:, 'audio_features'], audioFeatures_names)\n",
    "audio_features"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 10,
   "metadata": {
    "scrolled": true
   },
   "outputs": [
    {
     "data": {
      "text/html": [
       "<div>\n",
       "<style>\n",
       "    .dataframe thead tr:only-child th {\n",
       "        text-align: right;\n",
       "    }\n",
       "\n",
       "    .dataframe thead th {\n",
       "        text-align: left;\n",
       "    }\n",
       "\n",
       "    .dataframe tbody tr th {\n",
       "        vertical-align: top;\n",
       "    }\n",
       "</style>\n",
       "<table border=\"1\" class=\"dataframe\">\n",
       "  <thead>\n",
       "    <tr style=\"text-align: right;\">\n",
       "      <th></th>\n",
       "      <th>aggressive</th>\n",
       "      <th>angsty</th>\n",
       "      <th>atmospheric</th>\n",
       "      <th>campy</th>\n",
       "      <th>celebratory</th>\n",
       "      <th>classy</th>\n",
       "      <th>cocky</th>\n",
       "      <th>cold</th>\n",
       "      <th>earthy</th>\n",
       "      <th>energetic</th>\n",
       "      <th>...</th>\n",
       "      <th>seductive</th>\n",
       "      <th>sexual</th>\n",
       "      <th>soothing</th>\n",
       "      <th>spacey</th>\n",
       "      <th>sprightly</th>\n",
       "      <th>sweet</th>\n",
       "      <th>trashy</th>\n",
       "      <th>trippy</th>\n",
       "      <th>visceral</th>\n",
       "      <th>warm</th>\n",
       "    </tr>\n",
       "  </thead>\n",
       "  <tbody>\n",
       "    <tr>\n",
       "      <th>0</th>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>1</td>\n",
       "      <td>...</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>1</th>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>...</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>2</th>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>1</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>...</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>3</th>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>1</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>1</td>\n",
       "      <td>...</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>4</th>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>1</td>\n",
       "      <td>...</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>5</th>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>...</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>1</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>6</th>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>...</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>1</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>7</th>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>1</td>\n",
       "      <td>...</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>8</th>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>...</td>\n",
       "      <td>1</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>9</th>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>1</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>...</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>10</th>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>...</td>\n",
       "      <td>1</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>11</th>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>1</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>1</td>\n",
       "      <td>...</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>12</th>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>1</td>\n",
       "      <td>1</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>...</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>13</th>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>1</td>\n",
       "      <td>...</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>14</th>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>1</td>\n",
       "      <td>...</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>15</th>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>1</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>...</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>16</th>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>...</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>17</th>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>...</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>1</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>18</th>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>1</td>\n",
       "      <td>...</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>19</th>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>1</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>...</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>20</th>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>1</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>1</td>\n",
       "      <td>...</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>21</th>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>1</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>1</td>\n",
       "      <td>...</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>22</th>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>...</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>1</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>23</th>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>1</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>...</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>24</th>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>1</td>\n",
       "      <td>1</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>...</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>25</th>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>...</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>1</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>26</th>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>...</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>27</th>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>1</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>...</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>28</th>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>1</td>\n",
       "      <td>...</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>29</th>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>1</td>\n",
       "      <td>1</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>...</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>...</th>\n",
       "      <td>...</td>\n",
       "      <td>...</td>\n",
       "      <td>...</td>\n",
       "      <td>...</td>\n",
       "      <td>...</td>\n",
       "      <td>...</td>\n",
       "      <td>...</td>\n",
       "      <td>...</td>\n",
       "      <td>...</td>\n",
       "      <td>...</td>\n",
       "      <td>...</td>\n",
       "      <td>...</td>\n",
       "      <td>...</td>\n",
       "      <td>...</td>\n",
       "      <td>...</td>\n",
       "      <td>...</td>\n",
       "      <td>...</td>\n",
       "      <td>...</td>\n",
       "      <td>...</td>\n",
       "      <td>...</td>\n",
       "      <td>...</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>36703</th>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>1</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>1</td>\n",
       "      <td>...</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>36704</th>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>1</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>...</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>1</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>36705</th>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>...</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>1</td>\n",
       "      <td>0</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>36706</th>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>...</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>36707</th>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>1</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>...</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>36708</th>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>1</td>\n",
       "      <td>0</td>\n",
       "      <td>...</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>1</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>36709</th>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>1</td>\n",
       "      <td>0</td>\n",
       "      <td>...</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>1</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>36710</th>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>...</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>36711</th>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>...</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>36712</th>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>...</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>36713</th>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>...</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>1</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>36714</th>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>1</td>\n",
       "      <td>0</td>\n",
       "      <td>1</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>...</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>36715</th>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>...</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>36716</th>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>...</td>\n",
       "      <td>1</td>\n",
       "      <td>1</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>36717</th>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>...</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>1</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>36718</th>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>1</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>...</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>1</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>36719</th>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>...</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>1</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>36720</th>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>1</td>\n",
       "      <td>...</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>1</td>\n",
       "      <td>0</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>36721</th>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>1</td>\n",
       "      <td>0</td>\n",
       "      <td>1</td>\n",
       "      <td>...</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>36722</th>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>1</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>...</td>\n",
       "      <td>1</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>36723</th>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>1</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>...</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>36724</th>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>...</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>1</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>36725</th>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>1</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>...</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>36726</th>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>1</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>1</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>...</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>36727</th>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>...</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>1</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>36728</th>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>1</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>...</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>36729</th>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>1</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>...</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>1</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>36730</th>\n",
       "      <td>1</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>...</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>36731</th>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>...</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>36732</th>\n",
       "      <td>1</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>...</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "    </tr>\n",
       "  </tbody>\n",
       "</table>\n",
       "<p>36733 rows × 32 columns</p>\n",
       "</div>"
      ],
      "text/plain": [
       "       aggressive  angsty  atmospheric  campy  celebratory  classy  cocky  \\\n",
       "0               0       0            0      0            0       0      0   \n",
       "1               0       0            0      0            0       0      0   \n",
       "2               0       0            0      0            1       0      0   \n",
       "3               0       0            0      0            1       0      0   \n",
       "4               0       0            0      0            0       0      0   \n",
       "5               0       0            0      0            0       0      0   \n",
       "6               0       0            0      0            0       0      0   \n",
       "7               0       0            0      0            0       0      0   \n",
       "8               0       0            0      0            0       0      0   \n",
       "9               0       0            0      0            1       0      0   \n",
       "10              0       0            0      0            0       0      0   \n",
       "11              0       0            0      0            1       0      0   \n",
       "12              0       0            0      1            1       0      0   \n",
       "13              0       0            0      0            0       0      0   \n",
       "14              0       0            0      0            0       0      0   \n",
       "15              0       0            0      0            1       0      0   \n",
       "16              0       0            0      0            0       0      0   \n",
       "17              0       0            0      0            0       0      0   \n",
       "18              0       0            0      0            0       0      0   \n",
       "19              0       0            0      0            1       0      0   \n",
       "20              0       0            0      0            1       0      0   \n",
       "21              0       0            0      0            1       0      0   \n",
       "22              0       0            0      0            0       0      0   \n",
       "23              0       0            0      0            1       0      0   \n",
       "24              0       0            0      1            1       0      0   \n",
       "25              0       0            0      0            0       0      0   \n",
       "26              0       0            0      0            0       0      0   \n",
       "27              0       0            0      0            1       0      0   \n",
       "28              0       0            0      0            0       0      0   \n",
       "29              0       0            0      1            1       0      0   \n",
       "...           ...     ...          ...    ...          ...     ...    ...   \n",
       "36703           0       0            0      0            1       0      0   \n",
       "36704           0       0            1      0            0       0      0   \n",
       "36705           0       0            0      0            0       0      0   \n",
       "36706           0       0            0      0            0       0      0   \n",
       "36707           0       0            0      0            0       1      0   \n",
       "36708           0       0            0      0            0       0      0   \n",
       "36709           0       0            0      0            0       0      0   \n",
       "36710           0       0            0      0            0       0      0   \n",
       "36711           0       0            0      0            0       0      0   \n",
       "36712           0       0            0      0            0       0      0   \n",
       "36713           0       0            0      0            0       0      0   \n",
       "36714           0       0            0      1            0       1      0   \n",
       "36715           0       0            0      0            0       0      0   \n",
       "36716           0       0            0      0            0       0      0   \n",
       "36717           0       0            0      0            0       0      0   \n",
       "36718           0       0            0      1            0       0      0   \n",
       "36719           0       0            0      0            0       0      0   \n",
       "36720           0       0            0      0            0       0      0   \n",
       "36721           0       0            0      0            0       0      0   \n",
       "36722           0       0            0      0            0       1      0   \n",
       "36723           0       0            0      0            0       0      1   \n",
       "36724           0       0            0      0            0       0      0   \n",
       "36725           0       0            0      0            0       0      1   \n",
       "36726           0       0            1      0            0       0      0   \n",
       "36727           0       0            0      0            0       0      0   \n",
       "36728           0       0            0      0            0       1      0   \n",
       "36729           0       0            1      0            0       0      0   \n",
       "36730           1       0            0      0            0       0      0   \n",
       "36731           0       0            0      0            0       0      0   \n",
       "36732           1       0            0      0            0       0      0   \n",
       "\n",
       "       cold  earthy  energetic  ...   seductive  sexual  soothing  spacey  \\\n",
       "0         0       0          1  ...           0       0         0       0   \n",
       "1         0       0          0  ...           0       0         0       0   \n",
       "2         0       0          0  ...           0       0         0       0   \n",
       "3         0       0          1  ...           0       0         0       0   \n",
       "4         0       0          1  ...           0       0         0       0   \n",
       "5         0       0          0  ...           0       0         0       0   \n",
       "6         0       0          0  ...           0       0         0       0   \n",
       "7         0       0          1  ...           0       0         0       0   \n",
       "8         0       0          0  ...           1       0         0       0   \n",
       "9         0       0          0  ...           0       0         0       0   \n",
       "10        0       0          0  ...           1       0         0       0   \n",
       "11        0       0          1  ...           0       0         0       0   \n",
       "12        0       0          0  ...           0       0         0       0   \n",
       "13        0       0          1  ...           0       0         0       0   \n",
       "14        0       0          1  ...           0       0         0       0   \n",
       "15        0       0          0  ...           0       0         0       0   \n",
       "16        0       0          0  ...           0       0         0       0   \n",
       "17        0       0          0  ...           0       0         1       0   \n",
       "18        0       0          1  ...           0       0         0       0   \n",
       "19        0       0          0  ...           0       0         0       0   \n",
       "20        0       0          1  ...           0       0         0       0   \n",
       "21        0       0          1  ...           0       0         0       0   \n",
       "22        0       0          0  ...           0       0         0       0   \n",
       "23        0       0          0  ...           0       0         0       0   \n",
       "24        0       0          0  ...           0       0         0       0   \n",
       "25        0       0          0  ...           0       0         0       0   \n",
       "26        0       0          0  ...           0       0         0       0   \n",
       "27        0       0          0  ...           0       0         0       0   \n",
       "28        0       0          1  ...           0       0         0       0   \n",
       "29        0       0          0  ...           0       0         0       0   \n",
       "...     ...     ...        ...  ...         ...     ...       ...     ...   \n",
       "36703     0       0          1  ...           0       0         0       0   \n",
       "36704     0       0          0  ...           0       0         0       1   \n",
       "36705     0       0          0  ...           0       0         0       0   \n",
       "36706     0       0          0  ...           0       0         0       0   \n",
       "36707     0       0          0  ...           0       0         0       0   \n",
       "36708     0       1          0  ...           0       0         0       0   \n",
       "36709     0       1          0  ...           0       0         0       0   \n",
       "36710     0       0          0  ...           0       0         0       0   \n",
       "36711     0       0          0  ...           0       0         0       0   \n",
       "36712     0       0          0  ...           0       0         0       0   \n",
       "36713     0       0          0  ...           0       0         0       0   \n",
       "36714     0       0          0  ...           0       0         0       0   \n",
       "36715     0       0          0  ...           0       0         0       0   \n",
       "36716     0       0          0  ...           1       1         0       0   \n",
       "36717     0       0          0  ...           0       0         0       0   \n",
       "36718     0       0          0  ...           0       0         0       0   \n",
       "36719     0       0          0  ...           0       0         1       0   \n",
       "36720     0       0          1  ...           0       0         0       0   \n",
       "36721     1       0          1  ...           0       0         0       0   \n",
       "36722     0       0          0  ...           1       0         0       0   \n",
       "36723     0       0          0  ...           0       0         0       0   \n",
       "36724     0       0          0  ...           0       0         0       0   \n",
       "36725     0       0          0  ...           0       0         0       0   \n",
       "36726     1       0          0  ...           0       0         0       0   \n",
       "36727     0       0          0  ...           0       0         0       0   \n",
       "36728     0       0          0  ...           0       0         0       0   \n",
       "36729     0       0          0  ...           0       0         0       1   \n",
       "36730     0       0          0  ...           0       0         0       0   \n",
       "36731     0       0          0  ...           0       0         0       0   \n",
       "36732     0       0          0  ...           0       0         0       0   \n",
       "\n",
       "       sprightly  sweet  trashy  trippy  visceral  warm  \n",
       "0              0      0       0       0         0     0  \n",
       "1              0      0       0       0         0     0  \n",
       "2              0      0       0       0         0     0  \n",
       "3              0      0       0       0         0     0  \n",
       "4              0      0       0       0         0     0  \n",
       "5              1      0       0       0         0     0  \n",
       "6              0      0       0       0         0     1  \n",
       "7              0      0       0       0         0     0  \n",
       "8              0      0       0       0         0     0  \n",
       "9              0      0       0       0         0     0  \n",
       "10             0      0       0       0         0     0  \n",
       "11             0      0       0       0         0     0  \n",
       "12             0      0       0       0         0     0  \n",
       "13             0      0       0       0         0     0  \n",
       "14             0      0       0       0         0     0  \n",
       "15             0      0       0       0         0     0  \n",
       "16             0      0       0       0         0     0  \n",
       "17             0      0       0       0         0     0  \n",
       "18             0      0       0       0         0     0  \n",
       "19             0      0       0       0         0     0  \n",
       "20             0      0       0       0         0     0  \n",
       "21             0      0       0       0         0     0  \n",
       "22             1      0       0       0         0     0  \n",
       "23             0      0       0       0         0     0  \n",
       "24             0      0       0       0         0     0  \n",
       "25             0      0       0       1         0     0  \n",
       "26             0      0       0       0         0     0  \n",
       "27             0      0       0       0         0     0  \n",
       "28             0      0       0       0         0     0  \n",
       "29             0      0       0       0         0     0  \n",
       "...          ...    ...     ...     ...       ...   ...  \n",
       "36703          0      0       0       0         0     0  \n",
       "36704          0      0       0       0         0     0  \n",
       "36705          0      0       0       0         1     0  \n",
       "36706          0      0       0       0         0     0  \n",
       "36707          0      0       0       0         0     0  \n",
       "36708          1      0       0       0         0     0  \n",
       "36709          1      0       0       0         0     0  \n",
       "36710          0      0       0       0         0     0  \n",
       "36711          0      0       0       0         0     0  \n",
       "36712          0      0       0       0         0     0  \n",
       "36713          1      0       0       0         0     0  \n",
       "36714          0      0       0       0         0     0  \n",
       "36715          0      0       0       0         0     0  \n",
       "36716          0      0       0       0         0     0  \n",
       "36717          1      0       0       0         0     0  \n",
       "36718          0      1       0       0         0     0  \n",
       "36719          0      0       0       0         0     0  \n",
       "36720          0      0       0       0         1     0  \n",
       "36721          0      0       0       0         0     0  \n",
       "36722          0      0       0       0         0     0  \n",
       "36723          0      0       0       0         0     0  \n",
       "36724          0      0       0       1         0     0  \n",
       "36725          0      0       0       0         0     0  \n",
       "36726          0      0       0       0         0     0  \n",
       "36727          0      0       0       1         0     0  \n",
       "36728          0      0       0       0         0     0  \n",
       "36729          0      0       0       0         0     0  \n",
       "36730          0      0       0       0         0     0  \n",
       "36731          0      0       0       0         0     0  \n",
       "36732          0      0       0       0         0     0  \n",
       "\n",
       "[36733 rows x 32 columns]"
      ]
     },
     "execution_count": 10,
     "metadata": {},
     "output_type": "execute_result"
    }
   ],
   "source": [
    "mood_features = song_dataset.loc[:, 'moods'].str.get_dummies(sep=',')\n",
    "mood_features"
   ]
  },
  {
   "cell_type": "markdown",
   "metadata": {},
   "source": [
    "-------------------\n",
    "## Assignment 1"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 11,
   "metadata": {},
   "outputs": [
    {
     "data": {
      "text/html": [
       "<div>\n",
       "<style>\n",
       "    .dataframe thead tr:only-child th {\n",
       "        text-align: right;\n",
       "    }\n",
       "\n",
       "    .dataframe thead th {\n",
       "        text-align: left;\n",
       "    }\n",
       "\n",
       "    .dataframe tbody tr th {\n",
       "        vertical-align: top;\n",
       "    }\n",
       "</style>\n",
       "<table border=\"1\" class=\"dataframe\">\n",
       "  <thead>\n",
       "    <tr style=\"text-align: right;\">\n",
       "      <th></th>\n",
       "      <th>key</th>\n",
       "      <th>energy</th>\n",
       "      <th>liveliness</th>\n",
       "      <th>tempo</th>\n",
       "      <th>speechiness</th>\n",
       "      <th>acousticness</th>\n",
       "      <th>instrumentalness</th>\n",
       "      <th>time_signature</th>\n",
       "      <th>duration</th>\n",
       "      <th>loudness</th>\n",
       "      <th>...</th>\n",
       "      <th>sexual</th>\n",
       "      <th>soothing</th>\n",
       "      <th>spacey</th>\n",
       "      <th>sprightly</th>\n",
       "      <th>sweet</th>\n",
       "      <th>trashy</th>\n",
       "      <th>trippy</th>\n",
       "      <th>visceral</th>\n",
       "      <th>warm</th>\n",
       "      <th>genres</th>\n",
       "    </tr>\n",
       "  </thead>\n",
       "  <tbody>\n",
       "    <tr>\n",
       "      <th>0</th>\n",
       "      <td>11.0</td>\n",
       "      <td>0.912744</td>\n",
       "      <td>0.083704</td>\n",
       "      <td>132.069</td>\n",
       "      <td>0.293137</td>\n",
       "      <td>0.005423</td>\n",
       "      <td>0.000001</td>\n",
       "      <td>0.0</td>\n",
       "      <td>4.0</td>\n",
       "      <td>218.30667</td>\n",
       "      <td>...</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>pop</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>1</th>\n",
       "      <td>6.0</td>\n",
       "      <td>0.745704</td>\n",
       "      <td>0.119955</td>\n",
       "      <td>100.008</td>\n",
       "      <td>0.046255</td>\n",
       "      <td>0.026230</td>\n",
       "      <td>0.012727</td>\n",
       "      <td>1.0</td>\n",
       "      <td>4.0</td>\n",
       "      <td>235.06086</td>\n",
       "      <td>...</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>pop</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>2</th>\n",
       "      <td>5.0</td>\n",
       "      <td>0.709932</td>\n",
       "      <td>0.231455</td>\n",
       "      <td>130.030</td>\n",
       "      <td>0.121741</td>\n",
       "      <td>0.036662</td>\n",
       "      <td>0.000000</td>\n",
       "      <td>0.0</td>\n",
       "      <td>4.0</td>\n",
       "      <td>232.46104</td>\n",
       "      <td>...</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>NaN</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>3</th>\n",
       "      <td>3.0</td>\n",
       "      <td>0.705822</td>\n",
       "      <td>0.053292</td>\n",
       "      <td>126.009</td>\n",
       "      <td>0.126016</td>\n",
       "      <td>0.001966</td>\n",
       "      <td>0.000000</td>\n",
       "      <td>0.0</td>\n",
       "      <td>4.0</td>\n",
       "      <td>194.09333</td>\n",
       "      <td>...</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>dance</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>4</th>\n",
       "      <td>3.0</td>\n",
       "      <td>0.741757</td>\n",
       "      <td>0.072774</td>\n",
       "      <td>129.985</td>\n",
       "      <td>0.051255</td>\n",
       "      <td>0.096732</td>\n",
       "      <td>0.000474</td>\n",
       "      <td>0.0</td>\n",
       "      <td>4.0</td>\n",
       "      <td>285.42667</td>\n",
       "      <td>...</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>reggaeton</td>\n",
       "    </tr>\n",
       "  </tbody>\n",
       "</table>\n",
       "<p>5 rows × 50 columns</p>\n",
       "</div>"
      ],
      "text/plain": [
       "    key    energy  liveliness    tempo  speechiness  acousticness  \\\n",
       "0  11.0  0.912744    0.083704  132.069     0.293137      0.005423   \n",
       "1   6.0  0.745704    0.119955  100.008     0.046255      0.026230   \n",
       "2   5.0  0.709932    0.231455  130.030     0.121741      0.036662   \n",
       "3   3.0  0.705822    0.053292  126.009     0.126016      0.001966   \n",
       "4   3.0  0.741757    0.072774  129.985     0.051255      0.096732   \n",
       "\n",
       "   instrumentalness  time_signature  duration   loudness    ...      sexual  \\\n",
       "0          0.000001             0.0       4.0  218.30667    ...           0   \n",
       "1          0.012727             1.0       4.0  235.06086    ...           0   \n",
       "2          0.000000             0.0       4.0  232.46104    ...           0   \n",
       "3          0.000000             0.0       4.0  194.09333    ...           0   \n",
       "4          0.000474             0.0       4.0  285.42667    ...           0   \n",
       "\n",
       "   soothing  spacey  sprightly  sweet  trashy  trippy  visceral  warm  \\\n",
       "0         0       0          0      0       0       0         0     0   \n",
       "1         0       0          0      0       0       0         0     0   \n",
       "2         0       0          0      0       0       0         0     0   \n",
       "3         0       0          0      0       0       0         0     0   \n",
       "4         0       0          0      0       0       0         0     0   \n",
       "\n",
       "      genres  \n",
       "0        pop  \n",
       "1        pop  \n",
       "2        NaN  \n",
       "3      dance  \n",
       "4  reggaeton  \n",
       "\n",
       "[5 rows x 50 columns]"
      ]
     },
     "execution_count": 11,
     "metadata": {},
     "output_type": "execute_result"
    }
   ],
   "source": [
    "assignment_1_dataset = pd.concat([audio_features, mood_features, song_dataset.loc[:,'genres']], axis = 1)\n",
    "assignment_1_dataset.head()"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 12,
   "metadata": {
    "collapsed": true
   },
   "outputs": [],
   "source": [
    "assignment_1_dataset.to_csv(\"week1/assignment-1_dataset.csv\", header = True, index = False)"
   ]
  },
  {
   "cell_type": "markdown",
   "metadata": {
    "scrolled": true
   },
   "source": [
    "--------------------\n",
    "## Assignment 2"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 13,
   "metadata": {
    "collapsed": true
   },
   "outputs": [],
   "source": [
    "genres_list = ['rock', 'jazz', 'dance', 'rap']\n",
    "assignment_2_dataset = pd.DataFrame()\n",
    "\n",
    "for genre in genres_list:\n",
    "    \n",
    "    row_filter = ((song_dataset['genres'] == genre) & (song_dataset['yt_views'] > 1000))\n",
    "    \n",
    "    audio_subset = audio_features.loc[row_filter, :]\n",
    "    mood_genre_subset = song_dataset.loc[row_filter, ['moods', 'genres']]\n",
    "    \n",
    "    combined_subset = pd.concat([audio_subset, mood_genre_subset], axis = 1)\n",
    "    combined_subset.dropna(axis = 0, how = 'any',inplace = True)\n",
    "    \n",
    "    balanced_subset = combined_subset.sample(n = 1500, axis = 0, random_state = 101)\n",
    "    \n",
    "    assignment_2_dataset = pd.concat([assignment_2_dataset, balanced_subset], axis = 0)\n",
    "    "
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 14,
   "metadata": {},
   "outputs": [
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "(6000, 19)\n"
     ]
    },
    {
     "data": {
      "text/html": [
       "<div>\n",
       "<style>\n",
       "    .dataframe thead tr:only-child th {\n",
       "        text-align: right;\n",
       "    }\n",
       "\n",
       "    .dataframe thead th {\n",
       "        text-align: left;\n",
       "    }\n",
       "\n",
       "    .dataframe tbody tr th {\n",
       "        vertical-align: top;\n",
       "    }\n",
       "</style>\n",
       "<table border=\"1\" class=\"dataframe\">\n",
       "  <thead>\n",
       "    <tr style=\"text-align: right;\">\n",
       "      <th></th>\n",
       "      <th>key</th>\n",
       "      <th>energy</th>\n",
       "      <th>liveliness</th>\n",
       "      <th>tempo</th>\n",
       "      <th>speechiness</th>\n",
       "      <th>acousticness</th>\n",
       "      <th>instrumentalness</th>\n",
       "      <th>time_signature</th>\n",
       "      <th>duration</th>\n",
       "      <th>loudness</th>\n",
       "      <th>valence</th>\n",
       "      <th>danceability</th>\n",
       "      <th>mode</th>\n",
       "      <th>time_signature_confidence</th>\n",
       "      <th>tempo_confidence</th>\n",
       "      <th>key_confidence</th>\n",
       "      <th>mode_confidence</th>\n",
       "      <th>moods</th>\n",
       "      <th>genres</th>\n",
       "    </tr>\n",
       "  </thead>\n",
       "  <tbody>\n",
       "    <tr>\n",
       "      <th>25521</th>\n",
       "      <td>7.0</td>\n",
       "      <td>0.373302</td>\n",
       "      <td>0.065919</td>\n",
       "      <td>137.411</td>\n",
       "      <td>0.097201</td>\n",
       "      <td>0.000399</td>\n",
       "      <td>0.750563</td>\n",
       "      <td>1.0</td>\n",
       "      <td>4.0</td>\n",
       "      <td>348.44000</td>\n",
       "      <td>-17.928</td>\n",
       "      <td>0.071321</td>\n",
       "      <td>0.498209</td>\n",
       "      <td>1.000</td>\n",
       "      <td>0.915</td>\n",
       "      <td>0.245</td>\n",
       "      <td>0.798</td>\n",
       "      <td>cold,atmospheric</td>\n",
       "      <td>rock</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>23186</th>\n",
       "      <td>0.0</td>\n",
       "      <td>0.830998</td>\n",
       "      <td>0.133293</td>\n",
       "      <td>173.624</td>\n",
       "      <td>0.232938</td>\n",
       "      <td>0.121033</td>\n",
       "      <td>0.833052</td>\n",
       "      <td>1.0</td>\n",
       "      <td>4.0</td>\n",
       "      <td>424.99973</td>\n",
       "      <td>-11.181</td>\n",
       "      <td>0.308000</td>\n",
       "      <td>0.166843</td>\n",
       "      <td>0.344</td>\n",
       "      <td>0.496</td>\n",
       "      <td>0.315</td>\n",
       "      <td>0.952</td>\n",
       "      <td>trippy</td>\n",
       "      <td>rock</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>15021</th>\n",
       "      <td>0.0</td>\n",
       "      <td>0.672115</td>\n",
       "      <td>0.164654</td>\n",
       "      <td>133.887</td>\n",
       "      <td>0.037931</td>\n",
       "      <td>0.065867</td>\n",
       "      <td>0.000024</td>\n",
       "      <td>1.0</td>\n",
       "      <td>4.0</td>\n",
       "      <td>153.22667</td>\n",
       "      <td>-7.391</td>\n",
       "      <td>0.515320</td>\n",
       "      <td>0.321991</td>\n",
       "      <td>0.610</td>\n",
       "      <td>0.465</td>\n",
       "      <td>0.586</td>\n",
       "      <td>0.466</td>\n",
       "      <td>cocky,trashy,rowdy</td>\n",
       "      <td>rock</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>20811</th>\n",
       "      <td>0.0</td>\n",
       "      <td>0.312316</td>\n",
       "      <td>0.144085</td>\n",
       "      <td>68.877</td>\n",
       "      <td>0.045299</td>\n",
       "      <td>0.980134</td>\n",
       "      <td>0.492329</td>\n",
       "      <td>1.0</td>\n",
       "      <td>3.0</td>\n",
       "      <td>223.46667</td>\n",
       "      <td>-10.982</td>\n",
       "      <td>0.262868</td>\n",
       "      <td>0.331355</td>\n",
       "      <td>0.605</td>\n",
       "      <td>0.698</td>\n",
       "      <td>0.003</td>\n",
       "      <td>0.743</td>\n",
       "      <td>sweet</td>\n",
       "      <td>rock</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>31100</th>\n",
       "      <td>10.0</td>\n",
       "      <td>0.941656</td>\n",
       "      <td>0.594007</td>\n",
       "      <td>176.479</td>\n",
       "      <td>0.039777</td>\n",
       "      <td>0.946758</td>\n",
       "      <td>0.819777</td>\n",
       "      <td>1.0</td>\n",
       "      <td>4.0</td>\n",
       "      <td>117.24000</td>\n",
       "      <td>-9.285</td>\n",
       "      <td>0.899193</td>\n",
       "      <td>0.496625</td>\n",
       "      <td>0.007</td>\n",
       "      <td>0.258</td>\n",
       "      <td>0.462</td>\n",
       "      <td>1.000</td>\n",
       "      <td>raw</td>\n",
       "      <td>rock</td>\n",
       "    </tr>\n",
       "  </tbody>\n",
       "</table>\n",
       "</div>"
      ],
      "text/plain": [
       "        key    energy  liveliness    tempo  speechiness  acousticness  \\\n",
       "25521   7.0  0.373302    0.065919  137.411     0.097201      0.000399   \n",
       "23186   0.0  0.830998    0.133293  173.624     0.232938      0.121033   \n",
       "15021   0.0  0.672115    0.164654  133.887     0.037931      0.065867   \n",
       "20811   0.0  0.312316    0.144085   68.877     0.045299      0.980134   \n",
       "31100  10.0  0.941656    0.594007  176.479     0.039777      0.946758   \n",
       "\n",
       "       instrumentalness  time_signature  duration   loudness  valence  \\\n",
       "25521          0.750563             1.0       4.0  348.44000  -17.928   \n",
       "23186          0.833052             1.0       4.0  424.99973  -11.181   \n",
       "15021          0.000024             1.0       4.0  153.22667   -7.391   \n",
       "20811          0.492329             1.0       3.0  223.46667  -10.982   \n",
       "31100          0.819777             1.0       4.0  117.24000   -9.285   \n",
       "\n",
       "       danceability      mode  time_signature_confidence  tempo_confidence  \\\n",
       "25521      0.071321  0.498209                      1.000             0.915   \n",
       "23186      0.308000  0.166843                      0.344             0.496   \n",
       "15021      0.515320  0.321991                      0.610             0.465   \n",
       "20811      0.262868  0.331355                      0.605             0.698   \n",
       "31100      0.899193  0.496625                      0.007             0.258   \n",
       "\n",
       "       key_confidence  mode_confidence               moods genres  \n",
       "25521           0.245            0.798    cold,atmospheric   rock  \n",
       "23186           0.315            0.952              trippy   rock  \n",
       "15021           0.586            0.466  cocky,trashy,rowdy   rock  \n",
       "20811           0.003            0.743               sweet   rock  \n",
       "31100           0.462            1.000                 raw   rock  "
      ]
     },
     "execution_count": 14,
     "metadata": {},
     "output_type": "execute_result"
    }
   ],
   "source": [
    "print(assignment_2_dataset.shape)\n",
    "\n",
    "assignment_2_dataset.head()"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 15,
   "metadata": {
    "scrolled": true
   },
   "outputs": [
    {
     "data": {
      "text/html": [
       "<div>\n",
       "<style>\n",
       "    .dataframe thead tr:only-child th {\n",
       "        text-align: right;\n",
       "    }\n",
       "\n",
       "    .dataframe thead th {\n",
       "        text-align: left;\n",
       "    }\n",
       "\n",
       "    .dataframe tbody tr th {\n",
       "        vertical-align: top;\n",
       "    }\n",
       "</style>\n",
       "<table border=\"1\" class=\"dataframe\">\n",
       "  <thead>\n",
       "    <tr style=\"text-align: right;\">\n",
       "      <th></th>\n",
       "      <th>key</th>\n",
       "      <th>energy</th>\n",
       "      <th>liveliness</th>\n",
       "      <th>tempo</th>\n",
       "      <th>speechiness</th>\n",
       "      <th>acousticness</th>\n",
       "      <th>instrumentalness</th>\n",
       "      <th>time_signature</th>\n",
       "      <th>duration</th>\n",
       "      <th>loudness</th>\n",
       "      <th>valence</th>\n",
       "      <th>danceability</th>\n",
       "      <th>mode</th>\n",
       "      <th>time_signature_confidence</th>\n",
       "      <th>tempo_confidence</th>\n",
       "      <th>key_confidence</th>\n",
       "      <th>mode_confidence</th>\n",
       "      <th>moods</th>\n",
       "      <th>genres</th>\n",
       "    </tr>\n",
       "  </thead>\n",
       "  <tbody>\n",
       "    <tr>\n",
       "      <th>0</th>\n",
       "      <td>8.0</td>\n",
       "      <td>0.831914</td>\n",
       "      <td>0.056928</td>\n",
       "      <td>106.245</td>\n",
       "      <td>0.232162</td>\n",
       "      <td>0.072076</td>\n",
       "      <td>0.000000</td>\n",
       "      <td>1.0</td>\n",
       "      <td>4.0</td>\n",
       "      <td>254.60000</td>\n",
       "      <td>-5.984</td>\n",
       "      <td>0.555338</td>\n",
       "      <td>0.837518</td>\n",
       "      <td>0.555</td>\n",
       "      <td>0.408</td>\n",
       "      <td>0.664</td>\n",
       "      <td>1.000</td>\n",
       "      <td>motivational</td>\n",
       "      <td>rap</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>1</th>\n",
       "      <td>1.0</td>\n",
       "      <td>0.434896</td>\n",
       "      <td>0.113409</td>\n",
       "      <td>105.276</td>\n",
       "      <td>0.071236</td>\n",
       "      <td>0.456392</td>\n",
       "      <td>0.322574</td>\n",
       "      <td>1.0</td>\n",
       "      <td>4.0</td>\n",
       "      <td>242.21333</td>\n",
       "      <td>-13.902</td>\n",
       "      <td>0.290687</td>\n",
       "      <td>0.459532</td>\n",
       "      <td>0.127</td>\n",
       "      <td>0.225</td>\n",
       "      <td>0.743</td>\n",
       "      <td>0.907</td>\n",
       "      <td>classy</td>\n",
       "      <td>jazz</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>2</th>\n",
       "      <td>8.0</td>\n",
       "      <td>0.326645</td>\n",
       "      <td>0.843938</td>\n",
       "      <td>123.949</td>\n",
       "      <td>0.492508</td>\n",
       "      <td>0.016599</td>\n",
       "      <td>0.000000</td>\n",
       "      <td>0.0</td>\n",
       "      <td>4.0</td>\n",
       "      <td>275.01714</td>\n",
       "      <td>-10.846</td>\n",
       "      <td>0.568852</td>\n",
       "      <td>0.934398</td>\n",
       "      <td>0.562</td>\n",
       "      <td>0.418</td>\n",
       "      <td>0.704</td>\n",
       "      <td>0.813</td>\n",
       "      <td>sexual</td>\n",
       "      <td>rap</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>3</th>\n",
       "      <td>7.0</td>\n",
       "      <td>0.873642</td>\n",
       "      <td>0.281888</td>\n",
       "      <td>97.940</td>\n",
       "      <td>0.041479</td>\n",
       "      <td>0.327251</td>\n",
       "      <td>0.264088</td>\n",
       "      <td>0.0</td>\n",
       "      <td>4.0</td>\n",
       "      <td>234.78857</td>\n",
       "      <td>-10.682</td>\n",
       "      <td>0.224093</td>\n",
       "      <td>0.600045</td>\n",
       "      <td>0.000</td>\n",
       "      <td>0.064</td>\n",
       "      <td>0.167</td>\n",
       "      <td>0.983</td>\n",
       "      <td>funky,mellow</td>\n",
       "      <td>rap</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>4</th>\n",
       "      <td>5.0</td>\n",
       "      <td>0.117357</td>\n",
       "      <td>0.101048</td>\n",
       "      <td>129.386</td>\n",
       "      <td>0.062007</td>\n",
       "      <td>0.847661</td>\n",
       "      <td>0.010635</td>\n",
       "      <td>0.0</td>\n",
       "      <td>4.0</td>\n",
       "      <td>128.88045</td>\n",
       "      <td>-15.296</td>\n",
       "      <td>0.488700</td>\n",
       "      <td>0.778220</td>\n",
       "      <td>0.369</td>\n",
       "      <td>0.529</td>\n",
       "      <td>0.248</td>\n",
       "      <td>1.000</td>\n",
       "      <td>sprightly</td>\n",
       "      <td>jazz</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>5</th>\n",
       "      <td>4.0</td>\n",
       "      <td>0.211063</td>\n",
       "      <td>0.149438</td>\n",
       "      <td>106.492</td>\n",
       "      <td>0.031841</td>\n",
       "      <td>0.589955</td>\n",
       "      <td>0.174510</td>\n",
       "      <td>0.0</td>\n",
       "      <td>4.0</td>\n",
       "      <td>196.49333</td>\n",
       "      <td>-18.276</td>\n",
       "      <td>0.198158</td>\n",
       "      <td>0.432224</td>\n",
       "      <td>0.421</td>\n",
       "      <td>0.496</td>\n",
       "      <td>0.579</td>\n",
       "      <td>0.763</td>\n",
       "      <td>classy,seductive,mellow,lush</td>\n",
       "      <td>jazz</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>6</th>\n",
       "      <td>2.0</td>\n",
       "      <td>0.799820</td>\n",
       "      <td>0.102666</td>\n",
       "      <td>190.142</td>\n",
       "      <td>0.053073</td>\n",
       "      <td>0.001846</td>\n",
       "      <td>0.001744</td>\n",
       "      <td>1.0</td>\n",
       "      <td>4.0</td>\n",
       "      <td>400.29333</td>\n",
       "      <td>-8.121</td>\n",
       "      <td>0.670951</td>\n",
       "      <td>0.481638</td>\n",
       "      <td>0.227</td>\n",
       "      <td>0.191</td>\n",
       "      <td>0.216</td>\n",
       "      <td>1.000</td>\n",
       "      <td>energetic,visceral</td>\n",
       "      <td>dance</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>7</th>\n",
       "      <td>10.0</td>\n",
       "      <td>0.770380</td>\n",
       "      <td>0.346163</td>\n",
       "      <td>118.058</td>\n",
       "      <td>0.142171</td>\n",
       "      <td>0.601119</td>\n",
       "      <td>0.000008</td>\n",
       "      <td>0.0</td>\n",
       "      <td>4.0</td>\n",
       "      <td>156.46667</td>\n",
       "      <td>-6.388</td>\n",
       "      <td>0.741407</td>\n",
       "      <td>0.881608</td>\n",
       "      <td>0.159</td>\n",
       "      <td>0.295</td>\n",
       "      <td>0.088</td>\n",
       "      <td>1.000</td>\n",
       "      <td>cocky,visceral</td>\n",
       "      <td>dance</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>8</th>\n",
       "      <td>1.0</td>\n",
       "      <td>0.859680</td>\n",
       "      <td>0.205867</td>\n",
       "      <td>126.032</td>\n",
       "      <td>0.034965</td>\n",
       "      <td>0.080888</td>\n",
       "      <td>0.867131</td>\n",
       "      <td>0.0</td>\n",
       "      <td>4.0</td>\n",
       "      <td>201.30399</td>\n",
       "      <td>-7.121</td>\n",
       "      <td>0.493057</td>\n",
       "      <td>0.603445</td>\n",
       "      <td>0.124</td>\n",
       "      <td>0.399</td>\n",
       "      <td>0.950</td>\n",
       "      <td>0.773</td>\n",
       "      <td>celebratory</td>\n",
       "      <td>dance</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>9</th>\n",
       "      <td>9.0</td>\n",
       "      <td>0.712624</td>\n",
       "      <td>0.105198</td>\n",
       "      <td>98.025</td>\n",
       "      <td>0.043380</td>\n",
       "      <td>0.002306</td>\n",
       "      <td>0.851865</td>\n",
       "      <td>0.0</td>\n",
       "      <td>4.0</td>\n",
       "      <td>461.33333</td>\n",
       "      <td>-7.881</td>\n",
       "      <td>0.278169</td>\n",
       "      <td>0.530937</td>\n",
       "      <td>0.336</td>\n",
       "      <td>0.355</td>\n",
       "      <td>0.034</td>\n",
       "      <td>1.000</td>\n",
       "      <td>hypnotic,spacey</td>\n",
       "      <td>dance</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>10</th>\n",
       "      <td>6.0</td>\n",
       "      <td>0.734221</td>\n",
       "      <td>0.107920</td>\n",
       "      <td>140.763</td>\n",
       "      <td>0.057727</td>\n",
       "      <td>0.060503</td>\n",
       "      <td>0.855355</td>\n",
       "      <td>1.0</td>\n",
       "      <td>4.0</td>\n",
       "      <td>485.98667</td>\n",
       "      <td>-10.135</td>\n",
       "      <td>0.403387</td>\n",
       "      <td>0.421281</td>\n",
       "      <td>0.000</td>\n",
       "      <td>0.133</td>\n",
       "      <td>0.354</td>\n",
       "      <td>0.566</td>\n",
       "      <td>hypnotic,funky,spacey</td>\n",
       "      <td>rap</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>11</th>\n",
       "      <td>9.0</td>\n",
       "      <td>0.529422</td>\n",
       "      <td>0.591604</td>\n",
       "      <td>147.949</td>\n",
       "      <td>0.094531</td>\n",
       "      <td>0.063366</td>\n",
       "      <td>0.108790</td>\n",
       "      <td>1.0</td>\n",
       "      <td>4.0</td>\n",
       "      <td>392.28299</td>\n",
       "      <td>-12.335</td>\n",
       "      <td>0.901001</td>\n",
       "      <td>0.741657</td>\n",
       "      <td>0.238</td>\n",
       "      <td>0.323</td>\n",
       "      <td>0.769</td>\n",
       "      <td>1.000</td>\n",
       "      <td>trippy,lush</td>\n",
       "      <td>jazz</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>12</th>\n",
       "      <td>5.0</td>\n",
       "      <td>0.473317</td>\n",
       "      <td>0.271108</td>\n",
       "      <td>79.887</td>\n",
       "      <td>0.090009</td>\n",
       "      <td>0.927287</td>\n",
       "      <td>0.862108</td>\n",
       "      <td>0.0</td>\n",
       "      <td>4.0</td>\n",
       "      <td>187.13333</td>\n",
       "      <td>-9.732</td>\n",
       "      <td>0.917200</td>\n",
       "      <td>0.658137</td>\n",
       "      <td>0.364</td>\n",
       "      <td>0.416</td>\n",
       "      <td>0.007</td>\n",
       "      <td>0.943</td>\n",
       "      <td>happy,sprightly</td>\n",
       "      <td>jazz</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>13</th>\n",
       "      <td>6.0</td>\n",
       "      <td>0.737119</td>\n",
       "      <td>0.164180</td>\n",
       "      <td>179.112</td>\n",
       "      <td>0.438058</td>\n",
       "      <td>0.277475</td>\n",
       "      <td>0.000000</td>\n",
       "      <td>0.0</td>\n",
       "      <td>4.0</td>\n",
       "      <td>365.33333</td>\n",
       "      <td>-5.471</td>\n",
       "      <td>0.535163</td>\n",
       "      <td>0.512100</td>\n",
       "      <td>0.701</td>\n",
       "      <td>0.483</td>\n",
       "      <td>0.064</td>\n",
       "      <td>1.000</td>\n",
       "      <td>introspective,funky</td>\n",
       "      <td>rap</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>14</th>\n",
       "      <td>2.0</td>\n",
       "      <td>0.116875</td>\n",
       "      <td>0.102165</td>\n",
       "      <td>131.585</td>\n",
       "      <td>0.076917</td>\n",
       "      <td>0.706004</td>\n",
       "      <td>0.000000</td>\n",
       "      <td>0.0</td>\n",
       "      <td>4.0</td>\n",
       "      <td>160.29333</td>\n",
       "      <td>-16.719</td>\n",
       "      <td>0.658808</td>\n",
       "      <td>0.664899</td>\n",
       "      <td>0.250</td>\n",
       "      <td>0.054</td>\n",
       "      <td>0.498</td>\n",
       "      <td>0.781</td>\n",
       "      <td>introspective,sad,lush</td>\n",
       "      <td>jazz</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>15</th>\n",
       "      <td>9.0</td>\n",
       "      <td>0.757685</td>\n",
       "      <td>0.872421</td>\n",
       "      <td>129.328</td>\n",
       "      <td>0.096607</td>\n",
       "      <td>0.048222</td>\n",
       "      <td>0.012087</td>\n",
       "      <td>1.0</td>\n",
       "      <td>4.0</td>\n",
       "      <td>316.50667</td>\n",
       "      <td>-14.947</td>\n",
       "      <td>0.353514</td>\n",
       "      <td>0.367584</td>\n",
       "      <td>0.373</td>\n",
       "      <td>0.411</td>\n",
       "      <td>0.581</td>\n",
       "      <td>1.000</td>\n",
       "      <td>trippy</td>\n",
       "      <td>rock</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>16</th>\n",
       "      <td>9.0</td>\n",
       "      <td>0.891744</td>\n",
       "      <td>0.333309</td>\n",
       "      <td>94.973</td>\n",
       "      <td>0.025462</td>\n",
       "      <td>0.002141</td>\n",
       "      <td>0.000000</td>\n",
       "      <td>0.0</td>\n",
       "      <td>4.0</td>\n",
       "      <td>202.69333</td>\n",
       "      <td>-4.452</td>\n",
       "      <td>0.652764</td>\n",
       "      <td>0.558887</td>\n",
       "      <td>0.325</td>\n",
       "      <td>0.442</td>\n",
       "      <td>0.134</td>\n",
       "      <td>1.000</td>\n",
       "      <td>mellow</td>\n",
       "      <td>rock</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>17</th>\n",
       "      <td>5.0</td>\n",
       "      <td>0.890643</td>\n",
       "      <td>0.084202</td>\n",
       "      <td>92.011</td>\n",
       "      <td>0.054498</td>\n",
       "      <td>0.000022</td>\n",
       "      <td>0.010847</td>\n",
       "      <td>1.0</td>\n",
       "      <td>4.0</td>\n",
       "      <td>221.38667</td>\n",
       "      <td>-2.557</td>\n",
       "      <td>0.694601</td>\n",
       "      <td>0.500986</td>\n",
       "      <td>0.713</td>\n",
       "      <td>0.596</td>\n",
       "      <td>0.335</td>\n",
       "      <td>1.000</td>\n",
       "      <td>aggressive</td>\n",
       "      <td>rock</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>18</th>\n",
       "      <td>2.0</td>\n",
       "      <td>0.960691</td>\n",
       "      <td>0.332691</td>\n",
       "      <td>155.592</td>\n",
       "      <td>0.092927</td>\n",
       "      <td>0.010061</td>\n",
       "      <td>0.000052</td>\n",
       "      <td>1.0</td>\n",
       "      <td>4.0</td>\n",
       "      <td>322.97333</td>\n",
       "      <td>-5.300</td>\n",
       "      <td>0.633453</td>\n",
       "      <td>0.482498</td>\n",
       "      <td>0.310</td>\n",
       "      <td>0.314</td>\n",
       "      <td>0.725</td>\n",
       "      <td>0.745</td>\n",
       "      <td>trippy</td>\n",
       "      <td>rock</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>19</th>\n",
       "      <td>0.0</td>\n",
       "      <td>0.485852</td>\n",
       "      <td>0.115319</td>\n",
       "      <td>103.792</td>\n",
       "      <td>0.055659</td>\n",
       "      <td>0.363042</td>\n",
       "      <td>0.000181</td>\n",
       "      <td>0.0</td>\n",
       "      <td>4.0</td>\n",
       "      <td>241.93333</td>\n",
       "      <td>-12.321</td>\n",
       "      <td>0.636242</td>\n",
       "      <td>0.834960</td>\n",
       "      <td>0.545</td>\n",
       "      <td>0.617</td>\n",
       "      <td>0.542</td>\n",
       "      <td>1.000</td>\n",
       "      <td>mellow</td>\n",
       "      <td>jazz</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>20</th>\n",
       "      <td>10.0</td>\n",
       "      <td>0.908297</td>\n",
       "      <td>0.795194</td>\n",
       "      <td>89.352</td>\n",
       "      <td>0.147076</td>\n",
       "      <td>0.535824</td>\n",
       "      <td>0.783615</td>\n",
       "      <td>0.0</td>\n",
       "      <td>4.0</td>\n",
       "      <td>266.18667</td>\n",
       "      <td>-8.773</td>\n",
       "      <td>0.357849</td>\n",
       "      <td>0.187867</td>\n",
       "      <td>0.515</td>\n",
       "      <td>0.604</td>\n",
       "      <td>0.156</td>\n",
       "      <td>0.722</td>\n",
       "      <td>aggressive,motivational</td>\n",
       "      <td>rock</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>21</th>\n",
       "      <td>0.0</td>\n",
       "      <td>0.448413</td>\n",
       "      <td>0.195886</td>\n",
       "      <td>131.991</td>\n",
       "      <td>0.083772</td>\n",
       "      <td>0.714895</td>\n",
       "      <td>0.273746</td>\n",
       "      <td>1.0</td>\n",
       "      <td>4.0</td>\n",
       "      <td>215.49333</td>\n",
       "      <td>-15.516</td>\n",
       "      <td>0.791940</td>\n",
       "      <td>0.606194</td>\n",
       "      <td>0.626</td>\n",
       "      <td>0.512</td>\n",
       "      <td>0.762</td>\n",
       "      <td>0.881</td>\n",
       "      <td>happy,sprightly</td>\n",
       "      <td>jazz</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>22</th>\n",
       "      <td>9.0</td>\n",
       "      <td>0.918359</td>\n",
       "      <td>0.055892</td>\n",
       "      <td>114.307</td>\n",
       "      <td>0.062521</td>\n",
       "      <td>0.000088</td>\n",
       "      <td>0.002768</td>\n",
       "      <td>1.0</td>\n",
       "      <td>4.0</td>\n",
       "      <td>169.37333</td>\n",
       "      <td>-7.510</td>\n",
       "      <td>0.771978</td>\n",
       "      <td>0.411895</td>\n",
       "      <td>0.709</td>\n",
       "      <td>0.454</td>\n",
       "      <td>0.737</td>\n",
       "      <td>0.761</td>\n",
       "      <td>energetic,angsty,rowdy</td>\n",
       "      <td>rock</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>23</th>\n",
       "      <td>6.0</td>\n",
       "      <td>0.824910</td>\n",
       "      <td>0.282756</td>\n",
       "      <td>103.298</td>\n",
       "      <td>0.311362</td>\n",
       "      <td>0.001565</td>\n",
       "      <td>0.000000</td>\n",
       "      <td>0.0</td>\n",
       "      <td>4.0</td>\n",
       "      <td>161.28000</td>\n",
       "      <td>-7.041</td>\n",
       "      <td>0.568895</td>\n",
       "      <td>0.796683</td>\n",
       "      <td>0.664</td>\n",
       "      <td>0.527</td>\n",
       "      <td>0.625</td>\n",
       "      <td>1.000</td>\n",
       "      <td>funky</td>\n",
       "      <td>rap</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>24</th>\n",
       "      <td>2.0</td>\n",
       "      <td>0.494105</td>\n",
       "      <td>0.093251</td>\n",
       "      <td>94.104</td>\n",
       "      <td>0.051101</td>\n",
       "      <td>0.849306</td>\n",
       "      <td>0.767443</td>\n",
       "      <td>0.0</td>\n",
       "      <td>3.0</td>\n",
       "      <td>238.21333</td>\n",
       "      <td>-13.023</td>\n",
       "      <td>0.780673</td>\n",
       "      <td>0.752874</td>\n",
       "      <td>0.435</td>\n",
       "      <td>0.431</td>\n",
       "      <td>0.185</td>\n",
       "      <td>0.943</td>\n",
       "      <td>classy</td>\n",
       "      <td>jazz</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>25</th>\n",
       "      <td>7.0</td>\n",
       "      <td>0.677335</td>\n",
       "      <td>0.121160</td>\n",
       "      <td>95.930</td>\n",
       "      <td>0.321129</td>\n",
       "      <td>0.150821</td>\n",
       "      <td>0.000000</td>\n",
       "      <td>1.0</td>\n",
       "      <td>4.0</td>\n",
       "      <td>242.13769</td>\n",
       "      <td>-4.311</td>\n",
       "      <td>0.558315</td>\n",
       "      <td>0.844792</td>\n",
       "      <td>0.315</td>\n",
       "      <td>0.562</td>\n",
       "      <td>0.496</td>\n",
       "      <td>1.000</td>\n",
       "      <td>cocky,visceral</td>\n",
       "      <td>dance</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>26</th>\n",
       "      <td>7.0</td>\n",
       "      <td>0.438382</td>\n",
       "      <td>0.344461</td>\n",
       "      <td>121.338</td>\n",
       "      <td>0.039502</td>\n",
       "      <td>0.058154</td>\n",
       "      <td>0.069805</td>\n",
       "      <td>1.0</td>\n",
       "      <td>4.0</td>\n",
       "      <td>282.80000</td>\n",
       "      <td>-11.765</td>\n",
       "      <td>0.143837</td>\n",
       "      <td>0.324083</td>\n",
       "      <td>0.641</td>\n",
       "      <td>0.644</td>\n",
       "      <td>0.126</td>\n",
       "      <td>0.979</td>\n",
       "      <td>rowdy</td>\n",
       "      <td>rock</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>27</th>\n",
       "      <td>1.0</td>\n",
       "      <td>0.503706</td>\n",
       "      <td>0.103116</td>\n",
       "      <td>89.759</td>\n",
       "      <td>0.284512</td>\n",
       "      <td>0.039740</td>\n",
       "      <td>0.000013</td>\n",
       "      <td>1.0</td>\n",
       "      <td>4.0</td>\n",
       "      <td>197.89333</td>\n",
       "      <td>-17.233</td>\n",
       "      <td>0.652091</td>\n",
       "      <td>0.835839</td>\n",
       "      <td>0.650</td>\n",
       "      <td>0.571</td>\n",
       "      <td>0.727</td>\n",
       "      <td>1.000</td>\n",
       "      <td>funky,mellow</td>\n",
       "      <td>rap</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>28</th>\n",
       "      <td>4.0</td>\n",
       "      <td>0.872735</td>\n",
       "      <td>0.277321</td>\n",
       "      <td>109.976</td>\n",
       "      <td>0.037969</td>\n",
       "      <td>0.002669</td>\n",
       "      <td>0.000202</td>\n",
       "      <td>0.0</td>\n",
       "      <td>4.0</td>\n",
       "      <td>304.89333</td>\n",
       "      <td>-6.165</td>\n",
       "      <td>0.656647</td>\n",
       "      <td>0.624058</td>\n",
       "      <td>0.283</td>\n",
       "      <td>0.238</td>\n",
       "      <td>0.934</td>\n",
       "      <td>0.970</td>\n",
       "      <td>visceral,trippy,spacey,rowdy</td>\n",
       "      <td>dance</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>29</th>\n",
       "      <td>3.0</td>\n",
       "      <td>0.339460</td>\n",
       "      <td>0.091913</td>\n",
       "      <td>120.018</td>\n",
       "      <td>0.057594</td>\n",
       "      <td>0.026154</td>\n",
       "      <td>0.089393</td>\n",
       "      <td>1.0</td>\n",
       "      <td>4.0</td>\n",
       "      <td>418.35093</td>\n",
       "      <td>-10.160</td>\n",
       "      <td>0.424637</td>\n",
       "      <td>0.922620</td>\n",
       "      <td>1.000</td>\n",
       "      <td>0.795</td>\n",
       "      <td>0.871</td>\n",
       "      <td>1.000</td>\n",
       "      <td>hypnotic,visceral</td>\n",
       "      <td>dance</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>...</th>\n",
       "      <td>...</td>\n",
       "      <td>...</td>\n",
       "      <td>...</td>\n",
       "      <td>...</td>\n",
       "      <td>...</td>\n",
       "      <td>...</td>\n",
       "      <td>...</td>\n",
       "      <td>...</td>\n",
       "      <td>...</td>\n",
       "      <td>...</td>\n",
       "      <td>...</td>\n",
       "      <td>...</td>\n",
       "      <td>...</td>\n",
       "      <td>...</td>\n",
       "      <td>...</td>\n",
       "      <td>...</td>\n",
       "      <td>...</td>\n",
       "      <td>...</td>\n",
       "      <td>...</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>5970</th>\n",
       "      <td>8.0</td>\n",
       "      <td>0.888280</td>\n",
       "      <td>0.309638</td>\n",
       "      <td>126.868</td>\n",
       "      <td>0.092419</td>\n",
       "      <td>0.005333</td>\n",
       "      <td>0.000005</td>\n",
       "      <td>0.0</td>\n",
       "      <td>4.0</td>\n",
       "      <td>160.93333</td>\n",
       "      <td>-4.282</td>\n",
       "      <td>0.235612</td>\n",
       "      <td>0.429494</td>\n",
       "      <td>0.417</td>\n",
       "      <td>0.400</td>\n",
       "      <td>0.752</td>\n",
       "      <td>0.907</td>\n",
       "      <td>celebratory</td>\n",
       "      <td>dance</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>5971</th>\n",
       "      <td>10.0</td>\n",
       "      <td>0.314457</td>\n",
       "      <td>0.582517</td>\n",
       "      <td>78.708</td>\n",
       "      <td>0.047730</td>\n",
       "      <td>0.851942</td>\n",
       "      <td>0.000000</td>\n",
       "      <td>0.0</td>\n",
       "      <td>4.0</td>\n",
       "      <td>290.54667</td>\n",
       "      <td>-15.347</td>\n",
       "      <td>0.481374</td>\n",
       "      <td>0.473534</td>\n",
       "      <td>0.560</td>\n",
       "      <td>0.455</td>\n",
       "      <td>0.124</td>\n",
       "      <td>0.500</td>\n",
       "      <td>introspective,sad,lush</td>\n",
       "      <td>jazz</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>5972</th>\n",
       "      <td>6.0</td>\n",
       "      <td>0.670298</td>\n",
       "      <td>0.057365</td>\n",
       "      <td>91.707</td>\n",
       "      <td>0.359289</td>\n",
       "      <td>0.019557</td>\n",
       "      <td>0.000000</td>\n",
       "      <td>1.0</td>\n",
       "      <td>4.0</td>\n",
       "      <td>261.33333</td>\n",
       "      <td>-6.609</td>\n",
       "      <td>0.715711</td>\n",
       "      <td>0.750012</td>\n",
       "      <td>0.681</td>\n",
       "      <td>0.185</td>\n",
       "      <td>0.407</td>\n",
       "      <td>1.000</td>\n",
       "      <td>funky</td>\n",
       "      <td>rap</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>5973</th>\n",
       "      <td>0.0</td>\n",
       "      <td>0.429999</td>\n",
       "      <td>0.134476</td>\n",
       "      <td>134.235</td>\n",
       "      <td>0.034864</td>\n",
       "      <td>0.926811</td>\n",
       "      <td>0.650718</td>\n",
       "      <td>1.0</td>\n",
       "      <td>4.0</td>\n",
       "      <td>456.50000</td>\n",
       "      <td>-8.238</td>\n",
       "      <td>0.831814</td>\n",
       "      <td>0.545987</td>\n",
       "      <td>0.562</td>\n",
       "      <td>0.473</td>\n",
       "      <td>0.642</td>\n",
       "      <td>0.510</td>\n",
       "      <td>seductive,sad,classy,nocturnal,mellow</td>\n",
       "      <td>jazz</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>5974</th>\n",
       "      <td>1.0</td>\n",
       "      <td>0.350249</td>\n",
       "      <td>0.110357</td>\n",
       "      <td>79.644</td>\n",
       "      <td>0.042099</td>\n",
       "      <td>0.386249</td>\n",
       "      <td>0.730701</td>\n",
       "      <td>1.0</td>\n",
       "      <td>4.0</td>\n",
       "      <td>384.45333</td>\n",
       "      <td>-14.439</td>\n",
       "      <td>0.536254</td>\n",
       "      <td>0.574547</td>\n",
       "      <td>0.611</td>\n",
       "      <td>0.754</td>\n",
       "      <td>0.295</td>\n",
       "      <td>0.942</td>\n",
       "      <td>mellow</td>\n",
       "      <td>jazz</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>5975</th>\n",
       "      <td>10.0</td>\n",
       "      <td>0.311163</td>\n",
       "      <td>0.168269</td>\n",
       "      <td>150.281</td>\n",
       "      <td>0.041410</td>\n",
       "      <td>0.847481</td>\n",
       "      <td>0.067075</td>\n",
       "      <td>0.0</td>\n",
       "      <td>4.0</td>\n",
       "      <td>263.66404</td>\n",
       "      <td>-16.382</td>\n",
       "      <td>0.463241</td>\n",
       "      <td>0.615914</td>\n",
       "      <td>0.125</td>\n",
       "      <td>0.101</td>\n",
       "      <td>0.305</td>\n",
       "      <td>0.894</td>\n",
       "      <td>classy</td>\n",
       "      <td>jazz</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>5976</th>\n",
       "      <td>6.0</td>\n",
       "      <td>0.610988</td>\n",
       "      <td>0.231351</td>\n",
       "      <td>119.987</td>\n",
       "      <td>0.080490</td>\n",
       "      <td>0.208197</td>\n",
       "      <td>0.879979</td>\n",
       "      <td>1.0</td>\n",
       "      <td>4.0</td>\n",
       "      <td>306.84122</td>\n",
       "      <td>-10.658</td>\n",
       "      <td>0.457199</td>\n",
       "      <td>0.784958</td>\n",
       "      <td>0.049</td>\n",
       "      <td>0.516</td>\n",
       "      <td>0.888</td>\n",
       "      <td>0.968</td>\n",
       "      <td>cold,nocturnal</td>\n",
       "      <td>dance</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>5977</th>\n",
       "      <td>2.0</td>\n",
       "      <td>0.135102</td>\n",
       "      <td>0.101982</td>\n",
       "      <td>73.077</td>\n",
       "      <td>0.033218</td>\n",
       "      <td>0.959095</td>\n",
       "      <td>0.795216</td>\n",
       "      <td>1.0</td>\n",
       "      <td>4.0</td>\n",
       "      <td>275.30100</td>\n",
       "      <td>-19.802</td>\n",
       "      <td>0.082727</td>\n",
       "      <td>0.324176</td>\n",
       "      <td>0.000</td>\n",
       "      <td>0.000</td>\n",
       "      <td>0.041</td>\n",
       "      <td>0.823</td>\n",
       "      <td>classy</td>\n",
       "      <td>jazz</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>5978</th>\n",
       "      <td>8.0</td>\n",
       "      <td>0.565956</td>\n",
       "      <td>0.908394</td>\n",
       "      <td>141.354</td>\n",
       "      <td>0.062197</td>\n",
       "      <td>0.280775</td>\n",
       "      <td>0.002979</td>\n",
       "      <td>1.0</td>\n",
       "      <td>4.0</td>\n",
       "      <td>464.05211</td>\n",
       "      <td>-13.460</td>\n",
       "      <td>0.600426</td>\n",
       "      <td>0.439259</td>\n",
       "      <td>0.289</td>\n",
       "      <td>0.378</td>\n",
       "      <td>0.338</td>\n",
       "      <td>0.943</td>\n",
       "      <td>sprightly</td>\n",
       "      <td>jazz</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>5979</th>\n",
       "      <td>4.0</td>\n",
       "      <td>0.624842</td>\n",
       "      <td>0.719335</td>\n",
       "      <td>120.037</td>\n",
       "      <td>0.070853</td>\n",
       "      <td>0.122363</td>\n",
       "      <td>0.000000</td>\n",
       "      <td>0.0</td>\n",
       "      <td>4.0</td>\n",
       "      <td>212.10667</td>\n",
       "      <td>-4.947</td>\n",
       "      <td>0.641498</td>\n",
       "      <td>0.733978</td>\n",
       "      <td>0.535</td>\n",
       "      <td>0.518</td>\n",
       "      <td>0.239</td>\n",
       "      <td>0.420</td>\n",
       "      <td>funky,mellow</td>\n",
       "      <td>rap</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>5980</th>\n",
       "      <td>10.0</td>\n",
       "      <td>0.393793</td>\n",
       "      <td>0.292529</td>\n",
       "      <td>90.333</td>\n",
       "      <td>0.794493</td>\n",
       "      <td>0.517595</td>\n",
       "      <td>0.000002</td>\n",
       "      <td>1.0</td>\n",
       "      <td>4.0</td>\n",
       "      <td>49.47696</td>\n",
       "      <td>-13.089</td>\n",
       "      <td>0.643892</td>\n",
       "      <td>0.762073</td>\n",
       "      <td>0.723</td>\n",
       "      <td>0.403</td>\n",
       "      <td>0.066</td>\n",
       "      <td>0.864</td>\n",
       "      <td>cocky</td>\n",
       "      <td>rap</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>5981</th>\n",
       "      <td>5.0</td>\n",
       "      <td>0.951738</td>\n",
       "      <td>0.333906</td>\n",
       "      <td>100.009</td>\n",
       "      <td>0.078591</td>\n",
       "      <td>0.001706</td>\n",
       "      <td>0.000013</td>\n",
       "      <td>0.0</td>\n",
       "      <td>4.0</td>\n",
       "      <td>215.26667</td>\n",
       "      <td>-6.123</td>\n",
       "      <td>0.388601</td>\n",
       "      <td>0.547297</td>\n",
       "      <td>0.606</td>\n",
       "      <td>0.644</td>\n",
       "      <td>0.285</td>\n",
       "      <td>0.829</td>\n",
       "      <td>aggressive</td>\n",
       "      <td>rock</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>5982</th>\n",
       "      <td>7.0</td>\n",
       "      <td>0.339446</td>\n",
       "      <td>0.200164</td>\n",
       "      <td>97.117</td>\n",
       "      <td>0.823806</td>\n",
       "      <td>0.023704</td>\n",
       "      <td>0.000000</td>\n",
       "      <td>1.0</td>\n",
       "      <td>4.0</td>\n",
       "      <td>391.72000</td>\n",
       "      <td>-15.391</td>\n",
       "      <td>0.494321</td>\n",
       "      <td>0.686925</td>\n",
       "      <td>0.672</td>\n",
       "      <td>0.470</td>\n",
       "      <td>0.228</td>\n",
       "      <td>0.982</td>\n",
       "      <td>motivational</td>\n",
       "      <td>rap</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>5983</th>\n",
       "      <td>9.0</td>\n",
       "      <td>0.627389</td>\n",
       "      <td>0.352402</td>\n",
       "      <td>120.865</td>\n",
       "      <td>0.029232</td>\n",
       "      <td>0.137717</td>\n",
       "      <td>0.001639</td>\n",
       "      <td>1.0</td>\n",
       "      <td>4.0</td>\n",
       "      <td>343.22667</td>\n",
       "      <td>-10.788</td>\n",
       "      <td>0.348939</td>\n",
       "      <td>0.606676</td>\n",
       "      <td>0.555</td>\n",
       "      <td>0.477</td>\n",
       "      <td>0.840</td>\n",
       "      <td>1.000</td>\n",
       "      <td>introspective,atmospheric,mellow,lush</td>\n",
       "      <td>rock</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>5984</th>\n",
       "      <td>1.0</td>\n",
       "      <td>0.946412</td>\n",
       "      <td>0.355363</td>\n",
       "      <td>127.992</td>\n",
       "      <td>0.068679</td>\n",
       "      <td>0.000750</td>\n",
       "      <td>0.077679</td>\n",
       "      <td>1.0</td>\n",
       "      <td>4.0</td>\n",
       "      <td>292.90667</td>\n",
       "      <td>-2.876</td>\n",
       "      <td>0.332467</td>\n",
       "      <td>0.615690</td>\n",
       "      <td>0.410</td>\n",
       "      <td>0.436</td>\n",
       "      <td>0.779</td>\n",
       "      <td>0.948</td>\n",
       "      <td>energetic,cocky,visceral,motivational</td>\n",
       "      <td>dance</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>5985</th>\n",
       "      <td>9.0</td>\n",
       "      <td>0.489169</td>\n",
       "      <td>0.111141</td>\n",
       "      <td>190.022</td>\n",
       "      <td>0.412567</td>\n",
       "      <td>0.048178</td>\n",
       "      <td>0.000112</td>\n",
       "      <td>1.0</td>\n",
       "      <td>4.0</td>\n",
       "      <td>396.65333</td>\n",
       "      <td>-10.992</td>\n",
       "      <td>0.173986</td>\n",
       "      <td>0.796232</td>\n",
       "      <td>0.971</td>\n",
       "      <td>0.652</td>\n",
       "      <td>0.322</td>\n",
       "      <td>1.000</td>\n",
       "      <td>cocky,motivational</td>\n",
       "      <td>rap</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>5986</th>\n",
       "      <td>1.0</td>\n",
       "      <td>0.580709</td>\n",
       "      <td>0.229500</td>\n",
       "      <td>127.114</td>\n",
       "      <td>0.105809</td>\n",
       "      <td>0.033882</td>\n",
       "      <td>0.000000</td>\n",
       "      <td>1.0</td>\n",
       "      <td>4.0</td>\n",
       "      <td>268.60000</td>\n",
       "      <td>-4.704</td>\n",
       "      <td>0.506986</td>\n",
       "      <td>0.972205</td>\n",
       "      <td>0.294</td>\n",
       "      <td>0.307</td>\n",
       "      <td>0.924</td>\n",
       "      <td>1.000</td>\n",
       "      <td>funky,mellow</td>\n",
       "      <td>rap</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>5987</th>\n",
       "      <td>8.0</td>\n",
       "      <td>0.730658</td>\n",
       "      <td>0.101535</td>\n",
       "      <td>122.929</td>\n",
       "      <td>0.070489</td>\n",
       "      <td>0.024364</td>\n",
       "      <td>0.836136</td>\n",
       "      <td>1.0</td>\n",
       "      <td>4.0</td>\n",
       "      <td>468.60000</td>\n",
       "      <td>-9.901</td>\n",
       "      <td>0.740001</td>\n",
       "      <td>0.886605</td>\n",
       "      <td>0.432</td>\n",
       "      <td>0.187</td>\n",
       "      <td>0.968</td>\n",
       "      <td>1.000</td>\n",
       "      <td>energetic,visceral</td>\n",
       "      <td>dance</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>5988</th>\n",
       "      <td>1.0</td>\n",
       "      <td>0.933709</td>\n",
       "      <td>0.752768</td>\n",
       "      <td>87.169</td>\n",
       "      <td>0.308546</td>\n",
       "      <td>0.088382</td>\n",
       "      <td>0.000371</td>\n",
       "      <td>0.0</td>\n",
       "      <td>4.0</td>\n",
       "      <td>298.86667</td>\n",
       "      <td>-1.862</td>\n",
       "      <td>0.509463</td>\n",
       "      <td>0.652241</td>\n",
       "      <td>0.125</td>\n",
       "      <td>0.303</td>\n",
       "      <td>0.203</td>\n",
       "      <td>1.000</td>\n",
       "      <td>cocky</td>\n",
       "      <td>rap</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>5989</th>\n",
       "      <td>7.0</td>\n",
       "      <td>0.806922</td>\n",
       "      <td>0.946593</td>\n",
       "      <td>87.985</td>\n",
       "      <td>0.093150</td>\n",
       "      <td>0.512215</td>\n",
       "      <td>0.000029</td>\n",
       "      <td>1.0</td>\n",
       "      <td>4.0</td>\n",
       "      <td>166.04000</td>\n",
       "      <td>-9.325</td>\n",
       "      <td>0.658990</td>\n",
       "      <td>0.684071</td>\n",
       "      <td>0.437</td>\n",
       "      <td>0.589</td>\n",
       "      <td>0.239</td>\n",
       "      <td>1.000</td>\n",
       "      <td>classy,seductive,mellow,lush</td>\n",
       "      <td>jazz</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>5990</th>\n",
       "      <td>2.0</td>\n",
       "      <td>0.780344</td>\n",
       "      <td>0.056770</td>\n",
       "      <td>98.495</td>\n",
       "      <td>0.302122</td>\n",
       "      <td>0.074188</td>\n",
       "      <td>0.000000</td>\n",
       "      <td>1.0</td>\n",
       "      <td>4.0</td>\n",
       "      <td>296.12363</td>\n",
       "      <td>-6.434</td>\n",
       "      <td>0.806311</td>\n",
       "      <td>0.903785</td>\n",
       "      <td>0.306</td>\n",
       "      <td>0.252</td>\n",
       "      <td>0.363</td>\n",
       "      <td>1.000</td>\n",
       "      <td>rowdy</td>\n",
       "      <td>rap</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>5991</th>\n",
       "      <td>3.0</td>\n",
       "      <td>0.784097</td>\n",
       "      <td>0.433829</td>\n",
       "      <td>93.994</td>\n",
       "      <td>0.288136</td>\n",
       "      <td>0.431360</td>\n",
       "      <td>0.000021</td>\n",
       "      <td>0.0</td>\n",
       "      <td>4.0</td>\n",
       "      <td>194.24000</td>\n",
       "      <td>-14.996</td>\n",
       "      <td>0.487005</td>\n",
       "      <td>0.343684</td>\n",
       "      <td>0.244</td>\n",
       "      <td>0.183</td>\n",
       "      <td>0.101</td>\n",
       "      <td>1.000</td>\n",
       "      <td>cocky,motivational</td>\n",
       "      <td>rap</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>5992</th>\n",
       "      <td>9.0</td>\n",
       "      <td>0.915449</td>\n",
       "      <td>0.262760</td>\n",
       "      <td>137.645</td>\n",
       "      <td>0.103910</td>\n",
       "      <td>0.361401</td>\n",
       "      <td>0.001002</td>\n",
       "      <td>1.0</td>\n",
       "      <td>4.0</td>\n",
       "      <td>291.00000</td>\n",
       "      <td>-3.338</td>\n",
       "      <td>0.520971</td>\n",
       "      <td>0.449663</td>\n",
       "      <td>0.683</td>\n",
       "      <td>0.771</td>\n",
       "      <td>0.742</td>\n",
       "      <td>1.000</td>\n",
       "      <td>aggressive,raw</td>\n",
       "      <td>rock</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>5993</th>\n",
       "      <td>11.0</td>\n",
       "      <td>0.813546</td>\n",
       "      <td>0.102499</td>\n",
       "      <td>113.212</td>\n",
       "      <td>0.037043</td>\n",
       "      <td>0.155448</td>\n",
       "      <td>0.000165</td>\n",
       "      <td>0.0</td>\n",
       "      <td>4.0</td>\n",
       "      <td>314.09333</td>\n",
       "      <td>-10.561</td>\n",
       "      <td>0.962792</td>\n",
       "      <td>0.733775</td>\n",
       "      <td>0.481</td>\n",
       "      <td>0.457</td>\n",
       "      <td>0.799</td>\n",
       "      <td>1.000</td>\n",
       "      <td>energetic</td>\n",
       "      <td>dance</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>5994</th>\n",
       "      <td>5.0</td>\n",
       "      <td>0.980513</td>\n",
       "      <td>0.236613</td>\n",
       "      <td>127.998</td>\n",
       "      <td>0.205352</td>\n",
       "      <td>0.003058</td>\n",
       "      <td>0.041531</td>\n",
       "      <td>1.0</td>\n",
       "      <td>4.0</td>\n",
       "      <td>314.06249</td>\n",
       "      <td>-5.089</td>\n",
       "      <td>0.058105</td>\n",
       "      <td>0.791039</td>\n",
       "      <td>0.170</td>\n",
       "      <td>0.109</td>\n",
       "      <td>0.830</td>\n",
       "      <td>0.989</td>\n",
       "      <td>happy,energetic,celebratory</td>\n",
       "      <td>dance</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>5995</th>\n",
       "      <td>9.0</td>\n",
       "      <td>0.575255</td>\n",
       "      <td>0.135169</td>\n",
       "      <td>70.521</td>\n",
       "      <td>0.032945</td>\n",
       "      <td>0.049234</td>\n",
       "      <td>0.000090</td>\n",
       "      <td>0.0</td>\n",
       "      <td>4.0</td>\n",
       "      <td>189.30667</td>\n",
       "      <td>-6.488</td>\n",
       "      <td>0.201068</td>\n",
       "      <td>0.584273</td>\n",
       "      <td>0.732</td>\n",
       "      <td>0.565</td>\n",
       "      <td>0.141</td>\n",
       "      <td>1.000</td>\n",
       "      <td>aggressive</td>\n",
       "      <td>rock</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>5996</th>\n",
       "      <td>10.0</td>\n",
       "      <td>0.573068</td>\n",
       "      <td>0.131173</td>\n",
       "      <td>90.016</td>\n",
       "      <td>0.130624</td>\n",
       "      <td>0.153372</td>\n",
       "      <td>0.006717</td>\n",
       "      <td>0.0</td>\n",
       "      <td>4.0</td>\n",
       "      <td>195.40000</td>\n",
       "      <td>-7.961</td>\n",
       "      <td>0.393880</td>\n",
       "      <td>0.812248</td>\n",
       "      <td>0.687</td>\n",
       "      <td>0.628</td>\n",
       "      <td>0.817</td>\n",
       "      <td>1.000</td>\n",
       "      <td>funky</td>\n",
       "      <td>rap</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>5997</th>\n",
       "      <td>0.0</td>\n",
       "      <td>0.287646</td>\n",
       "      <td>0.108176</td>\n",
       "      <td>196.934</td>\n",
       "      <td>0.031969</td>\n",
       "      <td>0.946525</td>\n",
       "      <td>0.084028</td>\n",
       "      <td>1.0</td>\n",
       "      <td>3.0</td>\n",
       "      <td>211.22667</td>\n",
       "      <td>-8.205</td>\n",
       "      <td>0.490947</td>\n",
       "      <td>0.416804</td>\n",
       "      <td>0.487</td>\n",
       "      <td>0.626</td>\n",
       "      <td>0.287</td>\n",
       "      <td>1.000</td>\n",
       "      <td>sprightly</td>\n",
       "      <td>rock</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>5998</th>\n",
       "      <td>4.0</td>\n",
       "      <td>0.644623</td>\n",
       "      <td>0.089304</td>\n",
       "      <td>107.565</td>\n",
       "      <td>0.031378</td>\n",
       "      <td>0.431863</td>\n",
       "      <td>0.210129</td>\n",
       "      <td>1.0</td>\n",
       "      <td>4.0</td>\n",
       "      <td>348.08000</td>\n",
       "      <td>-11.143</td>\n",
       "      <td>0.790709</td>\n",
       "      <td>0.427386</td>\n",
       "      <td>0.617</td>\n",
       "      <td>0.215</td>\n",
       "      <td>0.722</td>\n",
       "      <td>1.000</td>\n",
       "      <td>lush</td>\n",
       "      <td>jazz</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>5999</th>\n",
       "      <td>8.0</td>\n",
       "      <td>0.676216</td>\n",
       "      <td>0.547466</td>\n",
       "      <td>91.041</td>\n",
       "      <td>0.439897</td>\n",
       "      <td>0.027474</td>\n",
       "      <td>0.000000</td>\n",
       "      <td>0.0</td>\n",
       "      <td>4.0</td>\n",
       "      <td>288.24000</td>\n",
       "      <td>-5.193</td>\n",
       "      <td>0.553324</td>\n",
       "      <td>0.713402</td>\n",
       "      <td>0.554</td>\n",
       "      <td>0.351</td>\n",
       "      <td>0.376</td>\n",
       "      <td>0.833</td>\n",
       "      <td>cocky</td>\n",
       "      <td>rap</td>\n",
       "    </tr>\n",
       "  </tbody>\n",
       "</table>\n",
       "<p>6000 rows × 19 columns</p>\n",
       "</div>"
      ],
      "text/plain": [
       "       key    energy  liveliness    tempo  speechiness  acousticness  \\\n",
       "0      8.0  0.831914    0.056928  106.245     0.232162      0.072076   \n",
       "1      1.0  0.434896    0.113409  105.276     0.071236      0.456392   \n",
       "2      8.0  0.326645    0.843938  123.949     0.492508      0.016599   \n",
       "3      7.0  0.873642    0.281888   97.940     0.041479      0.327251   \n",
       "4      5.0  0.117357    0.101048  129.386     0.062007      0.847661   \n",
       "5      4.0  0.211063    0.149438  106.492     0.031841      0.589955   \n",
       "6      2.0  0.799820    0.102666  190.142     0.053073      0.001846   \n",
       "7     10.0  0.770380    0.346163  118.058     0.142171      0.601119   \n",
       "8      1.0  0.859680    0.205867  126.032     0.034965      0.080888   \n",
       "9      9.0  0.712624    0.105198   98.025     0.043380      0.002306   \n",
       "10     6.0  0.734221    0.107920  140.763     0.057727      0.060503   \n",
       "11     9.0  0.529422    0.591604  147.949     0.094531      0.063366   \n",
       "12     5.0  0.473317    0.271108   79.887     0.090009      0.927287   \n",
       "13     6.0  0.737119    0.164180  179.112     0.438058      0.277475   \n",
       "14     2.0  0.116875    0.102165  131.585     0.076917      0.706004   \n",
       "15     9.0  0.757685    0.872421  129.328     0.096607      0.048222   \n",
       "16     9.0  0.891744    0.333309   94.973     0.025462      0.002141   \n",
       "17     5.0  0.890643    0.084202   92.011     0.054498      0.000022   \n",
       "18     2.0  0.960691    0.332691  155.592     0.092927      0.010061   \n",
       "19     0.0  0.485852    0.115319  103.792     0.055659      0.363042   \n",
       "20    10.0  0.908297    0.795194   89.352     0.147076      0.535824   \n",
       "21     0.0  0.448413    0.195886  131.991     0.083772      0.714895   \n",
       "22     9.0  0.918359    0.055892  114.307     0.062521      0.000088   \n",
       "23     6.0  0.824910    0.282756  103.298     0.311362      0.001565   \n",
       "24     2.0  0.494105    0.093251   94.104     0.051101      0.849306   \n",
       "25     7.0  0.677335    0.121160   95.930     0.321129      0.150821   \n",
       "26     7.0  0.438382    0.344461  121.338     0.039502      0.058154   \n",
       "27     1.0  0.503706    0.103116   89.759     0.284512      0.039740   \n",
       "28     4.0  0.872735    0.277321  109.976     0.037969      0.002669   \n",
       "29     3.0  0.339460    0.091913  120.018     0.057594      0.026154   \n",
       "...    ...       ...         ...      ...          ...           ...   \n",
       "5970   8.0  0.888280    0.309638  126.868     0.092419      0.005333   \n",
       "5971  10.0  0.314457    0.582517   78.708     0.047730      0.851942   \n",
       "5972   6.0  0.670298    0.057365   91.707     0.359289      0.019557   \n",
       "5973   0.0  0.429999    0.134476  134.235     0.034864      0.926811   \n",
       "5974   1.0  0.350249    0.110357   79.644     0.042099      0.386249   \n",
       "5975  10.0  0.311163    0.168269  150.281     0.041410      0.847481   \n",
       "5976   6.0  0.610988    0.231351  119.987     0.080490      0.208197   \n",
       "5977   2.0  0.135102    0.101982   73.077     0.033218      0.959095   \n",
       "5978   8.0  0.565956    0.908394  141.354     0.062197      0.280775   \n",
       "5979   4.0  0.624842    0.719335  120.037     0.070853      0.122363   \n",
       "5980  10.0  0.393793    0.292529   90.333     0.794493      0.517595   \n",
       "5981   5.0  0.951738    0.333906  100.009     0.078591      0.001706   \n",
       "5982   7.0  0.339446    0.200164   97.117     0.823806      0.023704   \n",
       "5983   9.0  0.627389    0.352402  120.865     0.029232      0.137717   \n",
       "5984   1.0  0.946412    0.355363  127.992     0.068679      0.000750   \n",
       "5985   9.0  0.489169    0.111141  190.022     0.412567      0.048178   \n",
       "5986   1.0  0.580709    0.229500  127.114     0.105809      0.033882   \n",
       "5987   8.0  0.730658    0.101535  122.929     0.070489      0.024364   \n",
       "5988   1.0  0.933709    0.752768   87.169     0.308546      0.088382   \n",
       "5989   7.0  0.806922    0.946593   87.985     0.093150      0.512215   \n",
       "5990   2.0  0.780344    0.056770   98.495     0.302122      0.074188   \n",
       "5991   3.0  0.784097    0.433829   93.994     0.288136      0.431360   \n",
       "5992   9.0  0.915449    0.262760  137.645     0.103910      0.361401   \n",
       "5993  11.0  0.813546    0.102499  113.212     0.037043      0.155448   \n",
       "5994   5.0  0.980513    0.236613  127.998     0.205352      0.003058   \n",
       "5995   9.0  0.575255    0.135169   70.521     0.032945      0.049234   \n",
       "5996  10.0  0.573068    0.131173   90.016     0.130624      0.153372   \n",
       "5997   0.0  0.287646    0.108176  196.934     0.031969      0.946525   \n",
       "5998   4.0  0.644623    0.089304  107.565     0.031378      0.431863   \n",
       "5999   8.0  0.676216    0.547466   91.041     0.439897      0.027474   \n",
       "\n",
       "      instrumentalness  time_signature  duration   loudness  valence  \\\n",
       "0             0.000000             1.0       4.0  254.60000   -5.984   \n",
       "1             0.322574             1.0       4.0  242.21333  -13.902   \n",
       "2             0.000000             0.0       4.0  275.01714  -10.846   \n",
       "3             0.264088             0.0       4.0  234.78857  -10.682   \n",
       "4             0.010635             0.0       4.0  128.88045  -15.296   \n",
       "5             0.174510             0.0       4.0  196.49333  -18.276   \n",
       "6             0.001744             1.0       4.0  400.29333   -8.121   \n",
       "7             0.000008             0.0       4.0  156.46667   -6.388   \n",
       "8             0.867131             0.0       4.0  201.30399   -7.121   \n",
       "9             0.851865             0.0       4.0  461.33333   -7.881   \n",
       "10            0.855355             1.0       4.0  485.98667  -10.135   \n",
       "11            0.108790             1.0       4.0  392.28299  -12.335   \n",
       "12            0.862108             0.0       4.0  187.13333   -9.732   \n",
       "13            0.000000             0.0       4.0  365.33333   -5.471   \n",
       "14            0.000000             0.0       4.0  160.29333  -16.719   \n",
       "15            0.012087             1.0       4.0  316.50667  -14.947   \n",
       "16            0.000000             0.0       4.0  202.69333   -4.452   \n",
       "17            0.010847             1.0       4.0  221.38667   -2.557   \n",
       "18            0.000052             1.0       4.0  322.97333   -5.300   \n",
       "19            0.000181             0.0       4.0  241.93333  -12.321   \n",
       "20            0.783615             0.0       4.0  266.18667   -8.773   \n",
       "21            0.273746             1.0       4.0  215.49333  -15.516   \n",
       "22            0.002768             1.0       4.0  169.37333   -7.510   \n",
       "23            0.000000             0.0       4.0  161.28000   -7.041   \n",
       "24            0.767443             0.0       3.0  238.21333  -13.023   \n",
       "25            0.000000             1.0       4.0  242.13769   -4.311   \n",
       "26            0.069805             1.0       4.0  282.80000  -11.765   \n",
       "27            0.000013             1.0       4.0  197.89333  -17.233   \n",
       "28            0.000202             0.0       4.0  304.89333   -6.165   \n",
       "29            0.089393             1.0       4.0  418.35093  -10.160   \n",
       "...                ...             ...       ...        ...      ...   \n",
       "5970          0.000005             0.0       4.0  160.93333   -4.282   \n",
       "5971          0.000000             0.0       4.0  290.54667  -15.347   \n",
       "5972          0.000000             1.0       4.0  261.33333   -6.609   \n",
       "5973          0.650718             1.0       4.0  456.50000   -8.238   \n",
       "5974          0.730701             1.0       4.0  384.45333  -14.439   \n",
       "5975          0.067075             0.0       4.0  263.66404  -16.382   \n",
       "5976          0.879979             1.0       4.0  306.84122  -10.658   \n",
       "5977          0.795216             1.0       4.0  275.30100  -19.802   \n",
       "5978          0.002979             1.0       4.0  464.05211  -13.460   \n",
       "5979          0.000000             0.0       4.0  212.10667   -4.947   \n",
       "5980          0.000002             1.0       4.0   49.47696  -13.089   \n",
       "5981          0.000013             0.0       4.0  215.26667   -6.123   \n",
       "5982          0.000000             1.0       4.0  391.72000  -15.391   \n",
       "5983          0.001639             1.0       4.0  343.22667  -10.788   \n",
       "5984          0.077679             1.0       4.0  292.90667   -2.876   \n",
       "5985          0.000112             1.0       4.0  396.65333  -10.992   \n",
       "5986          0.000000             1.0       4.0  268.60000   -4.704   \n",
       "5987          0.836136             1.0       4.0  468.60000   -9.901   \n",
       "5988          0.000371             0.0       4.0  298.86667   -1.862   \n",
       "5989          0.000029             1.0       4.0  166.04000   -9.325   \n",
       "5990          0.000000             1.0       4.0  296.12363   -6.434   \n",
       "5991          0.000021             0.0       4.0  194.24000  -14.996   \n",
       "5992          0.001002             1.0       4.0  291.00000   -3.338   \n",
       "5993          0.000165             0.0       4.0  314.09333  -10.561   \n",
       "5994          0.041531             1.0       4.0  314.06249   -5.089   \n",
       "5995          0.000090             0.0       4.0  189.30667   -6.488   \n",
       "5996          0.006717             0.0       4.0  195.40000   -7.961   \n",
       "5997          0.084028             1.0       3.0  211.22667   -8.205   \n",
       "5998          0.210129             1.0       4.0  348.08000  -11.143   \n",
       "5999          0.000000             0.0       4.0  288.24000   -5.193   \n",
       "\n",
       "      danceability      mode  time_signature_confidence  tempo_confidence  \\\n",
       "0         0.555338  0.837518                      0.555             0.408   \n",
       "1         0.290687  0.459532                      0.127             0.225   \n",
       "2         0.568852  0.934398                      0.562             0.418   \n",
       "3         0.224093  0.600045                      0.000             0.064   \n",
       "4         0.488700  0.778220                      0.369             0.529   \n",
       "5         0.198158  0.432224                      0.421             0.496   \n",
       "6         0.670951  0.481638                      0.227             0.191   \n",
       "7         0.741407  0.881608                      0.159             0.295   \n",
       "8         0.493057  0.603445                      0.124             0.399   \n",
       "9         0.278169  0.530937                      0.336             0.355   \n",
       "10        0.403387  0.421281                      0.000             0.133   \n",
       "11        0.901001  0.741657                      0.238             0.323   \n",
       "12        0.917200  0.658137                      0.364             0.416   \n",
       "13        0.535163  0.512100                      0.701             0.483   \n",
       "14        0.658808  0.664899                      0.250             0.054   \n",
       "15        0.353514  0.367584                      0.373             0.411   \n",
       "16        0.652764  0.558887                      0.325             0.442   \n",
       "17        0.694601  0.500986                      0.713             0.596   \n",
       "18        0.633453  0.482498                      0.310             0.314   \n",
       "19        0.636242  0.834960                      0.545             0.617   \n",
       "20        0.357849  0.187867                      0.515             0.604   \n",
       "21        0.791940  0.606194                      0.626             0.512   \n",
       "22        0.771978  0.411895                      0.709             0.454   \n",
       "23        0.568895  0.796683                      0.664             0.527   \n",
       "24        0.780673  0.752874                      0.435             0.431   \n",
       "25        0.558315  0.844792                      0.315             0.562   \n",
       "26        0.143837  0.324083                      0.641             0.644   \n",
       "27        0.652091  0.835839                      0.650             0.571   \n",
       "28        0.656647  0.624058                      0.283             0.238   \n",
       "29        0.424637  0.922620                      1.000             0.795   \n",
       "...            ...       ...                        ...               ...   \n",
       "5970      0.235612  0.429494                      0.417             0.400   \n",
       "5971      0.481374  0.473534                      0.560             0.455   \n",
       "5972      0.715711  0.750012                      0.681             0.185   \n",
       "5973      0.831814  0.545987                      0.562             0.473   \n",
       "5974      0.536254  0.574547                      0.611             0.754   \n",
       "5975      0.463241  0.615914                      0.125             0.101   \n",
       "5976      0.457199  0.784958                      0.049             0.516   \n",
       "5977      0.082727  0.324176                      0.000             0.000   \n",
       "5978      0.600426  0.439259                      0.289             0.378   \n",
       "5979      0.641498  0.733978                      0.535             0.518   \n",
       "5980      0.643892  0.762073                      0.723             0.403   \n",
       "5981      0.388601  0.547297                      0.606             0.644   \n",
       "5982      0.494321  0.686925                      0.672             0.470   \n",
       "5983      0.348939  0.606676                      0.555             0.477   \n",
       "5984      0.332467  0.615690                      0.410             0.436   \n",
       "5985      0.173986  0.796232                      0.971             0.652   \n",
       "5986      0.506986  0.972205                      0.294             0.307   \n",
       "5987      0.740001  0.886605                      0.432             0.187   \n",
       "5988      0.509463  0.652241                      0.125             0.303   \n",
       "5989      0.658990  0.684071                      0.437             0.589   \n",
       "5990      0.806311  0.903785                      0.306             0.252   \n",
       "5991      0.487005  0.343684                      0.244             0.183   \n",
       "5992      0.520971  0.449663                      0.683             0.771   \n",
       "5993      0.962792  0.733775                      0.481             0.457   \n",
       "5994      0.058105  0.791039                      0.170             0.109   \n",
       "5995      0.201068  0.584273                      0.732             0.565   \n",
       "5996      0.393880  0.812248                      0.687             0.628   \n",
       "5997      0.490947  0.416804                      0.487             0.626   \n",
       "5998      0.790709  0.427386                      0.617             0.215   \n",
       "5999      0.553324  0.713402                      0.554             0.351   \n",
       "\n",
       "      key_confidence  mode_confidence                                  moods  \\\n",
       "0              0.664            1.000                           motivational   \n",
       "1              0.743            0.907                                 classy   \n",
       "2              0.704            0.813                                 sexual   \n",
       "3              0.167            0.983                           funky,mellow   \n",
       "4              0.248            1.000                              sprightly   \n",
       "5              0.579            0.763           classy,seductive,mellow,lush   \n",
       "6              0.216            1.000                     energetic,visceral   \n",
       "7              0.088            1.000                         cocky,visceral   \n",
       "8              0.950            0.773                            celebratory   \n",
       "9              0.034            1.000                        hypnotic,spacey   \n",
       "10             0.354            0.566                  hypnotic,funky,spacey   \n",
       "11             0.769            1.000                            trippy,lush   \n",
       "12             0.007            0.943                        happy,sprightly   \n",
       "13             0.064            1.000                    introspective,funky   \n",
       "14             0.498            0.781                 introspective,sad,lush   \n",
       "15             0.581            1.000                                 trippy   \n",
       "16             0.134            1.000                                 mellow   \n",
       "17             0.335            1.000                             aggressive   \n",
       "18             0.725            0.745                                 trippy   \n",
       "19             0.542            1.000                                 mellow   \n",
       "20             0.156            0.722                aggressive,motivational   \n",
       "21             0.762            0.881                        happy,sprightly   \n",
       "22             0.737            0.761                 energetic,angsty,rowdy   \n",
       "23             0.625            1.000                                  funky   \n",
       "24             0.185            0.943                                 classy   \n",
       "25             0.496            1.000                         cocky,visceral   \n",
       "26             0.126            0.979                                  rowdy   \n",
       "27             0.727            1.000                           funky,mellow   \n",
       "28             0.934            0.970           visceral,trippy,spacey,rowdy   \n",
       "29             0.871            1.000                      hypnotic,visceral   \n",
       "...              ...              ...                                    ...   \n",
       "5970           0.752            0.907                            celebratory   \n",
       "5971           0.124            0.500                 introspective,sad,lush   \n",
       "5972           0.407            1.000                                  funky   \n",
       "5973           0.642            0.510  seductive,sad,classy,nocturnal,mellow   \n",
       "5974           0.295            0.942                                 mellow   \n",
       "5975           0.305            0.894                                 classy   \n",
       "5976           0.888            0.968                         cold,nocturnal   \n",
       "5977           0.041            0.823                                 classy   \n",
       "5978           0.338            0.943                              sprightly   \n",
       "5979           0.239            0.420                           funky,mellow   \n",
       "5980           0.066            0.864                                  cocky   \n",
       "5981           0.285            0.829                             aggressive   \n",
       "5982           0.228            0.982                           motivational   \n",
       "5983           0.840            1.000  introspective,atmospheric,mellow,lush   \n",
       "5984           0.779            0.948  energetic,cocky,visceral,motivational   \n",
       "5985           0.322            1.000                     cocky,motivational   \n",
       "5986           0.924            1.000                           funky,mellow   \n",
       "5987           0.968            1.000                     energetic,visceral   \n",
       "5988           0.203            1.000                                  cocky   \n",
       "5989           0.239            1.000           classy,seductive,mellow,lush   \n",
       "5990           0.363            1.000                                  rowdy   \n",
       "5991           0.101            1.000                     cocky,motivational   \n",
       "5992           0.742            1.000                         aggressive,raw   \n",
       "5993           0.799            1.000                              energetic   \n",
       "5994           0.830            0.989            happy,energetic,celebratory   \n",
       "5995           0.141            1.000                             aggressive   \n",
       "5996           0.817            1.000                                  funky   \n",
       "5997           0.287            1.000                              sprightly   \n",
       "5998           0.722            1.000                                   lush   \n",
       "5999           0.376            0.833                                  cocky   \n",
       "\n",
       "     genres  \n",
       "0       rap  \n",
       "1      jazz  \n",
       "2       rap  \n",
       "3       rap  \n",
       "4      jazz  \n",
       "5      jazz  \n",
       "6     dance  \n",
       "7     dance  \n",
       "8     dance  \n",
       "9     dance  \n",
       "10      rap  \n",
       "11     jazz  \n",
       "12     jazz  \n",
       "13      rap  \n",
       "14     jazz  \n",
       "15     rock  \n",
       "16     rock  \n",
       "17     rock  \n",
       "18     rock  \n",
       "19     jazz  \n",
       "20     rock  \n",
       "21     jazz  \n",
       "22     rock  \n",
       "23      rap  \n",
       "24     jazz  \n",
       "25    dance  \n",
       "26     rock  \n",
       "27      rap  \n",
       "28    dance  \n",
       "29    dance  \n",
       "...     ...  \n",
       "5970  dance  \n",
       "5971   jazz  \n",
       "5972    rap  \n",
       "5973   jazz  \n",
       "5974   jazz  \n",
       "5975   jazz  \n",
       "5976  dance  \n",
       "5977   jazz  \n",
       "5978   jazz  \n",
       "5979    rap  \n",
       "5980    rap  \n",
       "5981   rock  \n",
       "5982    rap  \n",
       "5983   rock  \n",
       "5984  dance  \n",
       "5985    rap  \n",
       "5986    rap  \n",
       "5987  dance  \n",
       "5988    rap  \n",
       "5989   jazz  \n",
       "5990    rap  \n",
       "5991    rap  \n",
       "5992   rock  \n",
       "5993  dance  \n",
       "5994  dance  \n",
       "5995   rock  \n",
       "5996    rap  \n",
       "5997   rock  \n",
       "5998   jazz  \n",
       "5999    rap  \n",
       "\n",
       "[6000 rows x 19 columns]"
      ]
     },
     "execution_count": 15,
     "metadata": {},
     "output_type": "execute_result"
    }
   ],
   "source": [
    "assignment_2_dataset = assignment_2_dataset.sample(frac = 1, axis = 0, random_state = 101) #shuffling the data\n",
    "assignment_2_dataset.reset_index(drop = True, inplace = True)\n",
    "assignment_2_dataset"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 16,
   "metadata": {
    "collapsed": true
   },
   "outputs": [],
   "source": [
    "assignment_2_dataset.to_csv('week2/assignment-2_dataset.csv', header = True, index = False)"
   ]
  },
  {
   "cell_type": "markdown",
   "metadata": {
    "collapsed": true
   },
   "source": [
    "## Assignment 3"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 17,
   "metadata": {
    "scrolled": true
   },
   "outputs": [],
   "source": [
    "moods_ass_3 = song_dataset['moods'].apply(lambda x: x.split(sep = \",\"))\n",
    "\n",
    "raw_df = pd.concat([song_dataset['lyrics_features'], moods_ass_3], axis = 1)"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 18,
   "metadata": {
    "scrolled": true
   },
   "outputs": [
    {
     "data": {
      "text/html": [
       "<div>\n",
       "<style>\n",
       "    .dataframe thead tr:only-child th {\n",
       "        text-align: right;\n",
       "    }\n",
       "\n",
       "    .dataframe thead th {\n",
       "        text-align: left;\n",
       "    }\n",
       "\n",
       "    .dataframe tbody tr th {\n",
       "        vertical-align: top;\n",
       "    }\n",
       "</style>\n",
       "<table border=\"1\" class=\"dataframe\">\n",
       "  <thead>\n",
       "    <tr style=\"text-align: right;\">\n",
       "      <th></th>\n",
       "      <th>lyrics_features</th>\n",
       "      <th>moods</th>\n",
       "    </tr>\n",
       "  </thead>\n",
       "  <tbody>\n",
       "    <tr>\n",
       "      <th>0</th>\n",
       "      <td>[oppa, gangnam, style, gangnam, style, najeneu...</td>\n",
       "      <td>[energetic, motivational]</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>1</th>\n",
       "      <td>[lately, i, ve, been, i, ve, been, losing, sle...</td>\n",
       "      <td>[happy]</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>2</th>\n",
       "      <td>[party, rock, yeah, woo, let, s, go, party, ro...</td>\n",
       "      <td>[happy, celebratory, rowdy]</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>3</th>\n",
       "      <td>[alagamun, lan, weh, wakun, heya, hanun, gon, ...</td>\n",
       "      <td>[happy, energetic, celebratory]</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>4</th>\n",
       "      <td>[j, lo, the, other, side, out, my, mine, it, s...</td>\n",
       "      <td>[energetic]</td>\n",
       "    </tr>\n",
       "  </tbody>\n",
       "</table>\n",
       "</div>"
      ],
      "text/plain": [
       "                                     lyrics_features  \\\n",
       "0  [oppa, gangnam, style, gangnam, style, najeneu...   \n",
       "1  [lately, i, ve, been, i, ve, been, losing, sle...   \n",
       "2  [party, rock, yeah, woo, let, s, go, party, ro...   \n",
       "3  [alagamun, lan, weh, wakun, heya, hanun, gon, ...   \n",
       "4  [j, lo, the, other, side, out, my, mine, it, s...   \n",
       "\n",
       "                             moods  \n",
       "0        [energetic, motivational]  \n",
       "1                          [happy]  \n",
       "2      [happy, celebratory, rowdy]  \n",
       "3  [happy, energetic, celebratory]  \n",
       "4                      [energetic]  "
      ]
     },
     "execution_count": 18,
     "metadata": {},
     "output_type": "execute_result"
    }
   ],
   "source": [
    "raw_df.head()"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 19,
   "metadata": {
    "collapsed": true
   },
   "outputs": [],
   "source": [
    "from string import punctuation\n",
    "from sklearn.feature_extraction.stop_words import ENGLISH_STOP_WORDS\n",
    "from nltk.stem.snowball import SnowballStemmer\n",
    "\n",
    "translator = str.maketrans('','', punctuation)\n",
    "stemmer = SnowballStemmer('english')\n",
    "\n",
    "def clean_text(raw_text):\n",
    "    \n",
    "    clean_words = []\n",
    "    \n",
    "    raw_text = raw_text.lower()\n",
    "    raw_text = raw_text.translate(translator)\n",
    "    \n",
    "    for word in raw_text.split():\n",
    "        \n",
    "        if word not in ENGLISH_STOP_WORDS:\n",
    "            \n",
    "            clean_words.append(stemmer.stem(word))\n",
    "    \n",
    "    return \" \".join(clean_words)"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 20,
   "metadata": {
    "scrolled": true
   },
   "outputs": [],
   "source": [
    "raw_df['lyrics_features'] = raw_df['lyrics_features'].apply(lambda x: clean_text(\" \".join(x)))"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 21,
   "metadata": {
    "scrolled": true
   },
   "outputs": [
    {
     "data": {
      "text/html": [
       "<div>\n",
       "<style>\n",
       "    .dataframe thead tr:only-child th {\n",
       "        text-align: right;\n",
       "    }\n",
       "\n",
       "    .dataframe thead th {\n",
       "        text-align: left;\n",
       "    }\n",
       "\n",
       "    .dataframe tbody tr th {\n",
       "        vertical-align: top;\n",
       "    }\n",
       "</style>\n",
       "<table border=\"1\" class=\"dataframe\">\n",
       "  <thead>\n",
       "    <tr style=\"text-align: right;\">\n",
       "      <th></th>\n",
       "      <th>lyrics_features</th>\n",
       "      <th>moods</th>\n",
       "    </tr>\n",
       "  </thead>\n",
       "  <tbody>\n",
       "    <tr>\n",
       "      <th>0</th>\n",
       "      <td>oppa gangnam style gangnam style najeneun ttas...</td>\n",
       "      <td>[energetic, motivational]</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>1</th>\n",
       "      <td>late ve ve lose sleep dream thing babi ve ve p...</td>\n",
       "      <td>[happy]</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>2</th>\n",
       "      <td>parti rock yeah woo let s parti rock hous toni...</td>\n",
       "      <td>[happy, celebratory, rowdy]</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>3</th>\n",
       "      <td>alagamun lan weh wakun heya hanun gon alagamun...</td>\n",
       "      <td>[happy, energetic, celebratory]</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>4</th>\n",
       "      <td>j lo s new generat mr worldwid parti peopl flo...</td>\n",
       "      <td>[energetic]</td>\n",
       "    </tr>\n",
       "  </tbody>\n",
       "</table>\n",
       "</div>"
      ],
      "text/plain": [
       "                                     lyrics_features  \\\n",
       "0  oppa gangnam style gangnam style najeneun ttas...   \n",
       "1  late ve ve lose sleep dream thing babi ve ve p...   \n",
       "2  parti rock yeah woo let s parti rock hous toni...   \n",
       "3  alagamun lan weh wakun heya hanun gon alagamun...   \n",
       "4  j lo s new generat mr worldwid parti peopl flo...   \n",
       "\n",
       "                             moods  \n",
       "0        [energetic, motivational]  \n",
       "1                          [happy]  \n",
       "2      [happy, celebratory, rowdy]  \n",
       "3  [happy, energetic, celebratory]  \n",
       "4                      [energetic]  "
      ]
     },
     "execution_count": 21,
     "metadata": {},
     "output_type": "execute_result"
    }
   ],
   "source": [
    "raw_df.head()"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 22,
   "metadata": {},
   "outputs": [
    {
     "data": {
      "text/plain": [
       "(36733, 2)"
      ]
     },
     "execution_count": 22,
     "metadata": {},
     "output_type": "execute_result"
    }
   ],
   "source": [
    "raw_df.shape"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 23,
   "metadata": {},
   "outputs": [
    {
     "data": {
      "text/plain": [
       "(20930, 2)"
      ]
     },
     "execution_count": 23,
     "metadata": {},
     "output_type": "execute_result"
    }
   ],
   "source": [
    "big_clean_df = raw_df[raw_df['lyrics_features'] != \"\"].copy()\n",
    "big_clean_df.shape"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 24,
   "metadata": {},
   "outputs": [
    {
     "data": {
      "text/html": [
       "<div>\n",
       "<style>\n",
       "    .dataframe thead tr:only-child th {\n",
       "        text-align: right;\n",
       "    }\n",
       "\n",
       "    .dataframe thead th {\n",
       "        text-align: left;\n",
       "    }\n",
       "\n",
       "    .dataframe tbody tr th {\n",
       "        vertical-align: top;\n",
       "    }\n",
       "</style>\n",
       "<table border=\"1\" class=\"dataframe\">\n",
       "  <thead>\n",
       "    <tr style=\"text-align: right;\">\n",
       "      <th></th>\n",
       "      <th>lyrics_features</th>\n",
       "      <th>moods</th>\n",
       "    </tr>\n",
       "  </thead>\n",
       "  <tbody>\n",
       "    <tr>\n",
       "      <th>0</th>\n",
       "      <td>oppa gangnam style gangnam style najeneun ttas...</td>\n",
       "      <td>[energetic, motivational]</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>1</th>\n",
       "      <td>late ve ve lose sleep dream thing babi ve ve p...</td>\n",
       "      <td>[happy]</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>2</th>\n",
       "      <td>parti rock yeah woo let s parti rock hous toni...</td>\n",
       "      <td>[happy, celebratory, rowdy]</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>3</th>\n",
       "      <td>alagamun lan weh wakun heya hanun gon alagamun...</td>\n",
       "      <td>[happy, energetic, celebratory]</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>4</th>\n",
       "      <td>j lo s new generat mr worldwid parti peopl flo...</td>\n",
       "      <td>[energetic]</td>\n",
       "    </tr>\n",
       "  </tbody>\n",
       "</table>\n",
       "</div>"
      ],
      "text/plain": [
       "                                     lyrics_features  \\\n",
       "0  oppa gangnam style gangnam style najeneun ttas...   \n",
       "1  late ve ve lose sleep dream thing babi ve ve p...   \n",
       "2  parti rock yeah woo let s parti rock hous toni...   \n",
       "3  alagamun lan weh wakun heya hanun gon alagamun...   \n",
       "4  j lo s new generat mr worldwid parti peopl flo...   \n",
       "\n",
       "                             moods  \n",
       "0        [energetic, motivational]  \n",
       "1                          [happy]  \n",
       "2      [happy, celebratory, rowdy]  \n",
       "3  [happy, energetic, celebratory]  \n",
       "4                      [energetic]  "
      ]
     },
     "execution_count": 24,
     "metadata": {},
     "output_type": "execute_result"
    }
   ],
   "source": [
    "big_clean_df.reset_index(drop = True, inplace = True)\n",
    "big_clean_df.head()"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 25,
   "metadata": {},
   "outputs": [],
   "source": [
    "def picking_moods (target_moods, moods_list_row):\n",
    "    \"\"\" \n",
    "    `moods_list_row` is a list of moods of a row (a song).\n",
    "    Two or more target moods may exist in one `moods_list_row`, but only the first one in the list is returned.\n",
    "    \"\"\"\n",
    "    for mood in moods_list_row:\n",
    "        \n",
    "        if mood in target_moods:\n",
    "            \n",
    "            return mood # Return the first mood in the list of moods that is wanted"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 26,
   "metadata": {
    "collapsed": true
   },
   "outputs": [],
   "source": [
    "target_moods_df = big_clean_df.copy()\n",
    "target_moods_df['moods'] = target_moods_df['moods'].apply(lambda x: picking_moods(['happy', 'aggressive'], x))\n",
    "\n",
    "target_moods_df.dropna(inplace = True)\n",
    "\n",
    "target_moods_df.reset_index(drop = False, inplace = True)"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 27,
   "metadata": {
    "scrolled": true
   },
   "outputs": [
    {
     "data": {
      "text/html": [
       "<div>\n",
       "<style>\n",
       "    .dataframe thead tr:only-child th {\n",
       "        text-align: right;\n",
       "    }\n",
       "\n",
       "    .dataframe thead th {\n",
       "        text-align: left;\n",
       "    }\n",
       "\n",
       "    .dataframe tbody tr th {\n",
       "        vertical-align: top;\n",
       "    }\n",
       "</style>\n",
       "<table border=\"1\" class=\"dataframe\">\n",
       "  <thead>\n",
       "    <tr style=\"text-align: right;\">\n",
       "      <th></th>\n",
       "      <th>index</th>\n",
       "      <th>lyrics_features</th>\n",
       "      <th>moods</th>\n",
       "    </tr>\n",
       "  </thead>\n",
       "  <tbody>\n",
       "    <tr>\n",
       "      <th>0</th>\n",
       "      <td>1</td>\n",
       "      <td>late ve ve lose sleep dream thing babi ve ve p...</td>\n",
       "      <td>happy</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>1</th>\n",
       "      <td>2</td>\n",
       "      <td>parti rock yeah woo let s parti rock hous toni...</td>\n",
       "      <td>happy</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>2</th>\n",
       "      <td>3</td>\n",
       "      <td>alagamun lan weh wakun heya hanun gon alagamun...</td>\n",
       "      <td>happy</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>3</th>\n",
       "      <td>5</td>\n",
       "      <td>today don t feel like do just wanna lay bed do...</td>\n",
       "      <td>happy</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>4</th>\n",
       "      <td>9</td>\n",
       "      <td>don t know turn head walk o don t need make co...</td>\n",
       "      <td>happy</td>\n",
       "    </tr>\n",
       "  </tbody>\n",
       "</table>\n",
       "</div>"
      ],
      "text/plain": [
       "   index                                    lyrics_features  moods\n",
       "0      1  late ve ve lose sleep dream thing babi ve ve p...  happy\n",
       "1      2  parti rock yeah woo let s parti rock hous toni...  happy\n",
       "2      3  alagamun lan weh wakun heya hanun gon alagamun...  happy\n",
       "3      5  today don t feel like do just wanna lay bed do...  happy\n",
       "4      9  don t know turn head walk o don t need make co...  happy"
      ]
     },
     "execution_count": 27,
     "metadata": {},
     "output_type": "execute_result"
    }
   ],
   "source": [
    "target_moods_df.head()"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 28,
   "metadata": {},
   "outputs": [
    {
     "data": {
      "text/plain": [
       "Index(['index', 'lyrics_features', 'moods'], dtype='object')"
      ]
     },
     "execution_count": 28,
     "metadata": {},
     "output_type": "execute_result"
    }
   ],
   "source": [
    "target_moods_df.columns"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 29,
   "metadata": {},
   "outputs": [
    {
     "data": {
      "text/plain": [
       "1683"
      ]
     },
     "execution_count": 29,
     "metadata": {},
     "output_type": "execute_result"
    }
   ],
   "source": [
    "len(target_moods_df[target_moods_df['moods'] == 'aggressive'])"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 30,
   "metadata": {},
   "outputs": [
    {
     "data": {
      "text/plain": [
       "1757"
      ]
     },
     "execution_count": 30,
     "metadata": {},
     "output_type": "execute_result"
    }
   ],
   "source": [
    "len(target_moods_df[target_moods_df['moods'] == 'happy'])"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 31,
   "metadata": {
    "collapsed": true
   },
   "outputs": [],
   "source": [
    "big_clean_df.to_csv('week3/assignment-3_big-dataset.csv', header = True, index = False)\n",
    "target_moods_df.to_csv('week3/assignment-3_small-dataset.csv', header = True, index = False)"
   ]
  }
 ],
 "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.6.3"
  }
 },
 "nbformat": 4,
 "nbformat_minor": 2
}