TEST03. Testing similaritydf to get dataframe.ipynb

{
 "cells": [
  {
   "cell_type": "code",
   "execution_count": 1,
   "metadata": {},
   "outputs": [],
   "source": [
    "import similaritydf\n",
    "from similaritydf import get_df"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 2,
   "metadata": {},
   "outputs": [],
   "source": [
    "topics = ['Kygo','Odesza','Armin van Buuren','Ilan Bluestone','Grum','Cascada','Kobe Bryant','LeBron James','Michael Jordan','Cristiano Ronaldo','Lionel Messi','Kylian Mbappé','Chemistry','Biology','Physics','Periodic table']"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 3,
   "metadata": {},
   "outputs": [
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "['Kygo', 'Odesza', 'Armin van Buuren', 'Ilan Bluestone', 'Grum', 'Cascada', 'Kobe Bryant', 'LeBron James', 'Michael Jordan', 'Cristiano Ronaldo', 'Lionel Messi', 'Kylian Mbappé', 'Chemistry', 'Biology', 'Physics', 'Periodic table']\n"
     ]
    }
   ],
   "source": [
    "test_df = get_df(topics)"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 4,
   "metadata": {},
   "outputs": [],
   "source": [
    "def see_topic(df, val):\n",
    "    return df[(df['Topic 1'] == topics[val]) | (df['Topic 2'] == topics[val])].sort_values('Probability', ascending=False)"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 5,
   "metadata": {},
   "outputs": [
    {
     "data": {
      "text/html": [
       "<div>\n",
       "<style scoped>\n",
       "    .dataframe tbody tr th:only-of-type {\n",
       "        vertical-align: middle;\n",
       "    }\n",
       "\n",
       "    .dataframe tbody tr th {\n",
       "        vertical-align: top;\n",
       "    }\n",
       "\n",
       "    .dataframe thead th {\n",
       "        text-align: right;\n",
       "    }\n",
       "</style>\n",
       "<table border=\"1\" class=\"dataframe\">\n",
       "  <thead>\n",
       "    <tr style=\"text-align: right;\">\n",
       "      <th></th>\n",
       "      <th>Topic 1</th>\n",
       "      <th>Topic 2</th>\n",
       "      <th>Probability</th>\n",
       "      <th>Similar?</th>\n",
       "    </tr>\n",
       "  </thead>\n",
       "  <tbody>\n",
       "    <tr>\n",
       "      <td>3</td>\n",
       "      <td>Kygo</td>\n",
       "      <td>Grum</td>\n",
       "      <td>100</td>\n",
       "      <td>Yes</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <td>4</td>\n",
       "      <td>Kygo</td>\n",
       "      <td>Cascada</td>\n",
       "      <td>91</td>\n",
       "      <td>Yes</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <td>0</td>\n",
       "      <td>Kygo</td>\n",
       "      <td>Odesza</td>\n",
       "      <td>66</td>\n",
       "      <td>Yes</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <td>1</td>\n",
       "      <td>Kygo</td>\n",
       "      <td>Armin van Buuren</td>\n",
       "      <td>65</td>\n",
       "      <td>Yes</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <td>2</td>\n",
       "      <td>Kygo</td>\n",
       "      <td>Ilan Bluestone</td>\n",
       "      <td>34</td>\n",
       "      <td>No</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <td>5</td>\n",
       "      <td>Kygo</td>\n",
       "      <td>Kobe Bryant</td>\n",
       "      <td>17</td>\n",
       "      <td>No</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <td>11</td>\n",
       "      <td>Kygo</td>\n",
       "      <td>Chemistry</td>\n",
       "      <td>14</td>\n",
       "      <td>No</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <td>8</td>\n",
       "      <td>Kygo</td>\n",
       "      <td>Cristiano Ronaldo</td>\n",
       "      <td>7</td>\n",
       "      <td>No</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <td>9</td>\n",
       "      <td>Kygo</td>\n",
       "      <td>Lionel Messi</td>\n",
       "      <td>7</td>\n",
       "      <td>No</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <td>12</td>\n",
       "      <td>Kygo</td>\n",
       "      <td>Biology</td>\n",
       "      <td>5</td>\n",
       "      <td>No</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <td>6</td>\n",
       "      <td>Kygo</td>\n",
       "      <td>LeBron James</td>\n",
       "      <td>4</td>\n",
       "      <td>No</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <td>7</td>\n",
       "      <td>Kygo</td>\n",
       "      <td>Michael Jordan</td>\n",
       "      <td>4</td>\n",
       "      <td>No</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <td>14</td>\n",
       "      <td>Kygo</td>\n",
       "      <td>Periodic table</td>\n",
       "      <td>4</td>\n",
       "      <td>No</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <td>13</td>\n",
       "      <td>Kygo</td>\n",
       "      <td>Physics</td>\n",
       "      <td>3</td>\n",
       "      <td>No</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <td>10</td>\n",
       "      <td>Kygo</td>\n",
       "      <td>Kylian Mbappé</td>\n",
       "      <td>1</td>\n",
       "      <td>No</td>\n",
       "    </tr>\n",
       "  </tbody>\n",
       "</table>\n",
       "</div>"
      ],
      "text/plain": [
       "   Topic 1            Topic 2  Probability Similar?\n",
       "3     Kygo               Grum          100      Yes\n",
       "4     Kygo            Cascada           91      Yes\n",
       "0     Kygo             Odesza           66      Yes\n",
       "1     Kygo   Armin van Buuren           65      Yes\n",
       "2     Kygo     Ilan Bluestone           34       No\n",
       "5     Kygo        Kobe Bryant           17       No\n",
       "11    Kygo          Chemistry           14       No\n",
       "8     Kygo  Cristiano Ronaldo            7       No\n",
       "9     Kygo       Lionel Messi            7       No\n",
       "12    Kygo            Biology            5       No\n",
       "6     Kygo       LeBron James            4       No\n",
       "7     Kygo     Michael Jordan            4       No\n",
       "14    Kygo     Periodic table            4       No\n",
       "13    Kygo            Physics            3       No\n",
       "10    Kygo      Kylian Mbappé            1       No"
      ]
     },
     "execution_count": 5,
     "metadata": {},
     "output_type": "execute_result"
    }
   ],
   "source": [
    "see_topic(test_df, 0)"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 6,
   "metadata": {},
   "outputs": [
    {
     "data": {
      "text/html": [
       "<div>\n",
       "<style scoped>\n",
       "    .dataframe tbody tr th:only-of-type {\n",
       "        vertical-align: middle;\n",
       "    }\n",
       "\n",
       "    .dataframe tbody tr th {\n",
       "        vertical-align: top;\n",
       "    }\n",
       "\n",
       "    .dataframe thead th {\n",
       "        text-align: right;\n",
       "    }\n",
       "</style>\n",
       "<table border=\"1\" class=\"dataframe\">\n",
       "  <thead>\n",
       "    <tr style=\"text-align: right;\">\n",
       "      <th></th>\n",
       "      <th>Topic 1</th>\n",
       "      <th>Topic 2</th>\n",
       "      <th>Probability</th>\n",
       "      <th>Similar?</th>\n",
       "    </tr>\n",
       "  </thead>\n",
       "  <tbody>\n",
       "    <tr>\n",
       "      <td>18</td>\n",
       "      <td>Odesza</td>\n",
       "      <td>Cascada</td>\n",
       "      <td>86</td>\n",
       "      <td>Yes</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <td>17</td>\n",
       "      <td>Odesza</td>\n",
       "      <td>Grum</td>\n",
       "      <td>82</td>\n",
       "      <td>Yes</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <td>15</td>\n",
       "      <td>Odesza</td>\n",
       "      <td>Armin van Buuren</td>\n",
       "      <td>79</td>\n",
       "      <td>Yes</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <td>0</td>\n",
       "      <td>Kygo</td>\n",
       "      <td>Odesza</td>\n",
       "      <td>66</td>\n",
       "      <td>Yes</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <td>16</td>\n",
       "      <td>Odesza</td>\n",
       "      <td>Ilan Bluestone</td>\n",
       "      <td>27</td>\n",
       "      <td>No</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <td>19</td>\n",
       "      <td>Odesza</td>\n",
       "      <td>Kobe Bryant</td>\n",
       "      <td>20</td>\n",
       "      <td>No</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <td>25</td>\n",
       "      <td>Odesza</td>\n",
       "      <td>Chemistry</td>\n",
       "      <td>18</td>\n",
       "      <td>No</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <td>23</td>\n",
       "      <td>Odesza</td>\n",
       "      <td>Lionel Messi</td>\n",
       "      <td>13</td>\n",
       "      <td>No</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <td>26</td>\n",
       "      <td>Odesza</td>\n",
       "      <td>Biology</td>\n",
       "      <td>12</td>\n",
       "      <td>No</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <td>24</td>\n",
       "      <td>Odesza</td>\n",
       "      <td>Kylian Mbappé</td>\n",
       "      <td>11</td>\n",
       "      <td>No</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <td>22</td>\n",
       "      <td>Odesza</td>\n",
       "      <td>Cristiano Ronaldo</td>\n",
       "      <td>10</td>\n",
       "      <td>No</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <td>20</td>\n",
       "      <td>Odesza</td>\n",
       "      <td>LeBron James</td>\n",
       "      <td>8</td>\n",
       "      <td>No</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <td>27</td>\n",
       "      <td>Odesza</td>\n",
       "      <td>Physics</td>\n",
       "      <td>8</td>\n",
       "      <td>No</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <td>21</td>\n",
       "      <td>Odesza</td>\n",
       "      <td>Michael Jordan</td>\n",
       "      <td>7</td>\n",
       "      <td>No</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <td>28</td>\n",
       "      <td>Odesza</td>\n",
       "      <td>Periodic table</td>\n",
       "      <td>6</td>\n",
       "      <td>No</td>\n",
       "    </tr>\n",
       "  </tbody>\n",
       "</table>\n",
       "</div>"
      ],
      "text/plain": [
       "   Topic 1            Topic 2  Probability Similar?\n",
       "18  Odesza            Cascada           86      Yes\n",
       "17  Odesza               Grum           82      Yes\n",
       "15  Odesza   Armin van Buuren           79      Yes\n",
       "0     Kygo             Odesza           66      Yes\n",
       "16  Odesza     Ilan Bluestone           27       No\n",
       "19  Odesza        Kobe Bryant           20       No\n",
       "25  Odesza          Chemistry           18       No\n",
       "23  Odesza       Lionel Messi           13       No\n",
       "26  Odesza            Biology           12       No\n",
       "24  Odesza      Kylian Mbappé           11       No\n",
       "22  Odesza  Cristiano Ronaldo           10       No\n",
       "20  Odesza       LeBron James            8       No\n",
       "27  Odesza            Physics            8       No\n",
       "21  Odesza     Michael Jordan            7       No\n",
       "28  Odesza     Periodic table            6       No"
      ]
     },
     "execution_count": 6,
     "metadata": {},
     "output_type": "execute_result"
    }
   ],
   "source": [
    "see_topic(test_df, 1)"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 7,
   "metadata": {},
   "outputs": [
    {
     "data": {
      "text/html": [
       "<div>\n",
       "<style scoped>\n",
       "    .dataframe tbody tr th:only-of-type {\n",
       "        vertical-align: middle;\n",
       "    }\n",
       "\n",
       "    .dataframe tbody tr th {\n",
       "        vertical-align: top;\n",
       "    }\n",
       "\n",
       "    .dataframe thead th {\n",
       "        text-align: right;\n",
       "    }\n",
       "</style>\n",
       "<table border=\"1\" class=\"dataframe\">\n",
       "  <thead>\n",
       "    <tr style=\"text-align: right;\">\n",
       "      <th></th>\n",
       "      <th>Topic 1</th>\n",
       "      <th>Topic 2</th>\n",
       "      <th>Probability</th>\n",
       "      <th>Similar?</th>\n",
       "    </tr>\n",
       "  </thead>\n",
       "  <tbody>\n",
       "    <tr>\n",
       "      <td>31</td>\n",
       "      <td>Armin van Buuren</td>\n",
       "      <td>Cascada</td>\n",
       "      <td>94</td>\n",
       "      <td>Yes</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <td>30</td>\n",
       "      <td>Armin van Buuren</td>\n",
       "      <td>Grum</td>\n",
       "      <td>91</td>\n",
       "      <td>Yes</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <td>15</td>\n",
       "      <td>Odesza</td>\n",
       "      <td>Armin van Buuren</td>\n",
       "      <td>79</td>\n",
       "      <td>Yes</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <td>1</td>\n",
       "      <td>Kygo</td>\n",
       "      <td>Armin van Buuren</td>\n",
       "      <td>65</td>\n",
       "      <td>Yes</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <td>29</td>\n",
       "      <td>Armin van Buuren</td>\n",
       "      <td>Ilan Bluestone</td>\n",
       "      <td>37</td>\n",
       "      <td>No</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <td>38</td>\n",
       "      <td>Armin van Buuren</td>\n",
       "      <td>Chemistry</td>\n",
       "      <td>22</td>\n",
       "      <td>No</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <td>39</td>\n",
       "      <td>Armin van Buuren</td>\n",
       "      <td>Biology</td>\n",
       "      <td>15</td>\n",
       "      <td>No</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <td>32</td>\n",
       "      <td>Armin van Buuren</td>\n",
       "      <td>Kobe Bryant</td>\n",
       "      <td>13</td>\n",
       "      <td>No</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <td>40</td>\n",
       "      <td>Armin van Buuren</td>\n",
       "      <td>Physics</td>\n",
       "      <td>13</td>\n",
       "      <td>No</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <td>36</td>\n",
       "      <td>Armin van Buuren</td>\n",
       "      <td>Lionel Messi</td>\n",
       "      <td>12</td>\n",
       "      <td>No</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <td>35</td>\n",
       "      <td>Armin van Buuren</td>\n",
       "      <td>Cristiano Ronaldo</td>\n",
       "      <td>11</td>\n",
       "      <td>No</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <td>41</td>\n",
       "      <td>Armin van Buuren</td>\n",
       "      <td>Periodic table</td>\n",
       "      <td>9</td>\n",
       "      <td>No</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <td>37</td>\n",
       "      <td>Armin van Buuren</td>\n",
       "      <td>Kylian Mbappé</td>\n",
       "      <td>8</td>\n",
       "      <td>No</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <td>33</td>\n",
       "      <td>Armin van Buuren</td>\n",
       "      <td>LeBron James</td>\n",
       "      <td>5</td>\n",
       "      <td>No</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <td>34</td>\n",
       "      <td>Armin van Buuren</td>\n",
       "      <td>Michael Jordan</td>\n",
       "      <td>1</td>\n",
       "      <td>No</td>\n",
       "    </tr>\n",
       "  </tbody>\n",
       "</table>\n",
       "</div>"
      ],
      "text/plain": [
       "             Topic 1            Topic 2  Probability Similar?\n",
       "31  Armin van Buuren            Cascada           94      Yes\n",
       "30  Armin van Buuren               Grum           91      Yes\n",
       "15            Odesza   Armin van Buuren           79      Yes\n",
       "1               Kygo   Armin van Buuren           65      Yes\n",
       "29  Armin van Buuren     Ilan Bluestone           37       No\n",
       "38  Armin van Buuren          Chemistry           22       No\n",
       "39  Armin van Buuren            Biology           15       No\n",
       "32  Armin van Buuren        Kobe Bryant           13       No\n",
       "40  Armin van Buuren            Physics           13       No\n",
       "36  Armin van Buuren       Lionel Messi           12       No\n",
       "35  Armin van Buuren  Cristiano Ronaldo           11       No\n",
       "41  Armin van Buuren     Periodic table            9       No\n",
       "37  Armin van Buuren      Kylian Mbappé            8       No\n",
       "33  Armin van Buuren       LeBron James            5       No\n",
       "34  Armin van Buuren     Michael Jordan            1       No"
      ]
     },
     "execution_count": 7,
     "metadata": {},
     "output_type": "execute_result"
    }
   ],
   "source": [
    "see_topic(test_df, 2)"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 8,
   "metadata": {},
   "outputs": [
    {
     "data": {
      "text/html": [
       "<div>\n",
       "<style scoped>\n",
       "    .dataframe tbody tr th:only-of-type {\n",
       "        vertical-align: middle;\n",
       "    }\n",
       "\n",
       "    .dataframe tbody tr th {\n",
       "        vertical-align: top;\n",
       "    }\n",
       "\n",
       "    .dataframe thead th {\n",
       "        text-align: right;\n",
       "    }\n",
       "</style>\n",
       "<table border=\"1\" class=\"dataframe\">\n",
       "  <thead>\n",
       "    <tr style=\"text-align: right;\">\n",
       "      <th></th>\n",
       "      <th>Topic 1</th>\n",
       "      <th>Topic 2</th>\n",
       "      <th>Probability</th>\n",
       "      <th>Similar?</th>\n",
       "    </tr>\n",
       "  </thead>\n",
       "  <tbody>\n",
       "    <tr>\n",
       "      <td>42</td>\n",
       "      <td>Ilan Bluestone</td>\n",
       "      <td>Grum</td>\n",
       "      <td>46</td>\n",
       "      <td>No</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <td>29</td>\n",
       "      <td>Armin van Buuren</td>\n",
       "      <td>Ilan Bluestone</td>\n",
       "      <td>37</td>\n",
       "      <td>No</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <td>2</td>\n",
       "      <td>Kygo</td>\n",
       "      <td>Ilan Bluestone</td>\n",
       "      <td>34</td>\n",
       "      <td>No</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <td>16</td>\n",
       "      <td>Odesza</td>\n",
       "      <td>Ilan Bluestone</td>\n",
       "      <td>27</td>\n",
       "      <td>No</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <td>50</td>\n",
       "      <td>Ilan Bluestone</td>\n",
       "      <td>Chemistry</td>\n",
       "      <td>20</td>\n",
       "      <td>No</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <td>43</td>\n",
       "      <td>Ilan Bluestone</td>\n",
       "      <td>Cascada</td>\n",
       "      <td>18</td>\n",
       "      <td>No</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <td>44</td>\n",
       "      <td>Ilan Bluestone</td>\n",
       "      <td>Kobe Bryant</td>\n",
       "      <td>13</td>\n",
       "      <td>No</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <td>48</td>\n",
       "      <td>Ilan Bluestone</td>\n",
       "      <td>Lionel Messi</td>\n",
       "      <td>10</td>\n",
       "      <td>No</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <td>51</td>\n",
       "      <td>Ilan Bluestone</td>\n",
       "      <td>Biology</td>\n",
       "      <td>10</td>\n",
       "      <td>No</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <td>47</td>\n",
       "      <td>Ilan Bluestone</td>\n",
       "      <td>Cristiano Ronaldo</td>\n",
       "      <td>9</td>\n",
       "      <td>No</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <td>52</td>\n",
       "      <td>Ilan Bluestone</td>\n",
       "      <td>Physics</td>\n",
       "      <td>6</td>\n",
       "      <td>No</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <td>53</td>\n",
       "      <td>Ilan Bluestone</td>\n",
       "      <td>Periodic table</td>\n",
       "      <td>6</td>\n",
       "      <td>No</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <td>45</td>\n",
       "      <td>Ilan Bluestone</td>\n",
       "      <td>LeBron James</td>\n",
       "      <td>5</td>\n",
       "      <td>No</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <td>49</td>\n",
       "      <td>Ilan Bluestone</td>\n",
       "      <td>Kylian Mbappé</td>\n",
       "      <td>5</td>\n",
       "      <td>No</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <td>46</td>\n",
       "      <td>Ilan Bluestone</td>\n",
       "      <td>Michael Jordan</td>\n",
       "      <td>3</td>\n",
       "      <td>No</td>\n",
       "    </tr>\n",
       "  </tbody>\n",
       "</table>\n",
       "</div>"
      ],
      "text/plain": [
       "             Topic 1            Topic 2  Probability Similar?\n",
       "42    Ilan Bluestone               Grum           46       No\n",
       "29  Armin van Buuren     Ilan Bluestone           37       No\n",
       "2               Kygo     Ilan Bluestone           34       No\n",
       "16            Odesza     Ilan Bluestone           27       No\n",
       "50    Ilan Bluestone          Chemistry           20       No\n",
       "43    Ilan Bluestone            Cascada           18       No\n",
       "44    Ilan Bluestone        Kobe Bryant           13       No\n",
       "48    Ilan Bluestone       Lionel Messi           10       No\n",
       "51    Ilan Bluestone            Biology           10       No\n",
       "47    Ilan Bluestone  Cristiano Ronaldo            9       No\n",
       "52    Ilan Bluestone            Physics            6       No\n",
       "53    Ilan Bluestone     Periodic table            6       No\n",
       "45    Ilan Bluestone       LeBron James            5       No\n",
       "49    Ilan Bluestone      Kylian Mbappé            5       No\n",
       "46    Ilan Bluestone     Michael Jordan            3       No"
      ]
     },
     "execution_count": 8,
     "metadata": {},
     "output_type": "execute_result"
    }
   ],
   "source": [
    "see_topic(test_df, 3)"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 9,
   "metadata": {},
   "outputs": [
    {
     "data": {
      "text/html": [
       "<div>\n",
       "<style scoped>\n",
       "    .dataframe tbody tr th:only-of-type {\n",
       "        vertical-align: middle;\n",
       "    }\n",
       "\n",
       "    .dataframe tbody tr th {\n",
       "        vertical-align: top;\n",
       "    }\n",
       "\n",
       "    .dataframe thead th {\n",
       "        text-align: right;\n",
       "    }\n",
       "</style>\n",
       "<table border=\"1\" class=\"dataframe\">\n",
       "  <thead>\n",
       "    <tr style=\"text-align: right;\">\n",
       "      <th></th>\n",
       "      <th>Topic 1</th>\n",
       "      <th>Topic 2</th>\n",
       "      <th>Probability</th>\n",
       "      <th>Similar?</th>\n",
       "    </tr>\n",
       "  </thead>\n",
       "  <tbody>\n",
       "    <tr>\n",
       "      <td>3</td>\n",
       "      <td>Kygo</td>\n",
       "      <td>Grum</td>\n",
       "      <td>100</td>\n",
       "      <td>Yes</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <td>54</td>\n",
       "      <td>Grum</td>\n",
       "      <td>Cascada</td>\n",
       "      <td>100</td>\n",
       "      <td>Yes</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <td>30</td>\n",
       "      <td>Armin van Buuren</td>\n",
       "      <td>Grum</td>\n",
       "      <td>91</td>\n",
       "      <td>Yes</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <td>17</td>\n",
       "      <td>Odesza</td>\n",
       "      <td>Grum</td>\n",
       "      <td>82</td>\n",
       "      <td>Yes</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <td>42</td>\n",
       "      <td>Ilan Bluestone</td>\n",
       "      <td>Grum</td>\n",
       "      <td>46</td>\n",
       "      <td>No</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <td>61</td>\n",
       "      <td>Grum</td>\n",
       "      <td>Chemistry</td>\n",
       "      <td>24</td>\n",
       "      <td>No</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <td>63</td>\n",
       "      <td>Grum</td>\n",
       "      <td>Physics</td>\n",
       "      <td>14</td>\n",
       "      <td>No</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <td>62</td>\n",
       "      <td>Grum</td>\n",
       "      <td>Biology</td>\n",
       "      <td>13</td>\n",
       "      <td>No</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <td>64</td>\n",
       "      <td>Grum</td>\n",
       "      <td>Periodic table</td>\n",
       "      <td>12</td>\n",
       "      <td>No</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <td>55</td>\n",
       "      <td>Grum</td>\n",
       "      <td>Kobe Bryant</td>\n",
       "      <td>9</td>\n",
       "      <td>No</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <td>59</td>\n",
       "      <td>Grum</td>\n",
       "      <td>Lionel Messi</td>\n",
       "      <td>9</td>\n",
       "      <td>No</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <td>58</td>\n",
       "      <td>Grum</td>\n",
       "      <td>Cristiano Ronaldo</td>\n",
       "      <td>8</td>\n",
       "      <td>No</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <td>60</td>\n",
       "      <td>Grum</td>\n",
       "      <td>Kylian Mbappé</td>\n",
       "      <td>6</td>\n",
       "      <td>No</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <td>56</td>\n",
       "      <td>Grum</td>\n",
       "      <td>LeBron James</td>\n",
       "      <td>2</td>\n",
       "      <td>No</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <td>57</td>\n",
       "      <td>Grum</td>\n",
       "      <td>Michael Jordan</td>\n",
       "      <td>0</td>\n",
       "      <td>No</td>\n",
       "    </tr>\n",
       "  </tbody>\n",
       "</table>\n",
       "</div>"
      ],
      "text/plain": [
       "             Topic 1            Topic 2  Probability Similar?\n",
       "3               Kygo               Grum          100      Yes\n",
       "54              Grum            Cascada          100      Yes\n",
       "30  Armin van Buuren               Grum           91      Yes\n",
       "17            Odesza               Grum           82      Yes\n",
       "42    Ilan Bluestone               Grum           46       No\n",
       "61              Grum          Chemistry           24       No\n",
       "63              Grum            Physics           14       No\n",
       "62              Grum            Biology           13       No\n",
       "64              Grum     Periodic table           12       No\n",
       "55              Grum        Kobe Bryant            9       No\n",
       "59              Grum       Lionel Messi            9       No\n",
       "58              Grum  Cristiano Ronaldo            8       No\n",
       "60              Grum      Kylian Mbappé            6       No\n",
       "56              Grum       LeBron James            2       No\n",
       "57              Grum     Michael Jordan            0       No"
      ]
     },
     "execution_count": 9,
     "metadata": {},
     "output_type": "execute_result"
    }
   ],
   "source": [
    "see_topic(test_df, 4)"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 10,
   "metadata": {},
   "outputs": [
    {
     "data": {
      "text/html": [
       "<div>\n",
       "<style scoped>\n",
       "    .dataframe tbody tr th:only-of-type {\n",
       "        vertical-align: middle;\n",
       "    }\n",
       "\n",
       "    .dataframe tbody tr th {\n",
       "        vertical-align: top;\n",
       "    }\n",
       "\n",
       "    .dataframe thead th {\n",
       "        text-align: right;\n",
       "    }\n",
       "</style>\n",
       "<table border=\"1\" class=\"dataframe\">\n",
       "  <thead>\n",
       "    <tr style=\"text-align: right;\">\n",
       "      <th></th>\n",
       "      <th>Topic 1</th>\n",
       "      <th>Topic 2</th>\n",
       "      <th>Probability</th>\n",
       "      <th>Similar?</th>\n",
       "    </tr>\n",
       "  </thead>\n",
       "  <tbody>\n",
       "    <tr>\n",
       "      <td>54</td>\n",
       "      <td>Grum</td>\n",
       "      <td>Cascada</td>\n",
       "      <td>100</td>\n",
       "      <td>Yes</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <td>31</td>\n",
       "      <td>Armin van Buuren</td>\n",
       "      <td>Cascada</td>\n",
       "      <td>94</td>\n",
       "      <td>Yes</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <td>4</td>\n",
       "      <td>Kygo</td>\n",
       "      <td>Cascada</td>\n",
       "      <td>91</td>\n",
       "      <td>Yes</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <td>18</td>\n",
       "      <td>Odesza</td>\n",
       "      <td>Cascada</td>\n",
       "      <td>86</td>\n",
       "      <td>Yes</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <td>73</td>\n",
       "      <td>Cascada</td>\n",
       "      <td>Physics</td>\n",
       "      <td>19</td>\n",
       "      <td>No</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <td>43</td>\n",
       "      <td>Ilan Bluestone</td>\n",
       "      <td>Cascada</td>\n",
       "      <td>18</td>\n",
       "      <td>No</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <td>65</td>\n",
       "      <td>Cascada</td>\n",
       "      <td>Kobe Bryant</td>\n",
       "      <td>14</td>\n",
       "      <td>No</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <td>68</td>\n",
       "      <td>Cascada</td>\n",
       "      <td>Cristiano Ronaldo</td>\n",
       "      <td>13</td>\n",
       "      <td>No</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <td>71</td>\n",
       "      <td>Cascada</td>\n",
       "      <td>Chemistry</td>\n",
       "      <td>13</td>\n",
       "      <td>No</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <td>69</td>\n",
       "      <td>Cascada</td>\n",
       "      <td>Lionel Messi</td>\n",
       "      <td>11</td>\n",
       "      <td>No</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <td>70</td>\n",
       "      <td>Cascada</td>\n",
       "      <td>Kylian Mbappé</td>\n",
       "      <td>10</td>\n",
       "      <td>No</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <td>72</td>\n",
       "      <td>Cascada</td>\n",
       "      <td>Biology</td>\n",
       "      <td>10</td>\n",
       "      <td>No</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <td>74</td>\n",
       "      <td>Cascada</td>\n",
       "      <td>Periodic table</td>\n",
       "      <td>4</td>\n",
       "      <td>No</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <td>66</td>\n",
       "      <td>Cascada</td>\n",
       "      <td>LeBron James</td>\n",
       "      <td>0</td>\n",
       "      <td>No</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <td>67</td>\n",
       "      <td>Cascada</td>\n",
       "      <td>Michael Jordan</td>\n",
       "      <td>0</td>\n",
       "      <td>No</td>\n",
       "    </tr>\n",
       "  </tbody>\n",
       "</table>\n",
       "</div>"
      ],
      "text/plain": [
       "             Topic 1            Topic 2  Probability Similar?\n",
       "54              Grum            Cascada          100      Yes\n",
       "31  Armin van Buuren            Cascada           94      Yes\n",
       "4               Kygo            Cascada           91      Yes\n",
       "18            Odesza            Cascada           86      Yes\n",
       "73           Cascada            Physics           19       No\n",
       "43    Ilan Bluestone            Cascada           18       No\n",
       "65           Cascada        Kobe Bryant           14       No\n",
       "68           Cascada  Cristiano Ronaldo           13       No\n",
       "71           Cascada          Chemistry           13       No\n",
       "69           Cascada       Lionel Messi           11       No\n",
       "70           Cascada      Kylian Mbappé           10       No\n",
       "72           Cascada            Biology           10       No\n",
       "74           Cascada     Periodic table            4       No\n",
       "66           Cascada       LeBron James            0       No\n",
       "67           Cascada     Michael Jordan            0       No"
      ]
     },
     "execution_count": 10,
     "metadata": {},
     "output_type": "execute_result"
    }
   ],
   "source": [
    "see_topic(test_df, 5)"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 11,
   "metadata": {},
   "outputs": [
    {
     "data": {
      "text/html": [
       "<div>\n",
       "<style scoped>\n",
       "    .dataframe tbody tr th:only-of-type {\n",
       "        vertical-align: middle;\n",
       "    }\n",
       "\n",
       "    .dataframe tbody tr th {\n",
       "        vertical-align: top;\n",
       "    }\n",
       "\n",
       "    .dataframe thead th {\n",
       "        text-align: right;\n",
       "    }\n",
       "</style>\n",
       "<table border=\"1\" class=\"dataframe\">\n",
       "  <thead>\n",
       "    <tr style=\"text-align: right;\">\n",
       "      <th></th>\n",
       "      <th>Topic 1</th>\n",
       "      <th>Topic 2</th>\n",
       "      <th>Probability</th>\n",
       "      <th>Similar?</th>\n",
       "    </tr>\n",
       "  </thead>\n",
       "  <tbody>\n",
       "    <tr>\n",
       "      <td>75</td>\n",
       "      <td>Kobe Bryant</td>\n",
       "      <td>LeBron James</td>\n",
       "      <td>100</td>\n",
       "      <td>Yes</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <td>76</td>\n",
       "      <td>Kobe Bryant</td>\n",
       "      <td>Michael Jordan</td>\n",
       "      <td>100</td>\n",
       "      <td>Yes</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <td>78</td>\n",
       "      <td>Kobe Bryant</td>\n",
       "      <td>Lionel Messi</td>\n",
       "      <td>57</td>\n",
       "      <td>Yes</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <td>77</td>\n",
       "      <td>Kobe Bryant</td>\n",
       "      <td>Cristiano Ronaldo</td>\n",
       "      <td>50</td>\n",
       "      <td>Yes</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <td>79</td>\n",
       "      <td>Kobe Bryant</td>\n",
       "      <td>Kylian Mbappé</td>\n",
       "      <td>39</td>\n",
       "      <td>No</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <td>19</td>\n",
       "      <td>Odesza</td>\n",
       "      <td>Kobe Bryant</td>\n",
       "      <td>20</td>\n",
       "      <td>No</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <td>80</td>\n",
       "      <td>Kobe Bryant</td>\n",
       "      <td>Chemistry</td>\n",
       "      <td>18</td>\n",
       "      <td>No</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <td>5</td>\n",
       "      <td>Kygo</td>\n",
       "      <td>Kobe Bryant</td>\n",
       "      <td>17</td>\n",
       "      <td>No</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <td>65</td>\n",
       "      <td>Cascada</td>\n",
       "      <td>Kobe Bryant</td>\n",
       "      <td>14</td>\n",
       "      <td>No</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <td>81</td>\n",
       "      <td>Kobe Bryant</td>\n",
       "      <td>Biology</td>\n",
       "      <td>14</td>\n",
       "      <td>No</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <td>32</td>\n",
       "      <td>Armin van Buuren</td>\n",
       "      <td>Kobe Bryant</td>\n",
       "      <td>13</td>\n",
       "      <td>No</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <td>44</td>\n",
       "      <td>Ilan Bluestone</td>\n",
       "      <td>Kobe Bryant</td>\n",
       "      <td>13</td>\n",
       "      <td>No</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <td>82</td>\n",
       "      <td>Kobe Bryant</td>\n",
       "      <td>Physics</td>\n",
       "      <td>11</td>\n",
       "      <td>No</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <td>55</td>\n",
       "      <td>Grum</td>\n",
       "      <td>Kobe Bryant</td>\n",
       "      <td>9</td>\n",
       "      <td>No</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <td>83</td>\n",
       "      <td>Kobe Bryant</td>\n",
       "      <td>Periodic table</td>\n",
       "      <td>8</td>\n",
       "      <td>No</td>\n",
       "    </tr>\n",
       "  </tbody>\n",
       "</table>\n",
       "</div>"
      ],
      "text/plain": [
       "             Topic 1            Topic 2  Probability Similar?\n",
       "75       Kobe Bryant       LeBron James          100      Yes\n",
       "76       Kobe Bryant     Michael Jordan          100      Yes\n",
       "78       Kobe Bryant       Lionel Messi           57      Yes\n",
       "77       Kobe Bryant  Cristiano Ronaldo           50      Yes\n",
       "79       Kobe Bryant      Kylian Mbappé           39       No\n",
       "19            Odesza        Kobe Bryant           20       No\n",
       "80       Kobe Bryant          Chemistry           18       No\n",
       "5               Kygo        Kobe Bryant           17       No\n",
       "65           Cascada        Kobe Bryant           14       No\n",
       "81       Kobe Bryant            Biology           14       No\n",
       "32  Armin van Buuren        Kobe Bryant           13       No\n",
       "44    Ilan Bluestone        Kobe Bryant           13       No\n",
       "82       Kobe Bryant            Physics           11       No\n",
       "55              Grum        Kobe Bryant            9       No\n",
       "83       Kobe Bryant     Periodic table            8       No"
      ]
     },
     "execution_count": 11,
     "metadata": {},
     "output_type": "execute_result"
    }
   ],
   "source": [
    "see_topic(test_df, 6)"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 12,
   "metadata": {},
   "outputs": [
    {
     "data": {
      "text/html": [
       "<div>\n",
       "<style scoped>\n",
       "    .dataframe tbody tr th:only-of-type {\n",
       "        vertical-align: middle;\n",
       "    }\n",
       "\n",
       "    .dataframe tbody tr th {\n",
       "        vertical-align: top;\n",
       "    }\n",
       "\n",
       "    .dataframe thead th {\n",
       "        text-align: right;\n",
       "    }\n",
       "</style>\n",
       "<table border=\"1\" class=\"dataframe\">\n",
       "  <thead>\n",
       "    <tr style=\"text-align: right;\">\n",
       "      <th></th>\n",
       "      <th>Topic 1</th>\n",
       "      <th>Topic 2</th>\n",
       "      <th>Probability</th>\n",
       "      <th>Similar?</th>\n",
       "    </tr>\n",
       "  </thead>\n",
       "  <tbody>\n",
       "    <tr>\n",
       "      <td>75</td>\n",
       "      <td>Kobe Bryant</td>\n",
       "      <td>LeBron James</td>\n",
       "      <td>100</td>\n",
       "      <td>Yes</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <td>84</td>\n",
       "      <td>LeBron James</td>\n",
       "      <td>Michael Jordan</td>\n",
       "      <td>100</td>\n",
       "      <td>Yes</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <td>86</td>\n",
       "      <td>LeBron James</td>\n",
       "      <td>Lionel Messi</td>\n",
       "      <td>27</td>\n",
       "      <td>No</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <td>85</td>\n",
       "      <td>LeBron James</td>\n",
       "      <td>Cristiano Ronaldo</td>\n",
       "      <td>24</td>\n",
       "      <td>No</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <td>87</td>\n",
       "      <td>LeBron James</td>\n",
       "      <td>Kylian Mbappé</td>\n",
       "      <td>20</td>\n",
       "      <td>No</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <td>88</td>\n",
       "      <td>LeBron James</td>\n",
       "      <td>Chemistry</td>\n",
       "      <td>13</td>\n",
       "      <td>No</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <td>91</td>\n",
       "      <td>LeBron James</td>\n",
       "      <td>Periodic table</td>\n",
       "      <td>11</td>\n",
       "      <td>No</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <td>20</td>\n",
       "      <td>Odesza</td>\n",
       "      <td>LeBron James</td>\n",
       "      <td>8</td>\n",
       "      <td>No</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <td>89</td>\n",
       "      <td>LeBron James</td>\n",
       "      <td>Biology</td>\n",
       "      <td>7</td>\n",
       "      <td>No</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <td>33</td>\n",
       "      <td>Armin van Buuren</td>\n",
       "      <td>LeBron James</td>\n",
       "      <td>5</td>\n",
       "      <td>No</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <td>45</td>\n",
       "      <td>Ilan Bluestone</td>\n",
       "      <td>LeBron James</td>\n",
       "      <td>5</td>\n",
       "      <td>No</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <td>6</td>\n",
       "      <td>Kygo</td>\n",
       "      <td>LeBron James</td>\n",
       "      <td>4</td>\n",
       "      <td>No</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <td>56</td>\n",
       "      <td>Grum</td>\n",
       "      <td>LeBron James</td>\n",
       "      <td>2</td>\n",
       "      <td>No</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <td>90</td>\n",
       "      <td>LeBron James</td>\n",
       "      <td>Physics</td>\n",
       "      <td>1</td>\n",
       "      <td>No</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <td>66</td>\n",
       "      <td>Cascada</td>\n",
       "      <td>LeBron James</td>\n",
       "      <td>0</td>\n",
       "      <td>No</td>\n",
       "    </tr>\n",
       "  </tbody>\n",
       "</table>\n",
       "</div>"
      ],
      "text/plain": [
       "             Topic 1            Topic 2  Probability Similar?\n",
       "75       Kobe Bryant       LeBron James          100      Yes\n",
       "84      LeBron James     Michael Jordan          100      Yes\n",
       "86      LeBron James       Lionel Messi           27       No\n",
       "85      LeBron James  Cristiano Ronaldo           24       No\n",
       "87      LeBron James      Kylian Mbappé           20       No\n",
       "88      LeBron James          Chemistry           13       No\n",
       "91      LeBron James     Periodic table           11       No\n",
       "20            Odesza       LeBron James            8       No\n",
       "89      LeBron James            Biology            7       No\n",
       "33  Armin van Buuren       LeBron James            5       No\n",
       "45    Ilan Bluestone       LeBron James            5       No\n",
       "6               Kygo       LeBron James            4       No\n",
       "56              Grum       LeBron James            2       No\n",
       "90      LeBron James            Physics            1       No\n",
       "66           Cascada       LeBron James            0       No"
      ]
     },
     "execution_count": 12,
     "metadata": {},
     "output_type": "execute_result"
    }
   ],
   "source": [
    "see_topic(test_df, 7)"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 13,
   "metadata": {},
   "outputs": [
    {
     "data": {
      "text/html": [
       "<div>\n",
       "<style scoped>\n",
       "    .dataframe tbody tr th:only-of-type {\n",
       "        vertical-align: middle;\n",
       "    }\n",
       "\n",
       "    .dataframe tbody tr th {\n",
       "        vertical-align: top;\n",
       "    }\n",
       "\n",
       "    .dataframe thead th {\n",
       "        text-align: right;\n",
       "    }\n",
       "</style>\n",
       "<table border=\"1\" class=\"dataframe\">\n",
       "  <thead>\n",
       "    <tr style=\"text-align: right;\">\n",
       "      <th></th>\n",
       "      <th>Topic 1</th>\n",
       "      <th>Topic 2</th>\n",
       "      <th>Probability</th>\n",
       "      <th>Similar?</th>\n",
       "    </tr>\n",
       "  </thead>\n",
       "  <tbody>\n",
       "    <tr>\n",
       "      <td>76</td>\n",
       "      <td>Kobe Bryant</td>\n",
       "      <td>Michael Jordan</td>\n",
       "      <td>100</td>\n",
       "      <td>Yes</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <td>84</td>\n",
       "      <td>LeBron James</td>\n",
       "      <td>Michael Jordan</td>\n",
       "      <td>100</td>\n",
       "      <td>Yes</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <td>93</td>\n",
       "      <td>Michael Jordan</td>\n",
       "      <td>Lionel Messi</td>\n",
       "      <td>40</td>\n",
       "      <td>No</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <td>92</td>\n",
       "      <td>Michael Jordan</td>\n",
       "      <td>Cristiano Ronaldo</td>\n",
       "      <td>37</td>\n",
       "      <td>No</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <td>94</td>\n",
       "      <td>Michael Jordan</td>\n",
       "      <td>Kylian Mbappé</td>\n",
       "      <td>29</td>\n",
       "      <td>No</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <td>98</td>\n",
       "      <td>Michael Jordan</td>\n",
       "      <td>Periodic table</td>\n",
       "      <td>17</td>\n",
       "      <td>No</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <td>95</td>\n",
       "      <td>Michael Jordan</td>\n",
       "      <td>Chemistry</td>\n",
       "      <td>11</td>\n",
       "      <td>No</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <td>21</td>\n",
       "      <td>Odesza</td>\n",
       "      <td>Michael Jordan</td>\n",
       "      <td>7</td>\n",
       "      <td>No</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <td>7</td>\n",
       "      <td>Kygo</td>\n",
       "      <td>Michael Jordan</td>\n",
       "      <td>4</td>\n",
       "      <td>No</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <td>96</td>\n",
       "      <td>Michael Jordan</td>\n",
       "      <td>Biology</td>\n",
       "      <td>4</td>\n",
       "      <td>No</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <td>46</td>\n",
       "      <td>Ilan Bluestone</td>\n",
       "      <td>Michael Jordan</td>\n",
       "      <td>3</td>\n",
       "      <td>No</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <td>34</td>\n",
       "      <td>Armin van Buuren</td>\n",
       "      <td>Michael Jordan</td>\n",
       "      <td>1</td>\n",
       "      <td>No</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <td>57</td>\n",
       "      <td>Grum</td>\n",
       "      <td>Michael Jordan</td>\n",
       "      <td>0</td>\n",
       "      <td>No</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <td>67</td>\n",
       "      <td>Cascada</td>\n",
       "      <td>Michael Jordan</td>\n",
       "      <td>0</td>\n",
       "      <td>No</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <td>97</td>\n",
       "      <td>Michael Jordan</td>\n",
       "      <td>Physics</td>\n",
       "      <td>0</td>\n",
       "      <td>No</td>\n",
       "    </tr>\n",
       "  </tbody>\n",
       "</table>\n",
       "</div>"
      ],
      "text/plain": [
       "             Topic 1            Topic 2  Probability Similar?\n",
       "76       Kobe Bryant     Michael Jordan          100      Yes\n",
       "84      LeBron James     Michael Jordan          100      Yes\n",
       "93    Michael Jordan       Lionel Messi           40       No\n",
       "92    Michael Jordan  Cristiano Ronaldo           37       No\n",
       "94    Michael Jordan      Kylian Mbappé           29       No\n",
       "98    Michael Jordan     Periodic table           17       No\n",
       "95    Michael Jordan          Chemistry           11       No\n",
       "21            Odesza     Michael Jordan            7       No\n",
       "7               Kygo     Michael Jordan            4       No\n",
       "96    Michael Jordan            Biology            4       No\n",
       "46    Ilan Bluestone     Michael Jordan            3       No\n",
       "34  Armin van Buuren     Michael Jordan            1       No\n",
       "57              Grum     Michael Jordan            0       No\n",
       "67           Cascada     Michael Jordan            0       No\n",
       "97    Michael Jordan            Physics            0       No"
      ]
     },
     "execution_count": 13,
     "metadata": {},
     "output_type": "execute_result"
    }
   ],
   "source": [
    "see_topic(test_df, 8)"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 14,
   "metadata": {},
   "outputs": [
    {
     "data": {
      "text/html": [
       "<div>\n",
       "<style scoped>\n",
       "    .dataframe tbody tr th:only-of-type {\n",
       "        vertical-align: middle;\n",
       "    }\n",
       "\n",
       "    .dataframe tbody tr th {\n",
       "        vertical-align: top;\n",
       "    }\n",
       "\n",
       "    .dataframe thead th {\n",
       "        text-align: right;\n",
       "    }\n",
       "</style>\n",
       "<table border=\"1\" class=\"dataframe\">\n",
       "  <thead>\n",
       "    <tr style=\"text-align: right;\">\n",
       "      <th></th>\n",
       "      <th>Topic 1</th>\n",
       "      <th>Topic 2</th>\n",
       "      <th>Probability</th>\n",
       "      <th>Similar?</th>\n",
       "    </tr>\n",
       "  </thead>\n",
       "  <tbody>\n",
       "    <tr>\n",
       "      <td>99</td>\n",
       "      <td>Cristiano Ronaldo</td>\n",
       "      <td>Lionel Messi</td>\n",
       "      <td>100</td>\n",
       "      <td>Yes</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <td>100</td>\n",
       "      <td>Cristiano Ronaldo</td>\n",
       "      <td>Kylian Mbappé</td>\n",
       "      <td>100</td>\n",
       "      <td>Yes</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <td>77</td>\n",
       "      <td>Kobe Bryant</td>\n",
       "      <td>Cristiano Ronaldo</td>\n",
       "      <td>50</td>\n",
       "      <td>Yes</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <td>92</td>\n",
       "      <td>Michael Jordan</td>\n",
       "      <td>Cristiano Ronaldo</td>\n",
       "      <td>37</td>\n",
       "      <td>No</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <td>85</td>\n",
       "      <td>LeBron James</td>\n",
       "      <td>Cristiano Ronaldo</td>\n",
       "      <td>24</td>\n",
       "      <td>No</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <td>104</td>\n",
       "      <td>Cristiano Ronaldo</td>\n",
       "      <td>Periodic table</td>\n",
       "      <td>23</td>\n",
       "      <td>No</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <td>101</td>\n",
       "      <td>Cristiano Ronaldo</td>\n",
       "      <td>Chemistry</td>\n",
       "      <td>16</td>\n",
       "      <td>No</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <td>68</td>\n",
       "      <td>Cascada</td>\n",
       "      <td>Cristiano Ronaldo</td>\n",
       "      <td>13</td>\n",
       "      <td>No</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <td>103</td>\n",
       "      <td>Cristiano Ronaldo</td>\n",
       "      <td>Physics</td>\n",
       "      <td>12</td>\n",
       "      <td>No</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <td>35</td>\n",
       "      <td>Armin van Buuren</td>\n",
       "      <td>Cristiano Ronaldo</td>\n",
       "      <td>11</td>\n",
       "      <td>No</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <td>22</td>\n",
       "      <td>Odesza</td>\n",
       "      <td>Cristiano Ronaldo</td>\n",
       "      <td>10</td>\n",
       "      <td>No</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <td>47</td>\n",
       "      <td>Ilan Bluestone</td>\n",
       "      <td>Cristiano Ronaldo</td>\n",
       "      <td>9</td>\n",
       "      <td>No</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <td>58</td>\n",
       "      <td>Grum</td>\n",
       "      <td>Cristiano Ronaldo</td>\n",
       "      <td>8</td>\n",
       "      <td>No</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <td>102</td>\n",
       "      <td>Cristiano Ronaldo</td>\n",
       "      <td>Biology</td>\n",
       "      <td>8</td>\n",
       "      <td>No</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <td>8</td>\n",
       "      <td>Kygo</td>\n",
       "      <td>Cristiano Ronaldo</td>\n",
       "      <td>7</td>\n",
       "      <td>No</td>\n",
       "    </tr>\n",
       "  </tbody>\n",
       "</table>\n",
       "</div>"
      ],
      "text/plain": [
       "               Topic 1            Topic 2  Probability Similar?\n",
       "99   Cristiano Ronaldo       Lionel Messi          100      Yes\n",
       "100  Cristiano Ronaldo      Kylian Mbappé          100      Yes\n",
       "77         Kobe Bryant  Cristiano Ronaldo           50      Yes\n",
       "92      Michael Jordan  Cristiano Ronaldo           37       No\n",
       "85        LeBron James  Cristiano Ronaldo           24       No\n",
       "104  Cristiano Ronaldo     Periodic table           23       No\n",
       "101  Cristiano Ronaldo          Chemistry           16       No\n",
       "68             Cascada  Cristiano Ronaldo           13       No\n",
       "103  Cristiano Ronaldo            Physics           12       No\n",
       "35    Armin van Buuren  Cristiano Ronaldo           11       No\n",
       "22              Odesza  Cristiano Ronaldo           10       No\n",
       "47      Ilan Bluestone  Cristiano Ronaldo            9       No\n",
       "58                Grum  Cristiano Ronaldo            8       No\n",
       "102  Cristiano Ronaldo            Biology            8       No\n",
       "8                 Kygo  Cristiano Ronaldo            7       No"
      ]
     },
     "execution_count": 14,
     "metadata": {},
     "output_type": "execute_result"
    }
   ],
   "source": [
    "see_topic(test_df, 9)"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 15,
   "metadata": {},
   "outputs": [
    {
     "data": {
      "text/html": [
       "<div>\n",
       "<style scoped>\n",
       "    .dataframe tbody tr th:only-of-type {\n",
       "        vertical-align: middle;\n",
       "    }\n",
       "\n",
       "    .dataframe tbody tr th {\n",
       "        vertical-align: top;\n",
       "    }\n",
       "\n",
       "    .dataframe thead th {\n",
       "        text-align: right;\n",
       "    }\n",
       "</style>\n",
       "<table border=\"1\" class=\"dataframe\">\n",
       "  <thead>\n",
       "    <tr style=\"text-align: right;\">\n",
       "      <th></th>\n",
       "      <th>Topic 1</th>\n",
       "      <th>Topic 2</th>\n",
       "      <th>Probability</th>\n",
       "      <th>Similar?</th>\n",
       "    </tr>\n",
       "  </thead>\n",
       "  <tbody>\n",
       "    <tr>\n",
       "      <td>99</td>\n",
       "      <td>Cristiano Ronaldo</td>\n",
       "      <td>Lionel Messi</td>\n",
       "      <td>100</td>\n",
       "      <td>Yes</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <td>105</td>\n",
       "      <td>Lionel Messi</td>\n",
       "      <td>Kylian Mbappé</td>\n",
       "      <td>92</td>\n",
       "      <td>Yes</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <td>78</td>\n",
       "      <td>Kobe Bryant</td>\n",
       "      <td>Lionel Messi</td>\n",
       "      <td>57</td>\n",
       "      <td>Yes</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <td>93</td>\n",
       "      <td>Michael Jordan</td>\n",
       "      <td>Lionel Messi</td>\n",
       "      <td>40</td>\n",
       "      <td>No</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <td>86</td>\n",
       "      <td>LeBron James</td>\n",
       "      <td>Lionel Messi</td>\n",
       "      <td>27</td>\n",
       "      <td>No</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <td>109</td>\n",
       "      <td>Lionel Messi</td>\n",
       "      <td>Periodic table</td>\n",
       "      <td>22</td>\n",
       "      <td>No</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <td>106</td>\n",
       "      <td>Lionel Messi</td>\n",
       "      <td>Chemistry</td>\n",
       "      <td>17</td>\n",
       "      <td>No</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <td>23</td>\n",
       "      <td>Odesza</td>\n",
       "      <td>Lionel Messi</td>\n",
       "      <td>13</td>\n",
       "      <td>No</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <td>36</td>\n",
       "      <td>Armin van Buuren</td>\n",
       "      <td>Lionel Messi</td>\n",
       "      <td>12</td>\n",
       "      <td>No</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <td>108</td>\n",
       "      <td>Lionel Messi</td>\n",
       "      <td>Physics</td>\n",
       "      <td>12</td>\n",
       "      <td>No</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <td>69</td>\n",
       "      <td>Cascada</td>\n",
       "      <td>Lionel Messi</td>\n",
       "      <td>11</td>\n",
       "      <td>No</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <td>48</td>\n",
       "      <td>Ilan Bluestone</td>\n",
       "      <td>Lionel Messi</td>\n",
       "      <td>10</td>\n",
       "      <td>No</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <td>59</td>\n",
       "      <td>Grum</td>\n",
       "      <td>Lionel Messi</td>\n",
       "      <td>9</td>\n",
       "      <td>No</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <td>107</td>\n",
       "      <td>Lionel Messi</td>\n",
       "      <td>Biology</td>\n",
       "      <td>9</td>\n",
       "      <td>No</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <td>9</td>\n",
       "      <td>Kygo</td>\n",
       "      <td>Lionel Messi</td>\n",
       "      <td>7</td>\n",
       "      <td>No</td>\n",
       "    </tr>\n",
       "  </tbody>\n",
       "</table>\n",
       "</div>"
      ],
      "text/plain": [
       "               Topic 1         Topic 2  Probability Similar?\n",
       "99   Cristiano Ronaldo    Lionel Messi          100      Yes\n",
       "105       Lionel Messi   Kylian Mbappé           92      Yes\n",
       "78         Kobe Bryant    Lionel Messi           57      Yes\n",
       "93      Michael Jordan    Lionel Messi           40       No\n",
       "86        LeBron James    Lionel Messi           27       No\n",
       "109       Lionel Messi  Periodic table           22       No\n",
       "106       Lionel Messi       Chemistry           17       No\n",
       "23              Odesza    Lionel Messi           13       No\n",
       "36    Armin van Buuren    Lionel Messi           12       No\n",
       "108       Lionel Messi         Physics           12       No\n",
       "69             Cascada    Lionel Messi           11       No\n",
       "48      Ilan Bluestone    Lionel Messi           10       No\n",
       "59                Grum    Lionel Messi            9       No\n",
       "107       Lionel Messi         Biology            9       No\n",
       "9                 Kygo    Lionel Messi            7       No"
      ]
     },
     "execution_count": 15,
     "metadata": {},
     "output_type": "execute_result"
    }
   ],
   "source": [
    "see_topic(test_df, 10)"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 16,
   "metadata": {},
   "outputs": [
    {
     "data": {
      "text/html": [
       "<div>\n",
       "<style scoped>\n",
       "    .dataframe tbody tr th:only-of-type {\n",
       "        vertical-align: middle;\n",
       "    }\n",
       "\n",
       "    .dataframe tbody tr th {\n",
       "        vertical-align: top;\n",
       "    }\n",
       "\n",
       "    .dataframe thead th {\n",
       "        text-align: right;\n",
       "    }\n",
       "</style>\n",
       "<table border=\"1\" class=\"dataframe\">\n",
       "  <thead>\n",
       "    <tr style=\"text-align: right;\">\n",
       "      <th></th>\n",
       "      <th>Topic 1</th>\n",
       "      <th>Topic 2</th>\n",
       "      <th>Probability</th>\n",
       "      <th>Similar?</th>\n",
       "    </tr>\n",
       "  </thead>\n",
       "  <tbody>\n",
       "    <tr>\n",
       "      <td>100</td>\n",
       "      <td>Cristiano Ronaldo</td>\n",
       "      <td>Kylian Mbappé</td>\n",
       "      <td>100</td>\n",
       "      <td>Yes</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <td>105</td>\n",
       "      <td>Lionel Messi</td>\n",
       "      <td>Kylian Mbappé</td>\n",
       "      <td>92</td>\n",
       "      <td>Yes</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <td>79</td>\n",
       "      <td>Kobe Bryant</td>\n",
       "      <td>Kylian Mbappé</td>\n",
       "      <td>39</td>\n",
       "      <td>No</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <td>94</td>\n",
       "      <td>Michael Jordan</td>\n",
       "      <td>Kylian Mbappé</td>\n",
       "      <td>29</td>\n",
       "      <td>No</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <td>87</td>\n",
       "      <td>LeBron James</td>\n",
       "      <td>Kylian Mbappé</td>\n",
       "      <td>20</td>\n",
       "      <td>No</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <td>113</td>\n",
       "      <td>Kylian Mbappé</td>\n",
       "      <td>Periodic table</td>\n",
       "      <td>15</td>\n",
       "      <td>No</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <td>110</td>\n",
       "      <td>Kylian Mbappé</td>\n",
       "      <td>Chemistry</td>\n",
       "      <td>14</td>\n",
       "      <td>No</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <td>112</td>\n",
       "      <td>Kylian Mbappé</td>\n",
       "      <td>Physics</td>\n",
       "      <td>12</td>\n",
       "      <td>No</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <td>24</td>\n",
       "      <td>Odesza</td>\n",
       "      <td>Kylian Mbappé</td>\n",
       "      <td>11</td>\n",
       "      <td>No</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <td>70</td>\n",
       "      <td>Cascada</td>\n",
       "      <td>Kylian Mbappé</td>\n",
       "      <td>10</td>\n",
       "      <td>No</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <td>37</td>\n",
       "      <td>Armin van Buuren</td>\n",
       "      <td>Kylian Mbappé</td>\n",
       "      <td>8</td>\n",
       "      <td>No</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <td>111</td>\n",
       "      <td>Kylian Mbappé</td>\n",
       "      <td>Biology</td>\n",
       "      <td>8</td>\n",
       "      <td>No</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <td>60</td>\n",
       "      <td>Grum</td>\n",
       "      <td>Kylian Mbappé</td>\n",
       "      <td>6</td>\n",
       "      <td>No</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <td>49</td>\n",
       "      <td>Ilan Bluestone</td>\n",
       "      <td>Kylian Mbappé</td>\n",
       "      <td>5</td>\n",
       "      <td>No</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <td>10</td>\n",
       "      <td>Kygo</td>\n",
       "      <td>Kylian Mbappé</td>\n",
       "      <td>1</td>\n",
       "      <td>No</td>\n",
       "    </tr>\n",
       "  </tbody>\n",
       "</table>\n",
       "</div>"
      ],
      "text/plain": [
       "               Topic 1         Topic 2  Probability Similar?\n",
       "100  Cristiano Ronaldo   Kylian Mbappé          100      Yes\n",
       "105       Lionel Messi   Kylian Mbappé           92      Yes\n",
       "79         Kobe Bryant   Kylian Mbappé           39       No\n",
       "94      Michael Jordan   Kylian Mbappé           29       No\n",
       "87        LeBron James   Kylian Mbappé           20       No\n",
       "113      Kylian Mbappé  Periodic table           15       No\n",
       "110      Kylian Mbappé       Chemistry           14       No\n",
       "112      Kylian Mbappé         Physics           12       No\n",
       "24              Odesza   Kylian Mbappé           11       No\n",
       "70             Cascada   Kylian Mbappé           10       No\n",
       "37    Armin van Buuren   Kylian Mbappé            8       No\n",
       "111      Kylian Mbappé         Biology            8       No\n",
       "60                Grum   Kylian Mbappé            6       No\n",
       "49      Ilan Bluestone   Kylian Mbappé            5       No\n",
       "10                Kygo   Kylian Mbappé            1       No"
      ]
     },
     "execution_count": 16,
     "metadata": {},
     "output_type": "execute_result"
    }
   ],
   "source": [
    "see_topic(test_df, 11)"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 17,
   "metadata": {},
   "outputs": [
    {
     "data": {
      "text/html": [
       "<div>\n",
       "<style scoped>\n",
       "    .dataframe tbody tr th:only-of-type {\n",
       "        vertical-align: middle;\n",
       "    }\n",
       "\n",
       "    .dataframe tbody tr th {\n",
       "        vertical-align: top;\n",
       "    }\n",
       "\n",
       "    .dataframe thead th {\n",
       "        text-align: right;\n",
       "    }\n",
       "</style>\n",
       "<table border=\"1\" class=\"dataframe\">\n",
       "  <thead>\n",
       "    <tr style=\"text-align: right;\">\n",
       "      <th></th>\n",
       "      <th>Topic 1</th>\n",
       "      <th>Topic 2</th>\n",
       "      <th>Probability</th>\n",
       "      <th>Similar?</th>\n",
       "    </tr>\n",
       "  </thead>\n",
       "  <tbody>\n",
       "    <tr>\n",
       "      <td>116</td>\n",
       "      <td>Chemistry</td>\n",
       "      <td>Periodic table</td>\n",
       "      <td>65</td>\n",
       "      <td>Yes</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <td>114</td>\n",
       "      <td>Chemistry</td>\n",
       "      <td>Biology</td>\n",
       "      <td>53</td>\n",
       "      <td>Yes</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <td>115</td>\n",
       "      <td>Chemistry</td>\n",
       "      <td>Physics</td>\n",
       "      <td>51</td>\n",
       "      <td>Yes</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <td>61</td>\n",
       "      <td>Grum</td>\n",
       "      <td>Chemistry</td>\n",
       "      <td>24</td>\n",
       "      <td>No</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <td>38</td>\n",
       "      <td>Armin van Buuren</td>\n",
       "      <td>Chemistry</td>\n",
       "      <td>22</td>\n",
       "      <td>No</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <td>50</td>\n",
       "      <td>Ilan Bluestone</td>\n",
       "      <td>Chemistry</td>\n",
       "      <td>20</td>\n",
       "      <td>No</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <td>25</td>\n",
       "      <td>Odesza</td>\n",
       "      <td>Chemistry</td>\n",
       "      <td>18</td>\n",
       "      <td>No</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <td>80</td>\n",
       "      <td>Kobe Bryant</td>\n",
       "      <td>Chemistry</td>\n",
       "      <td>18</td>\n",
       "      <td>No</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <td>106</td>\n",
       "      <td>Lionel Messi</td>\n",
       "      <td>Chemistry</td>\n",
       "      <td>17</td>\n",
       "      <td>No</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <td>101</td>\n",
       "      <td>Cristiano Ronaldo</td>\n",
       "      <td>Chemistry</td>\n",
       "      <td>16</td>\n",
       "      <td>No</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <td>11</td>\n",
       "      <td>Kygo</td>\n",
       "      <td>Chemistry</td>\n",
       "      <td>14</td>\n",
       "      <td>No</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <td>110</td>\n",
       "      <td>Kylian Mbappé</td>\n",
       "      <td>Chemistry</td>\n",
       "      <td>14</td>\n",
       "      <td>No</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <td>71</td>\n",
       "      <td>Cascada</td>\n",
       "      <td>Chemistry</td>\n",
       "      <td>13</td>\n",
       "      <td>No</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <td>88</td>\n",
       "      <td>LeBron James</td>\n",
       "      <td>Chemistry</td>\n",
       "      <td>13</td>\n",
       "      <td>No</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <td>95</td>\n",
       "      <td>Michael Jordan</td>\n",
       "      <td>Chemistry</td>\n",
       "      <td>11</td>\n",
       "      <td>No</td>\n",
       "    </tr>\n",
       "  </tbody>\n",
       "</table>\n",
       "</div>"
      ],
      "text/plain": [
       "               Topic 1         Topic 2  Probability Similar?\n",
       "116          Chemistry  Periodic table           65      Yes\n",
       "114          Chemistry         Biology           53      Yes\n",
       "115          Chemistry         Physics           51      Yes\n",
       "61                Grum       Chemistry           24       No\n",
       "38    Armin van Buuren       Chemistry           22       No\n",
       "50      Ilan Bluestone       Chemistry           20       No\n",
       "25              Odesza       Chemistry           18       No\n",
       "80         Kobe Bryant       Chemistry           18       No\n",
       "106       Lionel Messi       Chemistry           17       No\n",
       "101  Cristiano Ronaldo       Chemistry           16       No\n",
       "11                Kygo       Chemistry           14       No\n",
       "110      Kylian Mbappé       Chemistry           14       No\n",
       "71             Cascada       Chemistry           13       No\n",
       "88        LeBron James       Chemistry           13       No\n",
       "95      Michael Jordan       Chemistry           11       No"
      ]
     },
     "execution_count": 17,
     "metadata": {},
     "output_type": "execute_result"
    }
   ],
   "source": [
    "see_topic(test_df, 12)"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 18,
   "metadata": {},
   "outputs": [
    {
     "data": {
      "text/html": [
       "<div>\n",
       "<style scoped>\n",
       "    .dataframe tbody tr th:only-of-type {\n",
       "        vertical-align: middle;\n",
       "    }\n",
       "\n",
       "    .dataframe tbody tr th {\n",
       "        vertical-align: top;\n",
       "    }\n",
       "\n",
       "    .dataframe thead th {\n",
       "        text-align: right;\n",
       "    }\n",
       "</style>\n",
       "<table border=\"1\" class=\"dataframe\">\n",
       "  <thead>\n",
       "    <tr style=\"text-align: right;\">\n",
       "      <th></th>\n",
       "      <th>Topic 1</th>\n",
       "      <th>Topic 2</th>\n",
       "      <th>Probability</th>\n",
       "      <th>Similar?</th>\n",
       "    </tr>\n",
       "  </thead>\n",
       "  <tbody>\n",
       "    <tr>\n",
       "      <td>117</td>\n",
       "      <td>Biology</td>\n",
       "      <td>Physics</td>\n",
       "      <td>55</td>\n",
       "      <td>Yes</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <td>114</td>\n",
       "      <td>Chemistry</td>\n",
       "      <td>Biology</td>\n",
       "      <td>53</td>\n",
       "      <td>Yes</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <td>39</td>\n",
       "      <td>Armin van Buuren</td>\n",
       "      <td>Biology</td>\n",
       "      <td>15</td>\n",
       "      <td>No</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <td>81</td>\n",
       "      <td>Kobe Bryant</td>\n",
       "      <td>Biology</td>\n",
       "      <td>14</td>\n",
       "      <td>No</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <td>62</td>\n",
       "      <td>Grum</td>\n",
       "      <td>Biology</td>\n",
       "      <td>13</td>\n",
       "      <td>No</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <td>118</td>\n",
       "      <td>Biology</td>\n",
       "      <td>Periodic table</td>\n",
       "      <td>13</td>\n",
       "      <td>No</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <td>26</td>\n",
       "      <td>Odesza</td>\n",
       "      <td>Biology</td>\n",
       "      <td>12</td>\n",
       "      <td>No</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <td>51</td>\n",
       "      <td>Ilan Bluestone</td>\n",
       "      <td>Biology</td>\n",
       "      <td>10</td>\n",
       "      <td>No</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <td>72</td>\n",
       "      <td>Cascada</td>\n",
       "      <td>Biology</td>\n",
       "      <td>10</td>\n",
       "      <td>No</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <td>107</td>\n",
       "      <td>Lionel Messi</td>\n",
       "      <td>Biology</td>\n",
       "      <td>9</td>\n",
       "      <td>No</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <td>102</td>\n",
       "      <td>Cristiano Ronaldo</td>\n",
       "      <td>Biology</td>\n",
       "      <td>8</td>\n",
       "      <td>No</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <td>111</td>\n",
       "      <td>Kylian Mbappé</td>\n",
       "      <td>Biology</td>\n",
       "      <td>8</td>\n",
       "      <td>No</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <td>89</td>\n",
       "      <td>LeBron James</td>\n",
       "      <td>Biology</td>\n",
       "      <td>7</td>\n",
       "      <td>No</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <td>12</td>\n",
       "      <td>Kygo</td>\n",
       "      <td>Biology</td>\n",
       "      <td>5</td>\n",
       "      <td>No</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <td>96</td>\n",
       "      <td>Michael Jordan</td>\n",
       "      <td>Biology</td>\n",
       "      <td>4</td>\n",
       "      <td>No</td>\n",
       "    </tr>\n",
       "  </tbody>\n",
       "</table>\n",
       "</div>"
      ],
      "text/plain": [
       "               Topic 1         Topic 2  Probability Similar?\n",
       "117            Biology         Physics           55      Yes\n",
       "114          Chemistry         Biology           53      Yes\n",
       "39    Armin van Buuren         Biology           15       No\n",
       "81         Kobe Bryant         Biology           14       No\n",
       "62                Grum         Biology           13       No\n",
       "118            Biology  Periodic table           13       No\n",
       "26              Odesza         Biology           12       No\n",
       "51      Ilan Bluestone         Biology           10       No\n",
       "72             Cascada         Biology           10       No\n",
       "107       Lionel Messi         Biology            9       No\n",
       "102  Cristiano Ronaldo         Biology            8       No\n",
       "111      Kylian Mbappé         Biology            8       No\n",
       "89        LeBron James         Biology            7       No\n",
       "12                Kygo         Biology            5       No\n",
       "96      Michael Jordan         Biology            4       No"
      ]
     },
     "execution_count": 18,
     "metadata": {},
     "output_type": "execute_result"
    }
   ],
   "source": [
    "see_topic(test_df, 13)"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 19,
   "metadata": {},
   "outputs": [
    {
     "data": {
      "text/html": [
       "<div>\n",
       "<style scoped>\n",
       "    .dataframe tbody tr th:only-of-type {\n",
       "        vertical-align: middle;\n",
       "    }\n",
       "\n",
       "    .dataframe tbody tr th {\n",
       "        vertical-align: top;\n",
       "    }\n",
       "\n",
       "    .dataframe thead th {\n",
       "        text-align: right;\n",
       "    }\n",
       "</style>\n",
       "<table border=\"1\" class=\"dataframe\">\n",
       "  <thead>\n",
       "    <tr style=\"text-align: right;\">\n",
       "      <th></th>\n",
       "      <th>Topic 1</th>\n",
       "      <th>Topic 2</th>\n",
       "      <th>Probability</th>\n",
       "      <th>Similar?</th>\n",
       "    </tr>\n",
       "  </thead>\n",
       "  <tbody>\n",
       "    <tr>\n",
       "      <td>117</td>\n",
       "      <td>Biology</td>\n",
       "      <td>Physics</td>\n",
       "      <td>55</td>\n",
       "      <td>Yes</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <td>115</td>\n",
       "      <td>Chemistry</td>\n",
       "      <td>Physics</td>\n",
       "      <td>51</td>\n",
       "      <td>Yes</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <td>73</td>\n",
       "      <td>Cascada</td>\n",
       "      <td>Physics</td>\n",
       "      <td>19</td>\n",
       "      <td>No</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <td>119</td>\n",
       "      <td>Physics</td>\n",
       "      <td>Periodic table</td>\n",
       "      <td>17</td>\n",
       "      <td>No</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <td>63</td>\n",
       "      <td>Grum</td>\n",
       "      <td>Physics</td>\n",
       "      <td>14</td>\n",
       "      <td>No</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <td>40</td>\n",
       "      <td>Armin van Buuren</td>\n",
       "      <td>Physics</td>\n",
       "      <td>13</td>\n",
       "      <td>No</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <td>103</td>\n",
       "      <td>Cristiano Ronaldo</td>\n",
       "      <td>Physics</td>\n",
       "      <td>12</td>\n",
       "      <td>No</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <td>108</td>\n",
       "      <td>Lionel Messi</td>\n",
       "      <td>Physics</td>\n",
       "      <td>12</td>\n",
       "      <td>No</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <td>112</td>\n",
       "      <td>Kylian Mbappé</td>\n",
       "      <td>Physics</td>\n",
       "      <td>12</td>\n",
       "      <td>No</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <td>82</td>\n",
       "      <td>Kobe Bryant</td>\n",
       "      <td>Physics</td>\n",
       "      <td>11</td>\n",
       "      <td>No</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <td>27</td>\n",
       "      <td>Odesza</td>\n",
       "      <td>Physics</td>\n",
       "      <td>8</td>\n",
       "      <td>No</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <td>52</td>\n",
       "      <td>Ilan Bluestone</td>\n",
       "      <td>Physics</td>\n",
       "      <td>6</td>\n",
       "      <td>No</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <td>13</td>\n",
       "      <td>Kygo</td>\n",
       "      <td>Physics</td>\n",
       "      <td>3</td>\n",
       "      <td>No</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <td>90</td>\n",
       "      <td>LeBron James</td>\n",
       "      <td>Physics</td>\n",
       "      <td>1</td>\n",
       "      <td>No</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <td>97</td>\n",
       "      <td>Michael Jordan</td>\n",
       "      <td>Physics</td>\n",
       "      <td>0</td>\n",
       "      <td>No</td>\n",
       "    </tr>\n",
       "  </tbody>\n",
       "</table>\n",
       "</div>"
      ],
      "text/plain": [
       "               Topic 1         Topic 2  Probability Similar?\n",
       "117            Biology         Physics           55      Yes\n",
       "115          Chemistry         Physics           51      Yes\n",
       "73             Cascada         Physics           19       No\n",
       "119            Physics  Periodic table           17       No\n",
       "63                Grum         Physics           14       No\n",
       "40    Armin van Buuren         Physics           13       No\n",
       "103  Cristiano Ronaldo         Physics           12       No\n",
       "108       Lionel Messi         Physics           12       No\n",
       "112      Kylian Mbappé         Physics           12       No\n",
       "82         Kobe Bryant         Physics           11       No\n",
       "27              Odesza         Physics            8       No\n",
       "52      Ilan Bluestone         Physics            6       No\n",
       "13                Kygo         Physics            3       No\n",
       "90        LeBron James         Physics            1       No\n",
       "97      Michael Jordan         Physics            0       No"
      ]
     },
     "execution_count": 19,
     "metadata": {},
     "output_type": "execute_result"
    }
   ],
   "source": [
    "see_topic(test_df, 14)"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 20,
   "metadata": {},
   "outputs": [
    {
     "data": {
      "text/html": [
       "<div>\n",
       "<style scoped>\n",
       "    .dataframe tbody tr th:only-of-type {\n",
       "        vertical-align: middle;\n",
       "    }\n",
       "\n",
       "    .dataframe tbody tr th {\n",
       "        vertical-align: top;\n",
       "    }\n",
       "\n",
       "    .dataframe thead th {\n",
       "        text-align: right;\n",
       "    }\n",
       "</style>\n",
       "<table border=\"1\" class=\"dataframe\">\n",
       "  <thead>\n",
       "    <tr style=\"text-align: right;\">\n",
       "      <th></th>\n",
       "      <th>Topic 1</th>\n",
       "      <th>Topic 2</th>\n",
       "      <th>Probability</th>\n",
       "      <th>Similar?</th>\n",
       "    </tr>\n",
       "  </thead>\n",
       "  <tbody>\n",
       "    <tr>\n",
       "      <td>116</td>\n",
       "      <td>Chemistry</td>\n",
       "      <td>Periodic table</td>\n",
       "      <td>65</td>\n",
       "      <td>Yes</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <td>104</td>\n",
       "      <td>Cristiano Ronaldo</td>\n",
       "      <td>Periodic table</td>\n",
       "      <td>23</td>\n",
       "      <td>No</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <td>109</td>\n",
       "      <td>Lionel Messi</td>\n",
       "      <td>Periodic table</td>\n",
       "      <td>22</td>\n",
       "      <td>No</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <td>98</td>\n",
       "      <td>Michael Jordan</td>\n",
       "      <td>Periodic table</td>\n",
       "      <td>17</td>\n",
       "      <td>No</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <td>119</td>\n",
       "      <td>Physics</td>\n",
       "      <td>Periodic table</td>\n",
       "      <td>17</td>\n",
       "      <td>No</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <td>113</td>\n",
       "      <td>Kylian Mbappé</td>\n",
       "      <td>Periodic table</td>\n",
       "      <td>15</td>\n",
       "      <td>No</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <td>118</td>\n",
       "      <td>Biology</td>\n",
       "      <td>Periodic table</td>\n",
       "      <td>13</td>\n",
       "      <td>No</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <td>64</td>\n",
       "      <td>Grum</td>\n",
       "      <td>Periodic table</td>\n",
       "      <td>12</td>\n",
       "      <td>No</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <td>91</td>\n",
       "      <td>LeBron James</td>\n",
       "      <td>Periodic table</td>\n",
       "      <td>11</td>\n",
       "      <td>No</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <td>41</td>\n",
       "      <td>Armin van Buuren</td>\n",
       "      <td>Periodic table</td>\n",
       "      <td>9</td>\n",
       "      <td>No</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <td>83</td>\n",
       "      <td>Kobe Bryant</td>\n",
       "      <td>Periodic table</td>\n",
       "      <td>8</td>\n",
       "      <td>No</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <td>28</td>\n",
       "      <td>Odesza</td>\n",
       "      <td>Periodic table</td>\n",
       "      <td>6</td>\n",
       "      <td>No</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <td>53</td>\n",
       "      <td>Ilan Bluestone</td>\n",
       "      <td>Periodic table</td>\n",
       "      <td>6</td>\n",
       "      <td>No</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <td>14</td>\n",
       "      <td>Kygo</td>\n",
       "      <td>Periodic table</td>\n",
       "      <td>4</td>\n",
       "      <td>No</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <td>74</td>\n",
       "      <td>Cascada</td>\n",
       "      <td>Periodic table</td>\n",
       "      <td>4</td>\n",
       "      <td>No</td>\n",
       "    </tr>\n",
       "  </tbody>\n",
       "</table>\n",
       "</div>"
      ],
      "text/plain": [
       "               Topic 1         Topic 2  Probability Similar?\n",
       "116          Chemistry  Periodic table           65      Yes\n",
       "104  Cristiano Ronaldo  Periodic table           23       No\n",
       "109       Lionel Messi  Periodic table           22       No\n",
       "98      Michael Jordan  Periodic table           17       No\n",
       "119            Physics  Periodic table           17       No\n",
       "113      Kylian Mbappé  Periodic table           15       No\n",
       "118            Biology  Periodic table           13       No\n",
       "64                Grum  Periodic table           12       No\n",
       "91        LeBron James  Periodic table           11       No\n",
       "41    Armin van Buuren  Periodic table            9       No\n",
       "83         Kobe Bryant  Periodic table            8       No\n",
       "28              Odesza  Periodic table            6       No\n",
       "53      Ilan Bluestone  Periodic table            6       No\n",
       "14                Kygo  Periodic table            4       No\n",
       "74             Cascada  Periodic table            4       No"
      ]
     },
     "execution_count": 20,
     "metadata": {},
     "output_type": "execute_result"
    }
   ],
   "source": [
    "see_topic(test_df, 15)"
   ]
  }
 ],
 "metadata": {
  "kernelspec": {
   "display_name": "Python 3",
   "language": "python",
   "name": "python3"
  },
  "language_info": {
   "codemirror_mode": {
    "name": "ipython",
    "version": 3
   },
   "file_extension": ".py",
   "mimetype": "text/x-python",
   "name": "python",
   "nbconvert_exporter": "python",
   "pygments_lexer": "ipython3",
   "version": "3.7.4"
  }
 },
 "nbformat": 4,
 "nbformat_minor": 2
}