teste.ipynb

{
 "cells": [
  {
   "cell_type": "code",
   "execution_count": 1,
   "metadata": {},
   "outputs": [
    {
     "data": {
      "text/plain": [
       "0     0.744186\n",
       "1     0.294085\n",
       "2     0.747748\n",
       "3     0.905660\n",
       "4     0.897436\n",
       "5     0.698718\n",
       "6     0.488235\n",
       "7     0.953757\n",
       "8     0.497151\n",
       "9     0.580645\n",
       "10    0.767516\n",
       "11    0.513514\n",
       "dtype: float64"
      ]
     },
     "execution_count": 1,
     "metadata": {},
     "output_type": "execute_result"
    }
   ],
   "source": [
    "import pandas as pd\n",
    "import numpy as np\n",
    "\n",
    "# Tabela de contingência para TF-IDF com K-Means\n",
    "data = {\n",
    "    'A': [0, 47, 0, 0, 70, 0, 0, 0, 10, 0, 5, 2],\n",
    "    'B': [1, 211, 1, 1, 0, 5, 4, 1, 38, 3, 3, 2],\n",
    "    'C': [0, 3, 0, 0, 0, 4, 0, 495, 4, 0, 0, 1],\n",
    "    'D': [1, 256, 0, 0, 0, 1, 1, 0, 16, 1, 0, 0],\n",
    "    'E': [1, 527, 1, 1, 0, 12, 62, 2, 21, 1, 0, 0],\n",
    "    'F': [32, 321, 1, 0, 4, 6, 2, 11, 112, 4, 10, 19],\n",
    "    'G': [0, 17, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],\n",
    "    'H': [2, 303, 21, 48, 3, 15, 5, 9, 42, 7, 52, 1],\n",
    "    'I': [0, 11, 0, 0, 0, 0, 1, 0, 2, 0, 0, 0],\n",
    "    'J': [0, 2, 0, 0, 0, 3, 12, 1, 94, 10, 0, 0],\n",
    "    'K': [5, 4, 4, 2, 1, 109, 83, 0, 349, 36, 3, 10],\n",
    "    'L': [1, 90, 83, 1, 0, 1, 0, 0, 14, 0, 241, 2]\n",
    "}\n",
    "\n",
    "# Convertendo para DataFrame\n",
    "df = pd.DataFrame(data)\n",
    "\n",
    "# Calculando a pureza de cada cluster\n",
    "def calculate_cluster_purity(cluster_data):\n",
    "    # Maior frequência de categoria no cluster\n",
    "    max_count = max(cluster_data)\n",
    "    # Soma total de documentos no cluster\n",
    "    total_count = sum(cluster_data)\n",
    "    # Pureza do cluster\n",
    "    return max_count / total_count if total_count > 0 else 0\n",
    "\n",
    "# Aplicando a função em cada linha (cluster)\n",
    "cluster_purity = df.apply(calculate_cluster_purity, axis=1)\n",
    "cluster_purity"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 2,
   "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>A</th>\n",
       "      <th>B</th>\n",
       "      <th>C</th>\n",
       "      <th>D</th>\n",
       "      <th>E</th>\n",
       "      <th>F</th>\n",
       "      <th>G</th>\n",
       "      <th>H</th>\n",
       "      <th>I</th>\n",
       "      <th>J</th>\n",
       "      <th>K</th>\n",
       "      <th>L</th>\n",
       "    </tr>\n",
       "  </thead>\n",
       "  <tbody>\n",
       "    <tr>\n",
       "      <th>0</th>\n",
       "      <td>0</td>\n",
       "      <td>1</td>\n",
       "      <td>0</td>\n",
       "      <td>1</td>\n",
       "      <td>1</td>\n",
       "      <td>32</td>\n",
       "      <td>0</td>\n",
       "      <td>2</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>5</td>\n",
       "      <td>1</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>1</th>\n",
       "      <td>47</td>\n",
       "      <td>211</td>\n",
       "      <td>3</td>\n",
       "      <td>256</td>\n",
       "      <td>527</td>\n",
       "      <td>321</td>\n",
       "      <td>17</td>\n",
       "      <td>303</td>\n",
       "      <td>11</td>\n",
       "      <td>2</td>\n",
       "      <td>4</td>\n",
       "      <td>90</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>2</th>\n",
       "      <td>0</td>\n",
       "      <td>1</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>1</td>\n",
       "      <td>1</td>\n",
       "      <td>0</td>\n",
       "      <td>21</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>4</td>\n",
       "      <td>83</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>3</th>\n",
       "      <td>0</td>\n",
       "      <td>1</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>1</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>48</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>2</td>\n",
       "      <td>1</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>4</th>\n",
       "      <td>70</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>4</td>\n",
       "      <td>0</td>\n",
       "      <td>3</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>1</td>\n",
       "      <td>0</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>5</th>\n",
       "      <td>0</td>\n",
       "      <td>5</td>\n",
       "      <td>4</td>\n",
       "      <td>1</td>\n",
       "      <td>12</td>\n",
       "      <td>6</td>\n",
       "      <td>0</td>\n",
       "      <td>15</td>\n",
       "      <td>0</td>\n",
       "      <td>3</td>\n",
       "      <td>109</td>\n",
       "      <td>1</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>6</th>\n",
       "      <td>0</td>\n",
       "      <td>4</td>\n",
       "      <td>0</td>\n",
       "      <td>1</td>\n",
       "      <td>62</td>\n",
       "      <td>2</td>\n",
       "      <td>0</td>\n",
       "      <td>5</td>\n",
       "      <td>1</td>\n",
       "      <td>12</td>\n",
       "      <td>83</td>\n",
       "      <td>0</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>7</th>\n",
       "      <td>0</td>\n",
       "      <td>1</td>\n",
       "      <td>495</td>\n",
       "      <td>0</td>\n",
       "      <td>2</td>\n",
       "      <td>11</td>\n",
       "      <td>0</td>\n",
       "      <td>9</td>\n",
       "      <td>0</td>\n",
       "      <td>1</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>8</th>\n",
       "      <td>10</td>\n",
       "      <td>38</td>\n",
       "      <td>4</td>\n",
       "      <td>16</td>\n",
       "      <td>21</td>\n",
       "      <td>112</td>\n",
       "      <td>0</td>\n",
       "      <td>42</td>\n",
       "      <td>2</td>\n",
       "      <td>94</td>\n",
       "      <td>349</td>\n",
       "      <td>14</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>9</th>\n",
       "      <td>0</td>\n",
       "      <td>3</td>\n",
       "      <td>0</td>\n",
       "      <td>1</td>\n",
       "      <td>1</td>\n",
       "      <td>4</td>\n",
       "      <td>0</td>\n",
       "      <td>7</td>\n",
       "      <td>0</td>\n",
       "      <td>10</td>\n",
       "      <td>36</td>\n",
       "      <td>0</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>10</th>\n",
       "      <td>5</td>\n",
       "      <td>3</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>10</td>\n",
       "      <td>0</td>\n",
       "      <td>52</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>3</td>\n",
       "      <td>241</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>11</th>\n",
       "      <td>2</td>\n",
       "      <td>2</td>\n",
       "      <td>1</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>19</td>\n",
       "      <td>0</td>\n",
       "      <td>1</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>10</td>\n",
       "      <td>2</td>\n",
       "    </tr>\n",
       "  </tbody>\n",
       "</table>\n",
       "</div>"
      ],
      "text/plain": [
       "     A    B    C    D    E    F   G    H   I   J    K    L\n",
       "0    0    1    0    1    1   32   0    2   0   0    5    1\n",
       "1   47  211    3  256  527  321  17  303  11   2    4   90\n",
       "2    0    1    0    0    1    1   0   21   0   0    4   83\n",
       "3    0    1    0    0    1    0   0   48   0   0    2    1\n",
       "4   70    0    0    0    0    4   0    3   0   0    1    0\n",
       "5    0    5    4    1   12    6   0   15   0   3  109    1\n",
       "6    0    4    0    1   62    2   0    5   1  12   83    0\n",
       "7    0    1  495    0    2   11   0    9   0   1    0    0\n",
       "8   10   38    4   16   21  112   0   42   2  94  349   14\n",
       "9    0    3    0    1    1    4   0    7   0  10   36    0\n",
       "10   5    3    0    0    0   10   0   52   0   0    3  241\n",
       "11   2    2    1    0    0   19   0    1   0   0   10    2"
      ]
     },
     "execution_count": 2,
     "metadata": {},
     "output_type": "execute_result"
    }
   ],
   "source": [
    "df"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": null,
   "metadata": {},
   "outputs": [],
   "source": []
  }
 ],
 "metadata": {
  "kernelspec": {
   "display_name": "Python 3 (ipykernel)",
   "language": "python",
   "name": "python3"
  }
 },
 "nbformat": 4,
 "nbformat_minor": 2
}