easy_label_guohuan.ipynb

{
 "cells": [
  {
   "cell_type": "code",
   "execution_count": 2,
   "metadata": {},
   "outputs": [],
   "source": [
    "import numpy as np\n",
    "import os\n",
    "import pandas as pd\n",
    "import plotly.graph_objects as go\n",
    "import plotly.express as px\n",
    "import glob\n",
    "import matplotlib.pyplot as plt\n",
    "import tqdm\n",
    "import pywt\n",
    "import scipy.signal\n",
    "import scipy.ndimage\n",
    "import ecg\n",
    "import tqdm\n",
    "import joblib\n",
    "import pickle\n",
    "from sklearn.linear_model import LogisticRegression\n",
    "from sklearn.metrics import recall_score\n",
    "from sklearn.metrics import accuracy_score\n",
    "from sklearn.metrics import precision_score\n",
    "from sklearn.metrics import f1_score"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": null,
   "metadata": {},
   "outputs": [],
   "source": [
    "guohuan = pd.read_csv(r'D:\\\\byx\\\\baseline\\\\myEcg\\\\data\\\\过缓.csv')"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": null,
   "metadata": {},
   "outputs": [],
   "source": [
    "for index,age in zip(guohuan['index'],guohuan['age']):\n",
    "    print(index,age)"
   ]
  },
  {
   "cell_type": "markdown",
   "metadata": {},
   "source": [
    "## Test"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": null,
   "metadata": {},
   "outputs": [],
   "source": [
    "TP,FN = 0,0\n",
    "error = []\n",
    "for txtname,age in zip(guohuan['index'],guohuan['age']):\n",
    "    df = pd.read_csv(r'D:\\\\byx\\\\baseline\\\\myEcg\\\\data\\\\hf_round2_train\\\\%s'%txtname,delimiter=' ')\n",
    "    sampling_rate = 500\n",
    "    if ecg.raise_error(df['I']):\n",
    "        print('error',txtname)\n",
    "        continue\n",
    "    #peaks = R_peak(df,plot=False)\n",
    "    _,_,epochs = ecg.segment(df,plot=False)\n",
    "    #print((np.mean(r['T_Onsets']) - np.mean(r['P_Offsets'])) / 500)\n",
    "    #print('心率：',60 / len(epochs[0]) * sampling_rate)\n",
    "    xinlv = 60 / len(epochs[0]) * sampling_rate\n",
    "    if xinlv > 60:\n",
    "        #print(txtname)\n",
    "        #peaks = ecg.R_peak(df,plot=True)\n",
    "        #error.append(txtname)\n",
    "        FN += 1\n",
    "    else:\n",
    "        TP += 1\n",
    "print(TP/(TP+FN))"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": null,
   "metadata": {},
   "outputs": [],
   "source": [
    "FP,TN = 0,0\n",
    "trains = glob.glob(r'D:\\\\byx\\\\baseline\\\\myEcg\\\\data\\\\hf_round2_train\\\\**.txt')\n",
    "c = 0\n",
    "for txtname in tqdm.tqdm(trains[:100]):\n",
    "    txtn = txtname.split('\\\\')[-1]\n",
    "    if txtn not in guohuan['index']:\n",
    "        df = pd.read_csv(txtname,delimiter=' ')\n",
    "        if ecg.raise_error(df['I']):\n",
    "            print('error',txtname)\n",
    "            continue\n",
    "        _,_,epochs,_ = ecg.segment(df,plot=False)\n",
    "        xinlv = 60 / len(epochs[0]) * sampling_rate\n",
    "        if xinlv > 60:\n",
    "            #print(txtname)\n",
    "            #peaks = ecg.R_peak(df,plot=True)\n",
    "            #error.append(txtname)\n",
    "            TN += 1\n",
    "        else:\n",
    "            FP += 1\n",
    "            peaks = ecg.R_peak(df,plot=True)\n",
    "            print(xinlv)"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": null,
   "metadata": {},
   "outputs": [],
   "source": [
    "Precision = TP/(TP+FP)\n",
    "Recall = TP/(TP+FN)\n",
    "F1 = 2 * (Precision*Recall)/(Precision+Recall)\n",
    "Accuracy = (TP+TN) / (TP+FP+FN+TN)\n",
    "print(f'Precision = {Precision}')\n",
    "print(f'Recall = {Recall}')\n",
    "print(f'F1 = {F1}')\n",
    "print(f'Accuracy = {Accuracy}')"
   ]
  },
  {
   "cell_type": "markdown",
   "metadata": {},
   "source": [
    "# building dataset"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": null,
   "metadata": {},
   "outputs": [],
   "source": [
    "def heart_rating(txtname,sampling_rate=500):\n",
    "    df = pd.read_csv(r'D:\\\\byx\\\\baseline\\\\myEcg\\\\data\\\\hf_round2_train\\\\%s'%txtname,delimiter=' ')\n",
    "    if ecg.raise_error(df['I']):\n",
    "        print('error',txtname)\n",
    "        return np.nan\n",
    "    _,_,epochs,_ = ecg.segment(df,plot=False)\n",
    "    xinlv = 60 / len(epochs[0]) * sampling_rate\n",
    "    return xinlv"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": null,
   "metadata": {},
   "outputs": [],
   "source": [
    "dataset = []\n",
    "root = 'D:\\\\byx\\\\baseline\\\\myEcg\\\\data'\n",
    "train_csv = os.path.join(root,'hf_round2_label_train.csv')\n",
    "train = pd.read_csv(train_csv)\n",
    "train = train.dropna(axis=0,how='any')  "
   ]
  },
  {
   "cell_type": "code",
   "execution_count": null,
   "metadata": {},
   "outputs": [],
   "source": [
    "train['one_label'] = train['one_label'].map(lambda x: eval(x))\n",
    "train['guohuan'] = train['one_label'].map(lambda x:1 if x[18]==1.0 else 0)\n",
    "train['heart_ratings'] = train['index'].map(lambda x:heart_rating(x))\n",
    "train = train.dropna(axis=0,how='any')  "
   ]
  },
  {
   "cell_type": "code",
   "execution_count": null,
   "metadata": {},
   "outputs": [],
   "source": [
    "train['sex2'] = train['sex'].map(lambda x:1 if x=='MALE' else 0)\n",
    "train_size = int(len(train)*0.5)\n",
    "train_dataset = train[:train_size]\n",
    "test_dataset = train[train_size:]\n",
    "X_train = train_dataset[['age','heart_ratings','sex2']]\n",
    "y_train = train_dataset['guohuan']\n",
    "X_test = test_dataset[['age','heart_ratings','sex2']]\n",
    "y_test = test_dataset['guohuan']"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": null,
   "metadata": {},
   "outputs": [],
   "source": [
    "train.to_csv('train_heartrating.csv')"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": null,
   "metadata": {},
   "outputs": [],
   "source": [
    "clf = LogisticRegression(random_state=0).fit(X_train, y_train)\n",
    "y_pre = clf.predict(X_test)\n",
    "recall = recall_score(y_test,y_pre)\n",
    "accuracy = accuracy_score(y_test,y_pre)\n",
    "precision = precision_score(y_test,y_pre)\n",
    "F1 = f1_score(y_test,y_pre)\n",
    "print(f'Precision = {precision}')\n",
    "print(f'*Recall* = {recall}')\n",
    "print(f'F1 = {F1}')\n",
    "print(f'Accuracy = {accuracy}')"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": null,
   "metadata": {},
   "outputs": [],
   "source": [
    "train[train['guohuan']==1]"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": null,
   "metadata": {},
   "outputs": [],
   "source": [
    "clf.coef_,clf.intercept_"
   ]
  },
  {
   "cell_type": "markdown",
   "metadata": {},
   "source": [
    "# 过速"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": null,
   "metadata": {},
   "outputs": [],
   "source": [
    "train['guosu'] = train['one_label'].map(lambda x:1 if x[30]==1.0 else 0)"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": null,
   "metadata": {
    "scrolled": true
   },
   "outputs": [],
   "source": [
    "train_size = int(len(train)*0.5)\n",
    "train_dataset = train[:train_size]\n",
    "test_dataset = train[train_size:]\n",
    "X_train = train_dataset[['age','heart_ratings']]\n",
    "y_train = train_dataset['guosu']\n",
    "X_test = test_dataset[['age','heart_ratings']]\n",
    "y_test = test_dataset['guosu']\n",
    "\n",
    "clf = LogisticRegression(random_state=0).fit(X_train, y_train)\n",
    "y_pre = clf.predict(X_test)\n",
    "recall = recall_score(y_test,y_pre)\n",
    "accuracy = accuracy_score(y_test,y_pre)\n",
    "precision = precision_score(y_test,y_pre)\n",
    "F1 = f1_score(y_test,y_pre)\n",
    "print(f'Precision = {precision}')\n",
    "print(f'*Recall* = {recall}')\n",
    "print(f'F1 = {F1}')\n",
    "print(f'Accuracy = {accuracy}')"
   ]
  },
  {
   "cell_type": "markdown",
   "metadata": {},
   "source": [
    "### 过速 - 东阳数据集"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 5,
   "metadata": {},
   "outputs": [],
   "source": [
    "def heart_rating(txtname,sampling_rate=500):\n",
    "    df = pd.read_csv(r'D:\\\\byx\\\\Dongyang\\\\txt_data\\\\%s'%txtname,delimiter=' ')\n",
    "    if ecg.raise_error(df['I']) or ecg.raise_error(df['II']):\n",
    "        print('error',txtname)\n",
    "        return np.nan\n",
    "    try:\n",
    "        peaks = ecg.R_peak(df,lead='II')\n",
    "        xinlv = 60/((peaks[-1]-peaks[0])/(len(peaks)-1)/500)\n",
    "    except Exception as e:\n",
    "        print('***',txtname)\n",
    "        return np.nan\n",
    "    return xinlv"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 3,
   "metadata": {
    "collapsed": true
   },
   "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>Unnamed: 0</th>\n",
       "      <th>Unnamed: 0.1</th>\n",
       "      <th>0</th>\n",
       "      <th>1</th>\n",
       "      <th>过速</th>\n",
       "      <th>heart_rating</th>\n",
       "      <th>heart_rating_updated</th>\n",
       "    </tr>\n",
       "  </thead>\n",
       "  <tbody>\n",
       "    <tr>\n",
       "      <th>0</th>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>FILE@57f67388ce5ab304d8f0a72acf2339bf.txt</td>\n",
       "      <td>窦性心律　　正常心电图范围</td>\n",
       "      <td>0</td>\n",
       "      <td>68.649886</td>\n",
       "      <td>68.464730</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>1</th>\n",
       "      <td>1</td>\n",
       "      <td>1</td>\n",
       "      <td>FILE@c4505b5ef6c09c1afee93839e2de70ae.txt</td>\n",
       "      <td>窦性心律　　正常心电图范围</td>\n",
       "      <td>0</td>\n",
       "      <td>75.000000</td>\n",
       "      <td>74.859638</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>2</th>\n",
       "      <td>2</td>\n",
       "      <td>2</td>\n",
       "      <td>FILE@7b25b8706476ac7bbaaed23f902dd3b5.txt</td>\n",
       "      <td>窦性心律　　正常心电图范围</td>\n",
       "      <td>0</td>\n",
       "      <td>87.463557</td>\n",
       "      <td>87.073007</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>3</th>\n",
       "      <td>3</td>\n",
       "      <td>3</td>\n",
       "      <td>FILE@12eef1bdec5b766e4466dfd097c28a44.txt</td>\n",
       "      <td>窦性心律　　正常心电图范围</td>\n",
       "      <td>0</td>\n",
       "      <td>73.529412</td>\n",
       "      <td>73.349633</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>4</th>\n",
       "      <td>4</td>\n",
       "      <td>4</td>\n",
       "      <td>FILE@56e7521c27693266d32275536f1ff9e7.txt</td>\n",
       "      <td>1. 窦性心律\\n2. 房性早搏</td>\n",
       "      <td>0</td>\n",
       "      <td>71.770335</td>\n",
       "      <td>65.061809</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>5</th>\n",
       "      <td>5</td>\n",
       "      <td>5</td>\n",
       "      <td>FILE@4fe86725065038f57e4abca7391e5fd4.txt</td>\n",
       "      <td>窦性心律　　正常心电图范围</td>\n",
       "      <td>0</td>\n",
       "      <td>66.225166</td>\n",
       "      <td>66.006601</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>6</th>\n",
       "      <td>6</td>\n",
       "      <td>6</td>\n",
       "      <td>FILE@a5f945fd89f184ba13dc29dec0e81be0.txt</td>\n",
       "      <td>1. 窦性心律\\n2. 电轴左偏-39度</td>\n",
       "      <td>0</td>\n",
       "      <td>90.909091</td>\n",
       "      <td>90.869753</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>7</th>\n",
       "      <td>7</td>\n",
       "      <td>7</td>\n",
       "      <td>FILE@d83051d4ac81bba0d55fc9d2b3e7622e.txt</td>\n",
       "      <td>1. 窦性心律\\n2. 完全性左束支传导阻滞\\n3. ST-T改变</td>\n",
       "      <td>0</td>\n",
       "      <td>79.787234</td>\n",
       "      <td>65.919578</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>8</th>\n",
       "      <td>8</td>\n",
       "      <td>8</td>\n",
       "      <td>FILE@e1e98f0230ae026ffa216422e6117aa1.txt</td>\n",
       "      <td>窦性心律　　正常心电图范围</td>\n",
       "      <td>0</td>\n",
       "      <td>64.102564</td>\n",
       "      <td>63.938619</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>9</th>\n",
       "      <td>9</td>\n",
       "      <td>9</td>\n",
       "      <td>FILE@c02b925526cc24e490e453c7a5fc21f9.txt</td>\n",
       "      <td>窦性心律　　正常心电图范围</td>\n",
       "      <td>0</td>\n",
       "      <td>76.726343</td>\n",
       "      <td>76.514346</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>10</th>\n",
       "      <td>10</td>\n",
       "      <td>10</td>\n",
       "      <td>FILE@ea329dbb3e30e714e1d7e450a324bf72.txt</td>\n",
       "      <td>窦性心律　　正常心电图范围</td>\n",
       "      <td>0</td>\n",
       "      <td>81.521739</td>\n",
       "      <td>81.182348</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>11</th>\n",
       "      <td>11</td>\n",
       "      <td>11</td>\n",
       "      <td>FILE@74c6e7dce8a46afcce9aa460cdaff06d.txt</td>\n",
       "      <td>1. 窦性心律\\n2. 不完全性右束支传导阻滞</td>\n",
       "      <td>0</td>\n",
       "      <td>65.502183</td>\n",
       "      <td>65.387969</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>12</th>\n",
       "      <td>12</td>\n",
       "      <td>12</td>\n",
       "      <td>FILE@9f530f0ee2cc88d9be768889803d8f64.txt</td>\n",
       "      <td>窦性心律　　正常心电图范围</td>\n",
       "      <td>0</td>\n",
       "      <td>90.634441</td>\n",
       "      <td>90.400344</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>13</th>\n",
       "      <td>13</td>\n",
       "      <td>13</td>\n",
       "      <td>FILE@9d4194d88ab9caaf54694f9fde0295bf.txt</td>\n",
       "      <td>1. 窦性心律\\n2. Ⅰ度房室传导阻滞 (P-R间期214ms)\\n3. 左心室高电压\\n...</td>\n",
       "      <td>0</td>\n",
       "      <td>73.529412</td>\n",
       "      <td>73.170732</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>14</th>\n",
       "      <td>14</td>\n",
       "      <td>14</td>\n",
       "      <td>FILE@7a70d7e256068882e71d88cf25d4c999.txt</td>\n",
       "      <td>窦性心动过缓(50次/分)</td>\n",
       "      <td>0</td>\n",
       "      <td>56.603774</td>\n",
       "      <td>50.143266</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>15</th>\n",
       "      <td>15</td>\n",
       "      <td>15</td>\n",
       "      <td>FILE@33cc6c4556ebcdd2ea9fc50cfdd37ab6.txt</td>\n",
       "      <td>1. 窦性心律\\n2. T波改变</td>\n",
       "      <td>0</td>\n",
       "      <td>83.798883</td>\n",
       "      <td>83.529664</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>16</th>\n",
       "      <td>16</td>\n",
       "      <td>16</td>\n",
       "      <td>FILE@c6359a67e78fe3335d95d0a7118b8053.txt</td>\n",
       "      <td>窦性心律　　正常心电图范围</td>\n",
       "      <td>0</td>\n",
       "      <td>73.710074</td>\n",
       "      <td>73.644276</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>17</th>\n",
       "      <td>17</td>\n",
       "      <td>17</td>\n",
       "      <td>FILE@4cfdb9d56346bb87f6a56b3db405fa33.txt</td>\n",
       "      <td>不纯型 心房扑动 心室率 53次/分\\nT波改变</td>\n",
       "      <td>0</td>\n",
       "      <td>52.631579</td>\n",
       "      <td>52.481959</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>18</th>\n",
       "      <td>18</td>\n",
       "      <td>18</td>\n",
       "      <td>FILE@b12f78e872c542a5f204ef02c330b54c.txt</td>\n",
       "      <td>窦性心动过缓(50次/分)</td>\n",
       "      <td>0</td>\n",
       "      <td>50.335570</td>\n",
       "      <td>50.230222</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>19</th>\n",
       "      <td>19</td>\n",
       "      <td>19</td>\n",
       "      <td>FILE@732b92f424c809ab372de9a0ebe529e0.txt</td>\n",
       "      <td>窦性心律　\\n完全性右束支传导阻滞</td>\n",
       "      <td>0</td>\n",
       "      <td>75.000000</td>\n",
       "      <td>74.897867</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>20</th>\n",
       "      <td>20</td>\n",
       "      <td>20</td>\n",
       "      <td>FILE@2626bd8f28ddc63aa875a80e1f835d55.txt</td>\n",
       "      <td>1. 窦性心律\\n2. T波改变</td>\n",
       "      <td>0</td>\n",
       "      <td>80.645161</td>\n",
       "      <td>70.906747</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>21</th>\n",
       "      <td>21</td>\n",
       "      <td>21</td>\n",
       "      <td>FILE@d2c8e121b1ed98e76088714adeab5f25.txt</td>\n",
       "      <td>1. 窦性心动过缓(54次/分)\\n2. 房性早搏</td>\n",
       "      <td>0</td>\n",
       "      <td>54.347826</td>\n",
       "      <td>54.249548</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>22</th>\n",
       "      <td>22</td>\n",
       "      <td>22</td>\n",
       "      <td>FILE@16e6fbbfeb54246c0147ce3edc5743a8.txt</td>\n",
       "      <td>1. 窦性心律\\n2. T波轻度改变</td>\n",
       "      <td>0</td>\n",
       "      <td>68.027211</td>\n",
       "      <td>67.873303</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>23</th>\n",
       "      <td>23</td>\n",
       "      <td>23</td>\n",
       "      <td>FILE@d485582e8b439c384d1260fec96534e4.txt</td>\n",
       "      <td>窦性心律　　正常心电图范围</td>\n",
       "      <td>0</td>\n",
       "      <td>76.530612</td>\n",
       "      <td>76.465590</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>24</th>\n",
       "      <td>24</td>\n",
       "      <td>24</td>\n",
       "      <td>FILE@dcd874b83d5ed5265855e335106b1b5d.txt</td>\n",
       "      <td>窦性心动过缓伴不齐(59次/分)</td>\n",
       "      <td>0</td>\n",
       "      <td>66.225166</td>\n",
       "      <td>59.379811</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>25</th>\n",
       "      <td>25</td>\n",
       "      <td>25</td>\n",
       "      <td>FILE@f2fd8d5832a0f83e7ca29cf9c6f26dd0.txt</td>\n",
       "      <td>窦性心律　　正常心电图范围</td>\n",
       "      <td>0</td>\n",
       "      <td>88.757396</td>\n",
       "      <td>88.421053</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>26</th>\n",
       "      <td>26</td>\n",
       "      <td>26</td>\n",
       "      <td>FILE@2310e971d336c503ecf31d6a5e2b5494.txt</td>\n",
       "      <td>窦性心律　　正常心电图范围</td>\n",
       "      <td>0</td>\n",
       "      <td>71.428571</td>\n",
       "      <td>71.212775</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>27</th>\n",
       "      <td>27</td>\n",
       "      <td>27</td>\n",
       "      <td>FILE@8b29a4ed248393fc33a30e014db04c95.txt</td>\n",
       "      <td>窦性心律　　正常心电图范围</td>\n",
       "      <td>0</td>\n",
       "      <td>87.976540</td>\n",
       "      <td>87.755955</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>28</th>\n",
       "      <td>28</td>\n",
       "      <td>28</td>\n",
       "      <td>FILE@4158da9ffbb9710012d1d329e86f5671.txt</td>\n",
       "      <td>窦性心律　　正常心电图范围</td>\n",
       "      <td>0</td>\n",
       "      <td>70.588235</td>\n",
       "      <td>70.467649</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>29</th>\n",
       "      <td>29</td>\n",
       "      <td>29</td>\n",
       "      <td>FILE@46ecb99e99b72c0e1d168f19b1087228.txt</td>\n",
       "      <td>窦性心律　　正常心电图范围</td>\n",
       "      <td>0</td>\n",
       "      <td>65.217391</td>\n",
       "      <td>65.132436</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>...</th>\n",
       "      <td>...</td>\n",
       "      <td>...</td>\n",
       "      <td>...</td>\n",
       "      <td>...</td>\n",
       "      <td>...</td>\n",
       "      <td>...</td>\n",
       "      <td>...</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>58211</th>\n",
       "      <td>58211</td>\n",
       "      <td>58211</td>\n",
       "      <td>FILE@bd2c66a06b72f05fd6b061ce427d07ce.txt</td>\n",
       "      <td>1. 窦性心律\\n2. T波改变</td>\n",
       "      <td>0</td>\n",
       "      <td>61.475410</td>\n",
       "      <td>61.363636</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>58212</th>\n",
       "      <td>58212</td>\n",
       "      <td>58212</td>\n",
       "      <td>FILE@543636d28aa7b0c52cc67c6cac777fef.txt</td>\n",
       "      <td>窦性心律　　正常心电图范围</td>\n",
       "      <td>0</td>\n",
       "      <td>76.530612</td>\n",
       "      <td>76.371210</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>58213</th>\n",
       "      <td>58213</td>\n",
       "      <td>58213</td>\n",
       "      <td>FILE@0b1c29be605ebf014356ab92359d09cf.txt</td>\n",
       "      <td>窦性心律　　正常心电图范围</td>\n",
       "      <td>0</td>\n",
       "      <td>81.081081</td>\n",
       "      <td>80.789946</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>58214</th>\n",
       "      <td>58214</td>\n",
       "      <td>58214</td>\n",
       "      <td>FILE@ac5c01cdad4b189e80e45d44c50b1fdf.txt</td>\n",
       "      <td>窦性心律　　正常心电图范围</td>\n",
       "      <td>0</td>\n",
       "      <td>75.000000</td>\n",
       "      <td>74.750831</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>58215</th>\n",
       "      <td>58215</td>\n",
       "      <td>58215</td>\n",
       "      <td>FILE@3ee75c34a75fc87216c54c60432418ec.txt</td>\n",
       "      <td>窦性心律　　正常心电图范围</td>\n",
       "      <td>0</td>\n",
       "      <td>72.815534</td>\n",
       "      <td>72.591289</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>58216</th>\n",
       "      <td>58216</td>\n",
       "      <td>58216</td>\n",
       "      <td>FILE@9d39a8d5cba9031c1e37d9df18c40e75.txt</td>\n",
       "      <td>窦性心律　　正常心电图范围</td>\n",
       "      <td>0</td>\n",
       "      <td>66.371681</td>\n",
       "      <td>66.254417</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>58217</th>\n",
       "      <td>58217</td>\n",
       "      <td>58217</td>\n",
       "      <td>FILE@1abc430b71d7376a06ac30b5ceeed871.txt</td>\n",
       "      <td>窦性心律　　正常心电图范围</td>\n",
       "      <td>0</td>\n",
       "      <td>79.787234</td>\n",
       "      <td>79.575597</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>58218</th>\n",
       "      <td>58218</td>\n",
       "      <td>58218</td>\n",
       "      <td>FILE@28b69b923181474294a480cc6aa20c20.txt</td>\n",
       "      <td>窦性心律　　正常心电图范围</td>\n",
       "      <td>0</td>\n",
       "      <td>88.757396</td>\n",
       "      <td>88.315217</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>58219</th>\n",
       "      <td>58219</td>\n",
       "      <td>58219</td>\n",
       "      <td>FILE@af3991c5d21a9696f3c5520607ddc377.txt</td>\n",
       "      <td>窦性心律　　正常心电图范围</td>\n",
       "      <td>0</td>\n",
       "      <td>90.361446</td>\n",
       "      <td>90.051458</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>58220</th>\n",
       "      <td>58220</td>\n",
       "      <td>58220</td>\n",
       "      <td>FILE@acc8386dad0307f7fc860a043f1a3576.txt</td>\n",
       "      <td>窦性心律　　正常心电图范围</td>\n",
       "      <td>0</td>\n",
       "      <td>86.206897</td>\n",
       "      <td>86.920530</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>58221</th>\n",
       "      <td>58221</td>\n",
       "      <td>58221</td>\n",
       "      <td>FILE@5de7f63b960a3c83376fb44652a6a5c9.txt</td>\n",
       "      <td>窦性心律 正常心电图范围</td>\n",
       "      <td>0</td>\n",
       "      <td>63.694268</td>\n",
       "      <td>63.619227</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>58222</th>\n",
       "      <td>58222</td>\n",
       "      <td>58222</td>\n",
       "      <td>FILE@456eafa40e068ffa418f099e2d71788c.txt</td>\n",
       "      <td>窦性心律　　正常心电图范围</td>\n",
       "      <td>0</td>\n",
       "      <td>86.956522</td>\n",
       "      <td>86.859688</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>58223</th>\n",
       "      <td>58223</td>\n",
       "      <td>58223</td>\n",
       "      <td>FILE@f527688a78334bd802a266a57bad6ee9.txt</td>\n",
       "      <td>窦性心动过速(123次/分)</td>\n",
       "      <td>1</td>\n",
       "      <td>123.966942</td>\n",
       "      <td>123.355263</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>58224</th>\n",
       "      <td>58224</td>\n",
       "      <td>58224</td>\n",
       "      <td>FILE@ad77f34a0f2078972ae1e07591bb4d35.txt</td>\n",
       "      <td>窦性心动过速(101次/分)</td>\n",
       "      <td>1</td>\n",
       "      <td>101.010101</td>\n",
       "      <td>100.776821</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>58225</th>\n",
       "      <td>58225</td>\n",
       "      <td>58225</td>\n",
       "      <td>FILE@32a57329849dd8d209e8351e4e94ffb5.txt</td>\n",
       "      <td>窦性心动过速(111次/分)</td>\n",
       "      <td>1</td>\n",
       "      <td>111.111111</td>\n",
       "      <td>110.565111</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>58226</th>\n",
       "      <td>58226</td>\n",
       "      <td>58226</td>\n",
       "      <td>FILE@f686981ca76a606e2af8bc073394a35c.txt</td>\n",
       "      <td>窦性心律　　正常心电图范围</td>\n",
       "      <td>0</td>\n",
       "      <td>80.862534</td>\n",
       "      <td>80.554934</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>58227</th>\n",
       "      <td>58227</td>\n",
       "      <td>58227</td>\n",
       "      <td>FILE@d6cee8309b95fa4e483eb6aba2fab614.txt</td>\n",
       "      <td>窦性心律　　正常心电图范围</td>\n",
       "      <td>0</td>\n",
       "      <td>75.000000</td>\n",
       "      <td>74.781886</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>58228</th>\n",
       "      <td>58228</td>\n",
       "      <td>58228</td>\n",
       "      <td>FILE@bc1aec3a98d3f4da7765006e4ef8034b.txt</td>\n",
       "      <td>1. 窦性心动过缓(56次/分)\\n2. Ⅰ度房室传导阻滞 (P-R间期214ms)</td>\n",
       "      <td>0</td>\n",
       "      <td>63.157895</td>\n",
       "      <td>61.626952</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>58229</th>\n",
       "      <td>58229</td>\n",
       "      <td>58229</td>\n",
       "      <td>FILE@9eb83d55c154f2190b1d3d8d302f5f8e.txt</td>\n",
       "      <td>窦性心律　　正常心电图范围</td>\n",
       "      <td>0</td>\n",
       "      <td>72.992701</td>\n",
       "      <td>72.815534</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>58230</th>\n",
       "      <td>58230</td>\n",
       "      <td>58230</td>\n",
       "      <td>FILE@4b25436df649f1f5ae583bc1e97577ad.txt</td>\n",
       "      <td>窦性心律　　正常心电图范围</td>\n",
       "      <td>0</td>\n",
       "      <td>66.964286</td>\n",
       "      <td>66.815145</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>58231</th>\n",
       "      <td>58231</td>\n",
       "      <td>58231</td>\n",
       "      <td>FILE@ebdae8b47bcbc95c23162fc8faacc0be.txt</td>\n",
       "      <td>窦性心律　　正常心电图</td>\n",
       "      <td>0</td>\n",
       "      <td>95.846645</td>\n",
       "      <td>95.278425</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>58232</th>\n",
       "      <td>58232</td>\n",
       "      <td>58232</td>\n",
       "      <td>FILE@606df5637ef89dfe43a730d2f43dc216.txt</td>\n",
       "      <td>窦性心律　　正常心电图范围</td>\n",
       "      <td>0</td>\n",
       "      <td>96.463023</td>\n",
       "      <td>69.751221</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>58233</th>\n",
       "      <td>58233</td>\n",
       "      <td>58233</td>\n",
       "      <td>FILE@0c5c439c9e0cef4c038bdf54c1f833f2.txt</td>\n",
       "      <td>窦性心律  正常心电图</td>\n",
       "      <td>0</td>\n",
       "      <td>106.382979</td>\n",
       "      <td>105.890139</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>58234</th>\n",
       "      <td>58234</td>\n",
       "      <td>58234</td>\n",
       "      <td>FILE@8729bb0233969be833141490ba175923.txt</td>\n",
       "      <td>窦性心动过速(115次/分)</td>\n",
       "      <td>1</td>\n",
       "      <td>115.384615</td>\n",
       "      <td>114.698386</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>58235</th>\n",
       "      <td>58235</td>\n",
       "      <td>58235</td>\n",
       "      <td>FILE@5464f5974f8ee8455fa9607c1a3f7477.txt</td>\n",
       "      <td>窦性心律　　正常心电图范围</td>\n",
       "      <td>0</td>\n",
       "      <td>83.333333</td>\n",
       "      <td>82.996382</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>58236</th>\n",
       "      <td>58236</td>\n",
       "      <td>58236</td>\n",
       "      <td>FILE@b5c4d2e22a124d793059254175702bc6.txt</td>\n",
       "      <td>1. 窦性心律\\n2. T波改变</td>\n",
       "      <td>0</td>\n",
       "      <td>82.644628</td>\n",
       "      <td>82.469867</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>58237</th>\n",
       "      <td>58237</td>\n",
       "      <td>58237</td>\n",
       "      <td>FILE@a6cbb1112755b978b8c8aa861dd13a2d.txt</td>\n",
       "      <td>窦性心律　　正常心电图</td>\n",
       "      <td>0</td>\n",
       "      <td>82.191781</td>\n",
       "      <td>82.053440</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>58238</th>\n",
       "      <td>58238</td>\n",
       "      <td>58238</td>\n",
       "      <td>FILE@36baa1dce02de3fd2f2030873b90470b.txt</td>\n",
       "      <td>1. 窦性心律\\n2. 左心室高电压</td>\n",
       "      <td>0</td>\n",
       "      <td>90.909091</td>\n",
       "      <td>90.762404</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>58239</th>\n",
       "      <td>58239</td>\n",
       "      <td>58239</td>\n",
       "      <td>FILE@5badab33a8706f92f5421197c505eea5.txt</td>\n",
       "      <td>窦性心律　　正常心电图范围</td>\n",
       "      <td>0</td>\n",
       "      <td>61.099796</td>\n",
       "      <td>60.920578</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>58240</th>\n",
       "      <td>58240</td>\n",
       "      <td>58240</td>\n",
       "      <td>FILE@7841d4f004456a662f74721031cdca54.txt</td>\n",
       "      <td>窦性心动过速(110次/分)</td>\n",
       "      <td>1</td>\n",
       "      <td>110.701107</td>\n",
       "      <td>110.341843</td>\n",
       "    </tr>\n",
       "  </tbody>\n",
       "</table>\n",
       "<p>58241 rows × 7 columns</p>\n",
       "</div>"
      ],
      "text/plain": [
       "       Unnamed: 0  Unnamed: 0.1                                          0  \\\n",
       "0               0             0  FILE@57f67388ce5ab304d8f0a72acf2339bf.txt   \n",
       "1               1             1  FILE@c4505b5ef6c09c1afee93839e2de70ae.txt   \n",
       "2               2             2  FILE@7b25b8706476ac7bbaaed23f902dd3b5.txt   \n",
       "3               3             3  FILE@12eef1bdec5b766e4466dfd097c28a44.txt   \n",
       "4               4             4  FILE@56e7521c27693266d32275536f1ff9e7.txt   \n",
       "5               5             5  FILE@4fe86725065038f57e4abca7391e5fd4.txt   \n",
       "6               6             6  FILE@a5f945fd89f184ba13dc29dec0e81be0.txt   \n",
       "7               7             7  FILE@d83051d4ac81bba0d55fc9d2b3e7622e.txt   \n",
       "8               8             8  FILE@e1e98f0230ae026ffa216422e6117aa1.txt   \n",
       "9               9             9  FILE@c02b925526cc24e490e453c7a5fc21f9.txt   \n",
       "10             10            10  FILE@ea329dbb3e30e714e1d7e450a324bf72.txt   \n",
       "11             11            11  FILE@74c6e7dce8a46afcce9aa460cdaff06d.txt   \n",
       "12             12            12  FILE@9f530f0ee2cc88d9be768889803d8f64.txt   \n",
       "13             13            13  FILE@9d4194d88ab9caaf54694f9fde0295bf.txt   \n",
       "14             14            14  FILE@7a70d7e256068882e71d88cf25d4c999.txt   \n",
       "15             15            15  FILE@33cc6c4556ebcdd2ea9fc50cfdd37ab6.txt   \n",
       "16             16            16  FILE@c6359a67e78fe3335d95d0a7118b8053.txt   \n",
       "17             17            17  FILE@4cfdb9d56346bb87f6a56b3db405fa33.txt   \n",
       "18             18            18  FILE@b12f78e872c542a5f204ef02c330b54c.txt   \n",
       "19             19            19  FILE@732b92f424c809ab372de9a0ebe529e0.txt   \n",
       "20             20            20  FILE@2626bd8f28ddc63aa875a80e1f835d55.txt   \n",
       "21             21            21  FILE@d2c8e121b1ed98e76088714adeab5f25.txt   \n",
       "22             22            22  FILE@16e6fbbfeb54246c0147ce3edc5743a8.txt   \n",
       "23             23            23  FILE@d485582e8b439c384d1260fec96534e4.txt   \n",
       "24             24            24  FILE@dcd874b83d5ed5265855e335106b1b5d.txt   \n",
       "25             25            25  FILE@f2fd8d5832a0f83e7ca29cf9c6f26dd0.txt   \n",
       "26             26            26  FILE@2310e971d336c503ecf31d6a5e2b5494.txt   \n",
       "27             27            27  FILE@8b29a4ed248393fc33a30e014db04c95.txt   \n",
       "28             28            28  FILE@4158da9ffbb9710012d1d329e86f5671.txt   \n",
       "29             29            29  FILE@46ecb99e99b72c0e1d168f19b1087228.txt   \n",
       "...           ...           ...                                        ...   \n",
       "58211       58211         58211  FILE@bd2c66a06b72f05fd6b061ce427d07ce.txt   \n",
       "58212       58212         58212  FILE@543636d28aa7b0c52cc67c6cac777fef.txt   \n",
       "58213       58213         58213  FILE@0b1c29be605ebf014356ab92359d09cf.txt   \n",
       "58214       58214         58214  FILE@ac5c01cdad4b189e80e45d44c50b1fdf.txt   \n",
       "58215       58215         58215  FILE@3ee75c34a75fc87216c54c60432418ec.txt   \n",
       "58216       58216         58216  FILE@9d39a8d5cba9031c1e37d9df18c40e75.txt   \n",
       "58217       58217         58217  FILE@1abc430b71d7376a06ac30b5ceeed871.txt   \n",
       "58218       58218         58218  FILE@28b69b923181474294a480cc6aa20c20.txt   \n",
       "58219       58219         58219  FILE@af3991c5d21a9696f3c5520607ddc377.txt   \n",
       "58220       58220         58220  FILE@acc8386dad0307f7fc860a043f1a3576.txt   \n",
       "58221       58221         58221  FILE@5de7f63b960a3c83376fb44652a6a5c9.txt   \n",
       "58222       58222         58222  FILE@456eafa40e068ffa418f099e2d71788c.txt   \n",
       "58223       58223         58223  FILE@f527688a78334bd802a266a57bad6ee9.txt   \n",
       "58224       58224         58224  FILE@ad77f34a0f2078972ae1e07591bb4d35.txt   \n",
       "58225       58225         58225  FILE@32a57329849dd8d209e8351e4e94ffb5.txt   \n",
       "58226       58226         58226  FILE@f686981ca76a606e2af8bc073394a35c.txt   \n",
       "58227       58227         58227  FILE@d6cee8309b95fa4e483eb6aba2fab614.txt   \n",
       "58228       58228         58228  FILE@bc1aec3a98d3f4da7765006e4ef8034b.txt   \n",
       "58229       58229         58229  FILE@9eb83d55c154f2190b1d3d8d302f5f8e.txt   \n",
       "58230       58230         58230  FILE@4b25436df649f1f5ae583bc1e97577ad.txt   \n",
       "58231       58231         58231  FILE@ebdae8b47bcbc95c23162fc8faacc0be.txt   \n",
       "58232       58232         58232  FILE@606df5637ef89dfe43a730d2f43dc216.txt   \n",
       "58233       58233         58233  FILE@0c5c439c9e0cef4c038bdf54c1f833f2.txt   \n",
       "58234       58234         58234  FILE@8729bb0233969be833141490ba175923.txt   \n",
       "58235       58235         58235  FILE@5464f5974f8ee8455fa9607c1a3f7477.txt   \n",
       "58236       58236         58236  FILE@b5c4d2e22a124d793059254175702bc6.txt   \n",
       "58237       58237         58237  FILE@a6cbb1112755b978b8c8aa861dd13a2d.txt   \n",
       "58238       58238         58238  FILE@36baa1dce02de3fd2f2030873b90470b.txt   \n",
       "58239       58239         58239  FILE@5badab33a8706f92f5421197c505eea5.txt   \n",
       "58240       58240         58240  FILE@7841d4f004456a662f74721031cdca54.txt   \n",
       "\n",
       "                                                       1  过速  heart_rating  \\\n",
       "0                                          窦性心律　　正常心电图范围   0     68.649886   \n",
       "1                                          窦性心律　　正常心电图范围   0     75.000000   \n",
       "2                                          窦性心律　　正常心电图范围   0     87.463557   \n",
       "3                                          窦性心律　　正常心电图范围   0     73.529412   \n",
       "4                                       1. 窦性心律\\n2. 房性早搏   0     71.770335   \n",
       "5                                          窦性心律　　正常心电图范围   0     66.225166   \n",
       "6                                   1. 窦性心律\\n2. 电轴左偏-39度   0     90.909091   \n",
       "7                      1. 窦性心律\\n2. 完全性左束支传导阻滞\\n3. ST-T改变   0     79.787234   \n",
       "8                                          窦性心律　　正常心电图范围   0     64.102564   \n",
       "9                                          窦性心律　　正常心电图范围   0     76.726343   \n",
       "10                                         窦性心律　　正常心电图范围   0     81.521739   \n",
       "11                               1. 窦性心律\\n2. 不完全性右束支传导阻滞   0     65.502183   \n",
       "12                                         窦性心律　　正常心电图范围   0     90.634441   \n",
       "13     1. 窦性心律\\n2. Ⅰ度房室传导阻滞 (P-R间期214ms)\\n3. 左心室高电压\\n...   0     73.529412   \n",
       "14                                         窦性心动过缓(50次/分)   0     56.603774   \n",
       "15                                      1. 窦性心律\\n2. T波改变   0     83.798883   \n",
       "16                                         窦性心律　　正常心电图范围   0     73.710074   \n",
       "17                              不纯型 心房扑动 心室率 53次/分\\nT波改变   0     52.631579   \n",
       "18                                         窦性心动过缓(50次/分)   0     50.335570   \n",
       "19                                     窦性心律　\\n完全性右束支传导阻滞   0     75.000000   \n",
       "20                                      1. 窦性心律\\n2. T波改变   0     80.645161   \n",
       "21                             1. 窦性心动过缓(54次/分)\\n2. 房性早搏   0     54.347826   \n",
       "22                                    1. 窦性心律\\n2. T波轻度改变   0     68.027211   \n",
       "23                                         窦性心律　　正常心电图范围   0     76.530612   \n",
       "24                                      窦性心动过缓伴不齐(59次/分)   0     66.225166   \n",
       "25                                         窦性心律　　正常心电图范围   0     88.757396   \n",
       "26                                         窦性心律　　正常心电图范围   0     71.428571   \n",
       "27                                         窦性心律　　正常心电图范围   0     87.976540   \n",
       "28                                         窦性心律　　正常心电图范围   0     70.588235   \n",
       "29                                         窦性心律　　正常心电图范围   0     65.217391   \n",
       "...                                                  ...  ..           ...   \n",
       "58211                                   1. 窦性心律\\n2. T波改变   0     61.475410   \n",
       "58212                                      窦性心律　　正常心电图范围   0     76.530612   \n",
       "58213                                      窦性心律　　正常心电图范围   0     81.081081   \n",
       "58214                                      窦性心律　　正常心电图范围   0     75.000000   \n",
       "58215                                      窦性心律　　正常心电图范围   0     72.815534   \n",
       "58216                                      窦性心律　　正常心电图范围   0     66.371681   \n",
       "58217                                      窦性心律　　正常心电图范围   0     79.787234   \n",
       "58218                                      窦性心律　　正常心电图范围   0     88.757396   \n",
       "58219                                      窦性心律　　正常心电图范围   0     90.361446   \n",
       "58220                                      窦性心律　　正常心电图范围   0     86.206897   \n",
       "58221                                       窦性心律 正常心电图范围   0     63.694268   \n",
       "58222                                      窦性心律　　正常心电图范围   0     86.956522   \n",
       "58223                                     窦性心动过速(123次/分)   1    123.966942   \n",
       "58224                                     窦性心动过速(101次/分)   1    101.010101   \n",
       "58225                                     窦性心动过速(111次/分)   1    111.111111   \n",
       "58226                                      窦性心律　　正常心电图范围   0     80.862534   \n",
       "58227                                      窦性心律　　正常心电图范围   0     75.000000   \n",
       "58228         1. 窦性心动过缓(56次/分)\\n2. Ⅰ度房室传导阻滞 (P-R间期214ms)   0     63.157895   \n",
       "58229                                      窦性心律　　正常心电图范围   0     72.992701   \n",
       "58230                                      窦性心律　　正常心电图范围   0     66.964286   \n",
       "58231                                        窦性心律　　正常心电图   0     95.846645   \n",
       "58232                                      窦性心律　　正常心电图范围   0     96.463023   \n",
       "58233                                        窦性心律  正常心电图   0    106.382979   \n",
       "58234                                     窦性心动过速(115次/分)   1    115.384615   \n",
       "58235                                      窦性心律　　正常心电图范围   0     83.333333   \n",
       "58236                                   1. 窦性心律\\n2. T波改变   0     82.644628   \n",
       "58237                                        窦性心律　　正常心电图   0     82.191781   \n",
       "58238                                 1. 窦性心律\\n2. 左心室高电压   0     90.909091   \n",
       "58239                                      窦性心律　　正常心电图范围   0     61.099796   \n",
       "58240                                     窦性心动过速(110次/分)   1    110.701107   \n",
       "\n",
       "       heart_rating_updated  \n",
       "0                 68.464730  \n",
       "1                 74.859638  \n",
       "2                 87.073007  \n",
       "3                 73.349633  \n",
       "4                 65.061809  \n",
       "5                 66.006601  \n",
       "6                 90.869753  \n",
       "7                 65.919578  \n",
       "8                 63.938619  \n",
       "9                 76.514346  \n",
       "10                81.182348  \n",
       "11                65.387969  \n",
       "12                90.400344  \n",
       "13                73.170732  \n",
       "14                50.143266  \n",
       "15                83.529664  \n",
       "16                73.644276  \n",
       "17                52.481959  \n",
       "18                50.230222  \n",
       "19                74.897867  \n",
       "20                70.906747  \n",
       "21                54.249548  \n",
       "22                67.873303  \n",
       "23                76.465590  \n",
       "24                59.379811  \n",
       "25                88.421053  \n",
       "26                71.212775  \n",
       "27                87.755955  \n",
       "28                70.467649  \n",
       "29                65.132436  \n",
       "...                     ...  \n",
       "58211             61.363636  \n",
       "58212             76.371210  \n",
       "58213             80.789946  \n",
       "58214             74.750831  \n",
       "58215             72.591289  \n",
       "58216             66.254417  \n",
       "58217             79.575597  \n",
       "58218             88.315217  \n",
       "58219             90.051458  \n",
       "58220             86.920530  \n",
       "58221             63.619227  \n",
       "58222             86.859688  \n",
       "58223            123.355263  \n",
       "58224            100.776821  \n",
       "58225            110.565111  \n",
       "58226             80.554934  \n",
       "58227             74.781886  \n",
       "58228             61.626952  \n",
       "58229             72.815534  \n",
       "58230             66.815145  \n",
       "58231             95.278425  \n",
       "58232             69.751221  \n",
       "58233            105.890139  \n",
       "58234            114.698386  \n",
       "58235             82.996382  \n",
       "58236             82.469867  \n",
       "58237             82.053440  \n",
       "58238             90.762404  \n",
       "58239             60.920578  \n",
       "58240            110.341843  \n",
       "\n",
       "[58241 rows x 7 columns]"
      ]
     },
     "execution_count": 3,
     "metadata": {},
     "output_type": "execute_result"
    }
   ],
   "source": [
    "train = pd.read_csv(r'D:\\byx\\Dongyang\\label_add_heartrating_2.csv')\n",
    "train"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": null,
   "metadata": {},
   "outputs": [],
   "source": [
    "train = train.dropna(axis=0,how='any')  \n",
    "train_size = int(len(train)*0.5)\n",
    "train_dataset = train[:train_size]\n",
    "test_dataset = train[train_size:]\n",
    "X_train = train_dataset[['heart_rating']]\n",
    "y_train = train_dataset['过缓']\n",
    "X_test = test_dataset[['heart_rating']]\n",
    "y_test = test_dataset['过缓']\n",
    "\n",
    "clf = LogisticRegression(random_state=0).fit(X_train, y_train)\n",
    "y_pre = clf.predict(X_test)\n",
    "recall = recall_score(y_test,y_pre)\n",
    "accuracy = accuracy_score(y_test,y_pre)\n",
    "precision = precision_score(y_test,y_pre)\n",
    "F1 = f1_score(y_test,y_pre)\n",
    "print(f'Precision = {precision}')\n",
    "print(f'*Recall* = {recall}')\n",
    "print(f'F1 = {F1}')\n",
    "print(f'Accuracy = {accuracy}')"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 33,
   "metadata": {},
   "outputs": [
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "Precision = 0.7511910250499462\n",
      "*Recall* = 0.9920844327176781\n",
      "F1 = 0.8549938779079936\n",
      "Accuracy = 0.9715320822101269\n"
     ]
    }
   ],
   "source": [
    "y_pre = train['heart_rating_updated'].map(lambda x:1 if x <60else 0)\n",
    "y_test = train['过缓']\n",
    "recall = recall_score(y_test,y_pre)\n",
    "accuracy = accuracy_score(y_test,y_pre)\n",
    "precision = precision_score(y_test,y_pre)\n",
    "F1 = f1_score(y_test,y_pre)\n",
    "print(f'Precision = {precision}')\n",
    "print(f'*Recall* = {recall}')\n",
    "print(f'F1 = {F1}')\n",
    "print(f'Accuracy = {accuracy}')"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 34,
   "metadata": {},
   "outputs": [
    {
     "data": {
      "text/plain": [
       "(51695, 1619, 39, 4888)"
      ]
     },
     "execution_count": 34,
     "metadata": {},
     "output_type": "execute_result"
    }
   ],
   "source": [
    "from sklearn.metrics import confusion_matrix\n",
    "tn, fp, fn, tp = confusion_matrix(y_test,y_pre).ravel()\n",
    "tn, fp, fn, tp"
   ]
  }
 ],
 "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.3"
  }
 },
 "nbformat": 4,
 "nbformat_minor": 2
}