HW_准备绘制枝条颜色数据.ipynb

{
 "cells": [
  {
   "cell_type": "raw",
   "id": "61585f06-fc7a-4670-86b1-49147bc72f65",
   "metadata": {},
   "source": [
    "环境 py38data "
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 1,
   "id": "720d404f-7af7-442a-b430-d3c7718489f4",
   "metadata": {
    "execution": {
     "iopub.execute_input": "2024-03-04T02:40:42.182702Z",
     "iopub.status.busy": "2024-03-04T02:40:42.182394Z",
     "iopub.status.idle": "2024-03-04T02:40:42.395106Z",
     "shell.execute_reply": "2024-03-04T02:40:42.394607Z",
     "shell.execute_reply.started": "2024-03-04T02:40:42.182683Z"
    }
   },
   "outputs": [],
   "source": [
    "import pandas as pd"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 2,
   "id": "201fdcb5-8f14-4d4d-add8-c8ec728d539e",
   "metadata": {
    "execution": {
     "iopub.execute_input": "2024-03-04T02:40:42.396613Z",
     "iopub.status.busy": "2024-03-04T02:40:42.396340Z",
     "iopub.status.idle": "2024-03-04T02:40:42.410034Z",
     "shell.execute_reply": "2024-03-04T02:40:42.409522Z",
     "shell.execute_reply.started": "2024-03-04T02:40:42.396595Z"
    }
   },
   "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>0</th>\n",
       "      <th>1</th>\n",
       "      <th>taxa_id</th>\n",
       "      <th>docu</th>\n",
       "    </tr>\n",
       "  </thead>\n",
       "  <tbody>\n",
       "    <tr>\n",
       "      <th>0</th>\n",
       "      <td>[1]</td>\n",
       "      <td>'UIGHUR_XINJIANG_YILI_YINING'</td>\n",
       "      <td>1</td>\n",
       "      <td>UIGHUR_XINJIANG_YILI_YINING</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>1</th>\n",
       "      <td>[2]</td>\n",
       "      <td>'KAZAKH_XINJIANG_HABAHE'</td>\n",
       "      <td>2</td>\n",
       "      <td>KAZAKH_XINJIANG_HABAHE</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>2</th>\n",
       "      <td>[3]</td>\n",
       "      <td>'KAZAKH_XINJIANG_CHABUCHAER'</td>\n",
       "      <td>3</td>\n",
       "      <td>KAZAKH_XINJIANG_CHABUCHAER</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>3</th>\n",
       "      <td>[4]</td>\n",
       "      <td>'KIRGHIZ_XINJIANG_KEZILESU'</td>\n",
       "      <td>4</td>\n",
       "      <td>KIRGHIZ_XINJIANG_KEZILESU</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>4</th>\n",
       "      <td>[5]</td>\n",
       "      <td>'UZBEK_XINJIANG_ULUMUQI'</td>\n",
       "      <td>5</td>\n",
       "      <td>UZBEK_XINJIANG_ULUMUQI</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>5</th>\n",
       "      <td>[6]</td>\n",
       "      <td>'TATAR_XINJIANG_ULUMUQI'</td>\n",
       "      <td>6</td>\n",
       "      <td>TATAR_XINJIANG_ULUMUQI</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>6</th>\n",
       "      <td>[7]</td>\n",
       "      <td>'TUVA_XINJIANG_ALETAI_BUERJIN'</td>\n",
       "      <td>7</td>\n",
       "      <td>TUVA_XINJIANG_ALETAI_BUERJIN</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>7</th>\n",
       "      <td>[8]</td>\n",
       "      <td>'SALAER_QINGHAI_HAIDONG_XUNHUA'</td>\n",
       "      <td>8</td>\n",
       "      <td>SALAER_QINGHAI_HAIDONG_XUNHUA</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>8</th>\n",
       "      <td>[9]</td>\n",
       "      <td>'YUGU_GANSU_SUNAN_LIANHUA'</td>\n",
       "      <td>9</td>\n",
       "      <td>YUGU_GANSU_SUNAN_LIANHUA</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>9</th>\n",
       "      <td>[10]</td>\n",
       "      <td>'AYNU'</td>\n",
       "      <td>10</td>\n",
       "      <td>AYNU</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>10</th>\n",
       "      <td>[11]</td>\n",
       "      <td>'UYGHUR'</td>\n",
       "      <td>11</td>\n",
       "      <td>UYGHUR</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>11</th>\n",
       "      <td>[12]</td>\n",
       "      <td>'WEST_YUGUR'</td>\n",
       "      <td>12</td>\n",
       "      <td>WEST_YUGUR</td>\n",
       "    </tr>\n",
       "  </tbody>\n",
       "</table>\n",
       "</div>"
      ],
      "text/plain": [
       "       0                                1 taxa_id  \\\n",
       "0    [1]    'UIGHUR_XINJIANG_YILI_YINING'       1   \n",
       "1    [2]         'KAZAKH_XINJIANG_HABAHE'       2   \n",
       "2    [3]     'KAZAKH_XINJIANG_CHABUCHAER'       3   \n",
       "3    [4]      'KIRGHIZ_XINJIANG_KEZILESU'       4   \n",
       "4    [5]         'UZBEK_XINJIANG_ULUMUQI'       5   \n",
       "5    [6]         'TATAR_XINJIANG_ULUMUQI'       6   \n",
       "6    [7]   'TUVA_XINJIANG_ALETAI_BUERJIN'       7   \n",
       "7    [8]  'SALAER_QINGHAI_HAIDONG_XUNHUA'       8   \n",
       "8    [9]       'YUGU_GANSU_SUNAN_LIANHUA'       9   \n",
       "9   [10]                           'AYNU'      10   \n",
       "10  [11]                         'UYGHUR'      11   \n",
       "11  [12]                     'WEST_YUGUR'      12   \n",
       "\n",
       "                             docu  \n",
       "0     UIGHUR_XINJIANG_YILI_YINING  \n",
       "1          KAZAKH_XINJIANG_HABAHE  \n",
       "2      KAZAKH_XINJIANG_CHABUCHAER  \n",
       "3       KIRGHIZ_XINJIANG_KEZILESU  \n",
       "4          UZBEK_XINJIANG_ULUMUQI  \n",
       "5          TATAR_XINJIANG_ULUMUQI  \n",
       "6    TUVA_XINJIANG_ALETAI_BUERJIN  \n",
       "7   SALAER_QINGHAI_HAIDONG_XUNHUA  \n",
       "8        YUGU_GANSU_SUNAN_LIANHUA  \n",
       "9                            AYNU  \n",
       "10                         UYGHUR  \n",
       "11                     WEST_YUGUR  "
      ]
     },
     "execution_count": 2,
     "metadata": {},
     "output_type": "execute_result"
    }
   ],
   "source": [
    "## 读取 TAXLABELS 文件\n",
    "\n",
    "# 读取空格分隔的 txt 文件，不创建索引，没有表头\n",
    "df_taxlabels = pd.read_csv('TAXLABELS.txt', sep=' ', header=None, index_col=None)\n",
    "\n",
    "# 提取第一列方括号中的文本，并新建一列为 taxa_id\n",
    "df_taxlabels['taxa_id'] = df_taxlabels[0].str.extract(r'\\[(.*?)\\]', expand=False)\n",
    "\n",
    "# 提取第二列单引号中的文本，并新建一列为 docu\n",
    "df_taxlabels['docu'] = df_taxlabels[1].str.extract(r\"'(.*?)'\", expand=False)\n",
    "\n",
    "# 打印数据\n",
    "df_taxlabels"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 3,
   "id": "48f74f83-cb65-4f1a-8851-87f1cd69ba39",
   "metadata": {
    "execution": {
     "iopub.execute_input": "2024-03-04T02:40:42.425780Z",
     "iopub.status.busy": "2024-03-04T02:40:42.423787Z",
     "iopub.status.idle": "2024-03-04T02:40:42.436467Z",
     "shell.execute_reply": "2024-03-04T02:40:42.433099Z",
     "shell.execute_reply.started": "2024-03-04T02:40:42.425762Z"
    }
   },
   "outputs": [
    {
     "data": {
      "text/plain": [
       "12"
      ]
     },
     "execution_count": 3,
     "metadata": {},
     "output_type": "execute_result"
    }
   ],
   "source": [
    "# 计算 taxa_id 列中出现的非重复值的数量\n",
    "taxa_id_count = df_taxlabels['taxa_id'].nunique()\n",
    "taxa_id_count"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 4,
   "id": "8d8ff63b-870b-4b80-84d4-fcdde6af1096",
   "metadata": {
    "execution": {
     "iopub.execute_input": "2024-03-04T02:40:42.512425Z",
     "iopub.status.busy": "2024-03-04T02:40:42.512278Z",
     "iopub.status.idle": "2024-03-04T02:40:42.515100Z",
     "shell.execute_reply": "2024-03-04T02:40:42.514844Z",
     "shell.execute_reply.started": "2024-03-04T02:40:42.512414Z"
    }
   },
   "outputs": [
    {
     "data": {
      "text/plain": [
       "'1,2,3,4,5,6,7,8,9,10,11,12'"
      ]
     },
     "execution_count": 4,
     "metadata": {},
     "output_type": "execute_result"
    }
   ],
   "source": [
    "# 用逗号连接所有 taxa_id 的内容\n",
    "all_taxa_ids = ','.join(df_taxlabels['taxa_id'].astype(str))\n",
    "\n",
    "# 打印结果\n",
    "all_taxa_ids"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 5,
   "id": "234f2c7c-a7ae-43f4-9248-52fb25447edd",
   "metadata": {
    "execution": {
     "iopub.execute_input": "2024-03-04T02:40:42.686007Z",
     "iopub.status.busy": "2024-03-04T02:40:42.685785Z",
     "iopub.status.idle": "2024-03-04T02:40:42.696162Z",
     "shell.execute_reply": "2024-03-04T02:40:42.695818Z",
     "shell.execute_reply.started": "2024-03-04T02:40:42.685990Z"
    }
   },
   "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>0</th>\n",
       "      <th>taxa_ids</th>\n",
       "      <th>nsplits_id</th>\n",
       "      <th>taxacount</th>\n",
       "    </tr>\n",
       "  </thead>\n",
       "  <tbody>\n",
       "    <tr>\n",
       "      <th>0</th>\n",
       "      <td>[1, size=1]</td>\n",
       "      <td>1,</td>\n",
       "      <td>1</td>\n",
       "      <td>1</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>1</th>\n",
       "      <td>[6, size=1]</td>\n",
       "      <td>1,2,3,4,6,7,8,9,10,11,12,</td>\n",
       "      <td>6</td>\n",
       "      <td>11</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>2</th>\n",
       "      <td>[7, size=1]</td>\n",
       "      <td>1,2,3,4,5,6,7,8,9,10,12,</td>\n",
       "      <td>7</td>\n",
       "      <td>11</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>3</th>\n",
       "      <td>[11, size=1]</td>\n",
       "      <td>1,2,3,4,5,6,7,8,9,11,12,</td>\n",
       "      <td>11</td>\n",
       "      <td>11</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>4</th>\n",
       "      <td>[16, size=1]</td>\n",
       "      <td>1,2,3,4,5,6,8,9,10,11,12,</td>\n",
       "      <td>16</td>\n",
       "      <td>11</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>5</th>\n",
       "      <td>[19, size=1]</td>\n",
       "      <td>1,2,3,4,5,6,7,9,10,11,12,</td>\n",
       "      <td>19</td>\n",
       "      <td>11</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>6</th>\n",
       "      <td>[21, size=1]</td>\n",
       "      <td>1,2,3,4,5,6,7,8,10,11,12,</td>\n",
       "      <td>21</td>\n",
       "      <td>11</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>7</th>\n",
       "      <td>[24, size=1]</td>\n",
       "      <td>1,2,3,4,5,6,7,8,9,10,11,</td>\n",
       "      <td>24</td>\n",
       "      <td>11</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>8</th>\n",
       "      <td>[25, size=1]</td>\n",
       "      <td>1,3,4,5,6,7,8,9,10,11,12,</td>\n",
       "      <td>25</td>\n",
       "      <td>11</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>9</th>\n",
       "      <td>[29, size=1]</td>\n",
       "      <td>1,2,4,5,6,7,8,9,10,11,12,</td>\n",
       "      <td>29</td>\n",
       "      <td>11</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>10</th>\n",
       "      <td>[31, size=1]</td>\n",
       "      <td>1,2,3,5,6,7,8,9,10,11,12,</td>\n",
       "      <td>31</td>\n",
       "      <td>11</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>11</th>\n",
       "      <td>[32, size=1]</td>\n",
       "      <td>1,2,3,4,5,7,8,9,10,11,12,</td>\n",
       "      <td>32</td>\n",
       "      <td>11</td>\n",
       "    </tr>\n",
       "  </tbody>\n",
       "</table>\n",
       "</div>"
      ],
      "text/plain": [
       "                0                   taxa_ids nsplits_id  taxacount\n",
       "0    [1, size=1]                          1,          1          1\n",
       "1    [6, size=1]   1,2,3,4,6,7,8,9,10,11,12,          6         11\n",
       "2    [7, size=1]    1,2,3,4,5,6,7,8,9,10,12,          7         11\n",
       "3   [11, size=1]    1,2,3,4,5,6,7,8,9,11,12,         11         11\n",
       "4   [16, size=1]   1,2,3,4,5,6,8,9,10,11,12,         16         11\n",
       "5   [19, size=1]   1,2,3,4,5,6,7,9,10,11,12,         19         11\n",
       "6   [21, size=1]   1,2,3,4,5,6,7,8,10,11,12,         21         11\n",
       "7   [24, size=1]    1,2,3,4,5,6,7,8,9,10,11,         24         11\n",
       "8   [25, size=1]   1,3,4,5,6,7,8,9,10,11,12,         25         11\n",
       "9   [29, size=1]   1,2,4,5,6,7,8,9,10,11,12,         29         11\n",
       "10  [31, size=1]   1,2,3,5,6,7,8,9,10,11,12,         31         11\n",
       "11  [32, size=1]   1,2,3,4,5,7,8,9,10,11,12,         32         11"
      ]
     },
     "execution_count": 5,
     "metadata": {},
     "output_type": "execute_result"
    }
   ],
   "source": [
    "## 读取 MATRIX 文件\n",
    "\n",
    "# 读取 tab 分隔的 txt 文件，不创建索引，并且没有表头\n",
    "df_matrix = pd.read_csv('MATRIX.txt', sep='\\t', header=None, index_col=None)\n",
    "\n",
    "# 删除第一列中不包含 “size=1” 的行\n",
    "# “size=1” 表明这个节点只和一个 taxa 有关\n",
    "df_matrix = df_matrix[df_matrix[0].str.contains(\"size=1\")]\n",
    "\n",
    "# 删除第二列\n",
    "df_matrix = df_matrix.drop(columns=[1])\n",
    "\n",
    "# 将第二列中的空格替换为英文逗号\n",
    "df_matrix[2] = df_matrix[2].str.replace(' ', ',')\n",
    "\n",
    "# 删除第二列中出现在首位的英文逗号\n",
    "df_matrix[2] = df_matrix[2].str.lstrip(',')\n",
    "## 现在第二列中每个数字都有一个逗号，逗号的数量和数字的数量相等\n",
    "\n",
    "# 提取第一列中的数字\n",
    "df_matrix['nsplits_id'] = df_matrix[0].str.extract(r'\\[(\\d+),')\n",
    "\n",
    "# 计算第二列中逗号的数量，并存储在新的列中\n",
    "df_matrix['taxacount'] = df_matrix[2].str.count(',')\n",
    "\n",
    "# 重命名第二列为 'taxa_ids'\n",
    "df_matrix = df_matrix.rename(columns={2: 'taxa_ids'})\n",
    "\n",
    "# 重新设置索引\n",
    "df_matrix = df_matrix.reset_index(drop=True)\n",
    "\n",
    "# 打印读取的数据\n",
    "df_matrix"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 6,
   "id": "82d7046e-894a-41a2-a296-edacf2ab03f9",
   "metadata": {
    "execution": {
     "iopub.execute_input": "2024-03-04T02:40:42.861724Z",
     "iopub.status.busy": "2024-03-04T02:40:42.860997Z",
     "iopub.status.idle": "2024-03-04T02:40:42.878772Z",
     "shell.execute_reply": "2024-03-04T02:40:42.878081Z",
     "shell.execute_reply.started": "2024-03-04T02:40:42.861676Z"
    }
   },
   "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>0</th>\n",
       "      <th>taxa_ids</th>\n",
       "      <th>nsplits_id</th>\n",
       "      <th>taxacount</th>\n",
       "      <th>missing_taxa_id</th>\n",
       "    </tr>\n",
       "  </thead>\n",
       "  <tbody>\n",
       "    <tr>\n",
       "      <th>0</th>\n",
       "      <td>[1, size=1]</td>\n",
       "      <td>1,</td>\n",
       "      <td>1</td>\n",
       "      <td>1</td>\n",
       "      <td>1</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>1</th>\n",
       "      <td>[6, size=1]</td>\n",
       "      <td>1,2,3,4,6,7,8,9,10,11,12,</td>\n",
       "      <td>6</td>\n",
       "      <td>11</td>\n",
       "      <td>5</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>2</th>\n",
       "      <td>[7, size=1]</td>\n",
       "      <td>1,2,3,4,5,6,7,8,9,10,12,</td>\n",
       "      <td>7</td>\n",
       "      <td>11</td>\n",
       "      <td>11</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>3</th>\n",
       "      <td>[11, size=1]</td>\n",
       "      <td>1,2,3,4,5,6,7,8,9,11,12,</td>\n",
       "      <td>11</td>\n",
       "      <td>11</td>\n",
       "      <td>10</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>4</th>\n",
       "      <td>[16, size=1]</td>\n",
       "      <td>1,2,3,4,5,6,8,9,10,11,12,</td>\n",
       "      <td>16</td>\n",
       "      <td>11</td>\n",
       "      <td>7</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>5</th>\n",
       "      <td>[19, size=1]</td>\n",
       "      <td>1,2,3,4,5,6,7,9,10,11,12,</td>\n",
       "      <td>19</td>\n",
       "      <td>11</td>\n",
       "      <td>8</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>6</th>\n",
       "      <td>[21, size=1]</td>\n",
       "      <td>1,2,3,4,5,6,7,8,10,11,12,</td>\n",
       "      <td>21</td>\n",
       "      <td>11</td>\n",
       "      <td>9</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>7</th>\n",
       "      <td>[24, size=1]</td>\n",
       "      <td>1,2,3,4,5,6,7,8,9,10,11,</td>\n",
       "      <td>24</td>\n",
       "      <td>11</td>\n",
       "      <td>12</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>8</th>\n",
       "      <td>[25, size=1]</td>\n",
       "      <td>1,3,4,5,6,7,8,9,10,11,12,</td>\n",
       "      <td>25</td>\n",
       "      <td>11</td>\n",
       "      <td>2</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>9</th>\n",
       "      <td>[29, size=1]</td>\n",
       "      <td>1,2,4,5,6,7,8,9,10,11,12,</td>\n",
       "      <td>29</td>\n",
       "      <td>11</td>\n",
       "      <td>3</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>10</th>\n",
       "      <td>[31, size=1]</td>\n",
       "      <td>1,2,3,5,6,7,8,9,10,11,12,</td>\n",
       "      <td>31</td>\n",
       "      <td>11</td>\n",
       "      <td>4</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>11</th>\n",
       "      <td>[32, size=1]</td>\n",
       "      <td>1,2,3,4,5,7,8,9,10,11,12,</td>\n",
       "      <td>32</td>\n",
       "      <td>11</td>\n",
       "      <td>6</td>\n",
       "    </tr>\n",
       "  </tbody>\n",
       "</table>\n",
       "</div>"
      ],
      "text/plain": [
       "                0                   taxa_ids nsplits_id  taxacount  \\\n",
       "0    [1, size=1]                          1,          1          1   \n",
       "1    [6, size=1]   1,2,3,4,6,7,8,9,10,11,12,          6         11   \n",
       "2    [7, size=1]    1,2,3,4,5,6,7,8,9,10,12,          7         11   \n",
       "3   [11, size=1]    1,2,3,4,5,6,7,8,9,11,12,         11         11   \n",
       "4   [16, size=1]   1,2,3,4,5,6,8,9,10,11,12,         16         11   \n",
       "5   [19, size=1]   1,2,3,4,5,6,7,9,10,11,12,         19         11   \n",
       "6   [21, size=1]   1,2,3,4,5,6,7,8,10,11,12,         21         11   \n",
       "7   [24, size=1]    1,2,3,4,5,6,7,8,9,10,11,         24         11   \n",
       "8   [25, size=1]   1,3,4,5,6,7,8,9,10,11,12,         25         11   \n",
       "9   [29, size=1]   1,2,4,5,6,7,8,9,10,11,12,         29         11   \n",
       "10  [31, size=1]   1,2,3,5,6,7,8,9,10,11,12,         31         11   \n",
       "11  [32, size=1]   1,2,3,4,5,7,8,9,10,11,12,         32         11   \n",
       "\n",
       "   missing_taxa_id  \n",
       "0                1  \n",
       "1                5  \n",
       "2               11  \n",
       "3               10  \n",
       "4                7  \n",
       "5                8  \n",
       "6                9  \n",
       "7               12  \n",
       "8                2  \n",
       "9                3  \n",
       "10               4  \n",
       "11               6  "
      ]
     },
     "execution_count": 6,
     "metadata": {},
     "output_type": "execute_result"
    }
   ],
   "source": [
    "# # 制作 nsplits_id 和 taxa_id 的映射\n",
    "\n",
    "# 定义函数寻找在 taxa_ids 中没有出现的 id，即为该 nsplits_id 对应的 taxa_id\n",
    "# nsplits_id == \"1\" 除外，因为这个 1 就对应 taxa_id 的 1\n",
    "def find_missing_taxa_ids(row, nsplits_id):\n",
    "    if nsplits_id == \"1\":\n",
    "        return '1'\n",
    "    else:\n",
    "        missing_taxa_ids = []\n",
    "        row_taxa_ids = row.split(',')\n",
    "        for taxa_id in all_taxa_ids.split(','):\n",
    "            if taxa_id not in row_taxa_ids:\n",
    "                missing_taxa_ids.append(taxa_id)\n",
    "        return ','.join(missing_taxa_ids)\n",
    "\n",
    "# 应用函数到taxa_ids列\n",
    "df_matrix['missing_taxa_id'] = df_matrix.apply(lambda row: find_missing_taxa_ids(row['taxa_ids'], row['nsplits_id']), axis=1)\n",
    "\n",
    "# 打印结果\n",
    "df_matrix\n"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 7,
   "id": "72ed78b7-1a76-48ca-9a10-065c7f3c5958",
   "metadata": {
    "execution": {
     "iopub.execute_input": "2024-03-04T02:40:43.035142Z",
     "iopub.status.busy": "2024-03-04T02:40:43.034647Z",
     "iopub.status.idle": "2024-03-04T02:40:43.057760Z",
     "shell.execute_reply": "2024-03-04T02:40:43.057214Z",
     "shell.execute_reply.started": "2024-03-04T02:40:43.035109Z"
    }
   },
   "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>0</th>\n",
       "      <th>taxa_ids</th>\n",
       "      <th>nsplits_id</th>\n",
       "      <th>taxacount</th>\n",
       "      <th>missing_taxa_id</th>\n",
       "      <th>missing_taxa_docu</th>\n",
       "    </tr>\n",
       "  </thead>\n",
       "  <tbody>\n",
       "    <tr>\n",
       "      <th>0</th>\n",
       "      <td>[1, size=1]</td>\n",
       "      <td>1,</td>\n",
       "      <td>1</td>\n",
       "      <td>1</td>\n",
       "      <td>1</td>\n",
       "      <td>UIGHUR_XINJIANG_YILI_YINING</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>1</th>\n",
       "      <td>[6, size=1]</td>\n",
       "      <td>1,2,3,4,6,7,8,9,10,11,12,</td>\n",
       "      <td>6</td>\n",
       "      <td>11</td>\n",
       "      <td>5</td>\n",
       "      <td>UZBEK_XINJIANG_ULUMUQI</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>2</th>\n",
       "      <td>[7, size=1]</td>\n",
       "      <td>1,2,3,4,5,6,7,8,9,10,12,</td>\n",
       "      <td>7</td>\n",
       "      <td>11</td>\n",
       "      <td>11</td>\n",
       "      <td>UYGHUR</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>3</th>\n",
       "      <td>[11, size=1]</td>\n",
       "      <td>1,2,3,4,5,6,7,8,9,11,12,</td>\n",
       "      <td>11</td>\n",
       "      <td>11</td>\n",
       "      <td>10</td>\n",
       "      <td>AYNU</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>4</th>\n",
       "      <td>[16, size=1]</td>\n",
       "      <td>1,2,3,4,5,6,8,9,10,11,12,</td>\n",
       "      <td>16</td>\n",
       "      <td>11</td>\n",
       "      <td>7</td>\n",
       "      <td>TUVA_XINJIANG_ALETAI_BUERJIN</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>5</th>\n",
       "      <td>[19, size=1]</td>\n",
       "      <td>1,2,3,4,5,6,7,9,10,11,12,</td>\n",
       "      <td>19</td>\n",
       "      <td>11</td>\n",
       "      <td>8</td>\n",
       "      <td>SALAER_QINGHAI_HAIDONG_XUNHUA</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>6</th>\n",
       "      <td>[21, size=1]</td>\n",
       "      <td>1,2,3,4,5,6,7,8,10,11,12,</td>\n",
       "      <td>21</td>\n",
       "      <td>11</td>\n",
       "      <td>9</td>\n",
       "      <td>YUGU_GANSU_SUNAN_LIANHUA</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>7</th>\n",
       "      <td>[24, size=1]</td>\n",
       "      <td>1,2,3,4,5,6,7,8,9,10,11,</td>\n",
       "      <td>24</td>\n",
       "      <td>11</td>\n",
       "      <td>12</td>\n",
       "      <td>WEST_YUGUR</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>8</th>\n",
       "      <td>[25, size=1]</td>\n",
       "      <td>1,3,4,5,6,7,8,9,10,11,12,</td>\n",
       "      <td>25</td>\n",
       "      <td>11</td>\n",
       "      <td>2</td>\n",
       "      <td>KAZAKH_XINJIANG_HABAHE</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>9</th>\n",
       "      <td>[29, size=1]</td>\n",
       "      <td>1,2,4,5,6,7,8,9,10,11,12,</td>\n",
       "      <td>29</td>\n",
       "      <td>11</td>\n",
       "      <td>3</td>\n",
       "      <td>KAZAKH_XINJIANG_CHABUCHAER</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>10</th>\n",
       "      <td>[31, size=1]</td>\n",
       "      <td>1,2,3,5,6,7,8,9,10,11,12,</td>\n",
       "      <td>31</td>\n",
       "      <td>11</td>\n",
       "      <td>4</td>\n",
       "      <td>KIRGHIZ_XINJIANG_KEZILESU</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>11</th>\n",
       "      <td>[32, size=1]</td>\n",
       "      <td>1,2,3,4,5,7,8,9,10,11,12,</td>\n",
       "      <td>32</td>\n",
       "      <td>11</td>\n",
       "      <td>6</td>\n",
       "      <td>TATAR_XINJIANG_ULUMUQI</td>\n",
       "    </tr>\n",
       "  </tbody>\n",
       "</table>\n",
       "</div>"
      ],
      "text/plain": [
       "                0                   taxa_ids nsplits_id  taxacount  \\\n",
       "0    [1, size=1]                          1,          1          1   \n",
       "1    [6, size=1]   1,2,3,4,6,7,8,9,10,11,12,          6         11   \n",
       "2    [7, size=1]    1,2,3,4,5,6,7,8,9,10,12,          7         11   \n",
       "3   [11, size=1]    1,2,3,4,5,6,7,8,9,11,12,         11         11   \n",
       "4   [16, size=1]   1,2,3,4,5,6,8,9,10,11,12,         16         11   \n",
       "5   [19, size=1]   1,2,3,4,5,6,7,9,10,11,12,         19         11   \n",
       "6   [21, size=1]   1,2,3,4,5,6,7,8,10,11,12,         21         11   \n",
       "7   [24, size=1]    1,2,3,4,5,6,7,8,9,10,11,         24         11   \n",
       "8   [25, size=1]   1,3,4,5,6,7,8,9,10,11,12,         25         11   \n",
       "9   [29, size=1]   1,2,4,5,6,7,8,9,10,11,12,         29         11   \n",
       "10  [31, size=1]   1,2,3,5,6,7,8,9,10,11,12,         31         11   \n",
       "11  [32, size=1]   1,2,3,4,5,7,8,9,10,11,12,         32         11   \n",
       "\n",
       "   missing_taxa_id              missing_taxa_docu  \n",
       "0                1    UIGHUR_XINJIANG_YILI_YINING  \n",
       "1                5         UZBEK_XINJIANG_ULUMUQI  \n",
       "2               11                         UYGHUR  \n",
       "3               10                           AYNU  \n",
       "4                7   TUVA_XINJIANG_ALETAI_BUERJIN  \n",
       "5                8  SALAER_QINGHAI_HAIDONG_XUNHUA  \n",
       "6                9       YUGU_GANSU_SUNAN_LIANHUA  \n",
       "7               12                     WEST_YUGUR  \n",
       "8                2         KAZAKH_XINJIANG_HABAHE  \n",
       "9                3     KAZAKH_XINJIANG_CHABUCHAER  \n",
       "10               4      KIRGHIZ_XINJIANG_KEZILESU  \n",
       "11               6         TATAR_XINJIANG_ULUMUQI  "
      ]
     },
     "execution_count": 7,
     "metadata": {},
     "output_type": "execute_result"
    }
   ],
   "source": [
    "# 创建一个空的 Series，用于存储合并后的 docu 信息\n",
    "missing_taxa_docu = pd.Series(index=df_matrix.index)\n",
    "\n",
    "# 遍历 df_matrix 中的每一行\n",
    "for index, row in df_matrix.iterrows():\n",
    "    # 获取当前行的 missing_taxa_id\n",
    "    missing_taxa_id = row['missing_taxa_id']\n",
    "    \n",
    "    # 如果 missing_taxa_id 为空，则跳过\n",
    "    if pd.isnull(missing_taxa_id):\n",
    "        continue\n",
    "    \n",
    "    # 将 missing_taxa_id 拆分为单个 taxa_id\n",
    "    taxa_ids = [taxa_id.strip() for taxa_id in missing_taxa_id.split(',')]\n",
    "    \n",
    "    # 初始化 docu 列表\n",
    "    docu_list = []\n",
    "    \n",
    "    # 遍历 taxa_ids，根据每个 taxa_id 在 df_taxlabels 中查找对应的 docu 信息\n",
    "    for taxa_id in taxa_ids:\n",
    "        # 查找 taxa_id 对应的 docu 信息\n",
    "        docu = df_taxlabels.loc[df_taxlabels['taxa_id'] == taxa_id, 'docu'].values\n",
    "        \n",
    "        # 如果找到了 docu 信息，则添加到 docu_list 中\n",
    "        if len(docu) > 0:\n",
    "            docu_list.append(docu[0])\n",
    "    \n",
    "    # 将 docu_list 中的内容用逗号连接成一个字符串，并保存到 missing_taxa_docu 列中\n",
    "    missing_taxa_docu[index] = ', '.join(docu_list)\n",
    "\n",
    "# 将 missing_taxa_docu 列添加到 df_matrix 中\n",
    "df_matrix['missing_taxa_docu'] = missing_taxa_docu\n",
    "\n",
    "# 打印合并后的 df_matrix\n",
    "df_matrix\n"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 8,
   "id": "4cd93eba-af0b-41df-9bfa-b07c769e7541",
   "metadata": {
    "execution": {
     "iopub.execute_input": "2024-03-04T02:40:43.204659Z",
     "iopub.status.busy": "2024-03-04T02:40:43.204009Z",
     "iopub.status.idle": "2024-03-04T02:40:43.218868Z",
     "shell.execute_reply": "2024-03-04T02:40:43.218144Z",
     "shell.execute_reply.started": "2024-03-04T02:40:43.204613Z"
    }
   },
   "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>docu</th>\n",
       "      <th>label_c</th>\n",
       "      <th>label_background_c</th>\n",
       "      <th>edge_c</th>\n",
       "    </tr>\n",
       "  </thead>\n",
       "  <tbody>\n",
       "    <tr>\n",
       "      <th>0</th>\n",
       "      <td>UIGHUR_XINJIANG_YILI_YINING</td>\n",
       "      <td>1</td>\n",
       "      <td>0</td>\n",
       "      <td>1</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>1</th>\n",
       "      <td>KAZAKH_XINJIANG_HABAHE</td>\n",
       "      <td>1</td>\n",
       "      <td>0</td>\n",
       "      <td>1</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>2</th>\n",
       "      <td>KAZAKH_XINJIANG_CHABUCHAER</td>\n",
       "      <td>1</td>\n",
       "      <td>0</td>\n",
       "      <td>1</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>3</th>\n",
       "      <td>KIRGHIZ_XINJIANG_KEZILESU</td>\n",
       "      <td>2</td>\n",
       "      <td>0</td>\n",
       "      <td>2</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>4</th>\n",
       "      <td>UZBEK_XINJIANG_ULUMUQI</td>\n",
       "      <td>2</td>\n",
       "      <td>0</td>\n",
       "      <td>2</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>5</th>\n",
       "      <td>TATAR_XINJIANG_ULUMUQI</td>\n",
       "      <td>2</td>\n",
       "      <td>0</td>\n",
       "      <td>2</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>6</th>\n",
       "      <td>TUVA_XINJIANG_ALETAI_BUERJIN</td>\n",
       "      <td>2</td>\n",
       "      <td>0</td>\n",
       "      <td>2</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>7</th>\n",
       "      <td>SALAER_QINGHAI_HAIDONG_XUNHUA</td>\n",
       "      <td>3</td>\n",
       "      <td>0</td>\n",
       "      <td>3</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>8</th>\n",
       "      <td>YUGU_GANSU_SUNAN_LIANHUA</td>\n",
       "      <td>3</td>\n",
       "      <td>0</td>\n",
       "      <td>3</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>9</th>\n",
       "      <td>AYNU</td>\n",
       "      <td>3</td>\n",
       "      <td>0</td>\n",
       "      <td>3</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>10</th>\n",
       "      <td>UYGHUR</td>\n",
       "      <td>4</td>\n",
       "      <td>0</td>\n",
       "      <td>4</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>11</th>\n",
       "      <td>WEST_YUGUR</td>\n",
       "      <td>4</td>\n",
       "      <td>0</td>\n",
       "      <td>4</td>\n",
       "    </tr>\n",
       "  </tbody>\n",
       "</table>\n",
       "</div>"
      ],
      "text/plain": [
       "                             docu  label_c  label_background_c  edge_c\n",
       "0     UIGHUR_XINJIANG_YILI_YINING        1                   0       1\n",
       "1          KAZAKH_XINJIANG_HABAHE        1                   0       1\n",
       "2      KAZAKH_XINJIANG_CHABUCHAER        1                   0       1\n",
       "3       KIRGHIZ_XINJIANG_KEZILESU        2                   0       2\n",
       "4          UZBEK_XINJIANG_ULUMUQI        2                   0       2\n",
       "5          TATAR_XINJIANG_ULUMUQI        2                   0       2\n",
       "6    TUVA_XINJIANG_ALETAI_BUERJIN        2                   0       2\n",
       "7   SALAER_QINGHAI_HAIDONG_XUNHUA        3                   0       3\n",
       "8        YUGU_GANSU_SUNAN_LIANHUA        3                   0       3\n",
       "9                            AYNU        3                   0       3\n",
       "10                         UYGHUR        4                   0       4\n",
       "11                     WEST_YUGUR        4                   0       4"
      ]
     },
     "execution_count": 8,
     "metadata": {},
     "output_type": "execute_result"
    }
   ],
   "source": [
    "# 读取 tab 分隔的 docu_color.txt 文件\n",
    "df_docu_color = pd.read_csv('docu_color.txt', sep='\\t', header=0)\n",
    "\n",
    "# 打印\n",
    "df_docu_color"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 1,
   "id": "59985ef6-4294-4c95-82ed-2136d07e2bf9",
   "metadata": {
    "execution": {
     "iopub.execute_input": "2024-03-04T15:02:29.235078Z",
     "iopub.status.busy": "2024-03-04T15:02:29.234349Z",
     "iopub.status.idle": "2024-03-04T15:02:29.261338Z",
     "shell.execute_reply": "2024-03-04T15:02:29.260730Z",
     "shell.execute_reply.started": "2024-03-04T15:02:29.235008Z"
    }
   },
   "outputs": [],
   "source": [
    "## 创建颜色映射字典 ChatGPT 生成的，没有逐一测试是否正确：\n",
    "\n",
    "# 可以根据自己的需要调整数字和对应颜色的映射\n",
    "color_mapping = {\n",
    "    0: {\"rgba\": \"0x00000000\", \"description\": \"透明色\"},\n",
    "    1: {\"hex\": \"#ffffff\", \"rgba\": \"0x00ffffff\", \"description\": \"白色\"},\n",
    "    13: {\"hex\": \"#000000\", \"rgba\": \"0x000000ff\", \"description\": \"黑色\"},\n",
    "    3: {\"hex\": \"#ff0000\", \"rgba\": \"0xff0000ff\", \"description\": \"红色\"},\n",
    "    4: {\"hex\": \"#00ff00\", \"rgba\": \"0x00ff00ff\", \"description\": \"绿色\"},\n",
    "    2: {\"hex\": \"#0000ff\", \"rgba\": \"0x0000ffff\", \"description\": \"蓝色\"},\n",
    "    6: {\"hex\": \"#ffff00\", \"rgba\": \"0xffff00ff\", \"description\": \"黄色\"},\n",
    "    7: {\"hex\": \"#00ffff\", \"rgba\": \"0x00ffffff\", \"description\": \"青色\"},\n",
    "    8: {\"hex\": \"#ff00ff\", \"rgba\": \"0xff00ffff\", \"description\": \"品红\"},\n",
    "    9: {\"hex\": \"#800000\", \"rgba\": \"0x800000ff\", \"description\": \"栗色\"},\n",
    "    10: {\"hex\": \"#008000\", \"rgba\": \"0x008000ff\", \"description\": \"深绿色\"},\n",
    "    11: {\"hex\": \"#000080\", \"rgba\": \"0x000080ff\", \"description\": \"深蓝色\"},\n",
    "    12: {\"hex\": \"#808000\", \"rgba\": \"0x808000ff\", \"description\": \"橄榄色\"},\n",
    "    5: {\"hex\": \"#800080\", \"rgba\": \"0x800080ff\", \"description\": \"紫色\"},\n",
    "    14: {\"hex\": \"#008080\", \"rgba\": \"0x008080ff\", \"description\": \"墨绿色\"},\n",
    "    15: {\"hex\": \"#c0c0c0\", \"rgba\": \"0xc0c0c0ff\", \"description\": \"银色\"},\n",
    "    16: {\"hex\": \"#808080\", \"rgba\": \"0x808080ff\", \"description\": \"灰色\"},\n",
    "    17: {\"hex\": \"#ff9999\", \"rgba\": \"0xff9999ff\", \"description\": \"淡红色\"},\n",
    "    18: {\"hex\": \"#99ff99\", \"rgba\": \"0x99ff99ff\", \"description\": \"淡绿色\"},\n",
    "    19: {\"hex\": \"#9999ff\", \"rgba\": \"0x9999ffff\", \"description\": \"淡蓝色\"},\n",
    "    20: {\"hex\": \"#ffff99\", \"rgba\": \"0xffff99ff\", \"description\": \"浅黄色\"},\n",
    "}\n"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 10,
   "id": "2a1168e1-f502-470c-b8ef-aa9227ec17e8",
   "metadata": {
    "execution": {
     "iopub.execute_input": "2024-03-04T02:40:43.781359Z",
     "iopub.status.busy": "2024-03-04T02:40:43.780749Z",
     "iopub.status.idle": "2024-03-04T02:40:43.802240Z",
     "shell.execute_reply": "2024-03-04T02:40:43.801424Z",
     "shell.execute_reply.started": "2024-03-04T02:40:43.781326Z"
    }
   },
   "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>docu</th>\n",
       "      <th>label_c</th>\n",
       "      <th>label_background_c</th>\n",
       "      <th>edge_c</th>\n",
       "      <th>lab_c</th>\n",
       "      <th>lab_bg</th>\n",
       "      <th>edg_c</th>\n",
       "    </tr>\n",
       "  </thead>\n",
       "  <tbody>\n",
       "    <tr>\n",
       "      <th>0</th>\n",
       "      <td>UIGHUR_XINJIANG_YILI_YINING</td>\n",
       "      <td>1</td>\n",
       "      <td>0</td>\n",
       "      <td>1</td>\n",
       "      <td>0x00ffffff</td>\n",
       "      <td>0x00000000</td>\n",
       "      <td>0x00ffffff</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>1</th>\n",
       "      <td>KAZAKH_XINJIANG_HABAHE</td>\n",
       "      <td>1</td>\n",
       "      <td>0</td>\n",
       "      <td>1</td>\n",
       "      <td>0x00ffffff</td>\n",
       "      <td>0x00000000</td>\n",
       "      <td>0x00ffffff</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>2</th>\n",
       "      <td>KAZAKH_XINJIANG_CHABUCHAER</td>\n",
       "      <td>1</td>\n",
       "      <td>0</td>\n",
       "      <td>1</td>\n",
       "      <td>0x00ffffff</td>\n",
       "      <td>0x00000000</td>\n",
       "      <td>0x00ffffff</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>3</th>\n",
       "      <td>KIRGHIZ_XINJIANG_KEZILESU</td>\n",
       "      <td>2</td>\n",
       "      <td>0</td>\n",
       "      <td>2</td>\n",
       "      <td>0x0000ffff</td>\n",
       "      <td>0x00000000</td>\n",
       "      <td>0x0000ffff</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>4</th>\n",
       "      <td>UZBEK_XINJIANG_ULUMUQI</td>\n",
       "      <td>2</td>\n",
       "      <td>0</td>\n",
       "      <td>2</td>\n",
       "      <td>0x0000ffff</td>\n",
       "      <td>0x00000000</td>\n",
       "      <td>0x0000ffff</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>5</th>\n",
       "      <td>TATAR_XINJIANG_ULUMUQI</td>\n",
       "      <td>2</td>\n",
       "      <td>0</td>\n",
       "      <td>2</td>\n",
       "      <td>0x0000ffff</td>\n",
       "      <td>0x00000000</td>\n",
       "      <td>0x0000ffff</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>6</th>\n",
       "      <td>TUVA_XINJIANG_ALETAI_BUERJIN</td>\n",
       "      <td>2</td>\n",
       "      <td>0</td>\n",
       "      <td>2</td>\n",
       "      <td>0x0000ffff</td>\n",
       "      <td>0x00000000</td>\n",
       "      <td>0x0000ffff</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>7</th>\n",
       "      <td>SALAER_QINGHAI_HAIDONG_XUNHUA</td>\n",
       "      <td>3</td>\n",
       "      <td>0</td>\n",
       "      <td>3</td>\n",
       "      <td>0xff0000ff</td>\n",
       "      <td>0x00000000</td>\n",
       "      <td>0xff0000ff</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>8</th>\n",
       "      <td>YUGU_GANSU_SUNAN_LIANHUA</td>\n",
       "      <td>3</td>\n",
       "      <td>0</td>\n",
       "      <td>3</td>\n",
       "      <td>0xff0000ff</td>\n",
       "      <td>0x00000000</td>\n",
       "      <td>0xff0000ff</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>9</th>\n",
       "      <td>AYNU</td>\n",
       "      <td>3</td>\n",
       "      <td>0</td>\n",
       "      <td>3</td>\n",
       "      <td>0xff0000ff</td>\n",
       "      <td>0x00000000</td>\n",
       "      <td>0xff0000ff</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>10</th>\n",
       "      <td>UYGHUR</td>\n",
       "      <td>4</td>\n",
       "      <td>0</td>\n",
       "      <td>4</td>\n",
       "      <td>0x00ff00ff</td>\n",
       "      <td>0x00000000</td>\n",
       "      <td>0x00ff00ff</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>11</th>\n",
       "      <td>WEST_YUGUR</td>\n",
       "      <td>4</td>\n",
       "      <td>0</td>\n",
       "      <td>4</td>\n",
       "      <td>0x00ff00ff</td>\n",
       "      <td>0x00000000</td>\n",
       "      <td>0x00ff00ff</td>\n",
       "    </tr>\n",
       "  </tbody>\n",
       "</table>\n",
       "</div>"
      ],
      "text/plain": [
       "                             docu  label_c  label_background_c  edge_c  \\\n",
       "0     UIGHUR_XINJIANG_YILI_YINING        1                   0       1   \n",
       "1          KAZAKH_XINJIANG_HABAHE        1                   0       1   \n",
       "2      KAZAKH_XINJIANG_CHABUCHAER        1                   0       1   \n",
       "3       KIRGHIZ_XINJIANG_KEZILESU        2                   0       2   \n",
       "4          UZBEK_XINJIANG_ULUMUQI        2                   0       2   \n",
       "5          TATAR_XINJIANG_ULUMUQI        2                   0       2   \n",
       "6    TUVA_XINJIANG_ALETAI_BUERJIN        2                   0       2   \n",
       "7   SALAER_QINGHAI_HAIDONG_XUNHUA        3                   0       3   \n",
       "8        YUGU_GANSU_SUNAN_LIANHUA        3                   0       3   \n",
       "9                            AYNU        3                   0       3   \n",
       "10                         UYGHUR        4                   0       4   \n",
       "11                     WEST_YUGUR        4                   0       4   \n",
       "\n",
       "         lab_c      lab_bg       edg_c  \n",
       "0   0x00ffffff  0x00000000  0x00ffffff  \n",
       "1   0x00ffffff  0x00000000  0x00ffffff  \n",
       "2   0x00ffffff  0x00000000  0x00ffffff  \n",
       "3   0x0000ffff  0x00000000  0x0000ffff  \n",
       "4   0x0000ffff  0x00000000  0x0000ffff  \n",
       "5   0x0000ffff  0x00000000  0x0000ffff  \n",
       "6   0x0000ffff  0x00000000  0x0000ffff  \n",
       "7   0xff0000ff  0x00000000  0xff0000ff  \n",
       "8   0xff0000ff  0x00000000  0xff0000ff  \n",
       "9   0xff0000ff  0x00000000  0xff0000ff  \n",
       "10  0x00ff00ff  0x00000000  0x00ff00ff  \n",
       "11  0x00ff00ff  0x00000000  0x00ff00ff  "
      ]
     },
     "execution_count": 10,
     "metadata": {},
     "output_type": "execute_result"
    }
   ],
   "source": [
    "# 读取 label_c 列，并在 color_mapping 中查找对应的颜色信息\n",
    "df_docu_color['lab_c'] = df_docu_color['label_c'].apply(lambda x: color_mapping[int(x)]['rgba'] if int(x) in color_mapping else None)\n",
    "df_docu_color['lab_bg'] = df_docu_color['label_background_c'].apply(lambda x: color_mapping[int(x)]['rgba'] if int(x) in color_mapping else None)\n",
    "df_docu_color['edg_c'] = df_docu_color['edge_c'].apply(lambda x: color_mapping[int(x)]['rgba'] if int(x) in color_mapping else None)\n",
    "\n",
    "# 打印\n",
    "df_docu_color"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 11,
   "id": "3a2aadf5-d60a-4f29-8245-46e0fcb45a67",
   "metadata": {
    "execution": {
     "iopub.execute_input": "2024-03-04T02:40:44.006726Z",
     "iopub.status.busy": "2024-03-04T02:40:44.005875Z",
     "iopub.status.idle": "2024-03-04T02:40:44.027327Z",
     "shell.execute_reply": "2024-03-04T02:40:44.026800Z",
     "shell.execute_reply.started": "2024-03-04T02:40:44.006673Z"
    }
   },
   "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>0</th>\n",
       "      <th>taxa_ids</th>\n",
       "      <th>nsplits_id</th>\n",
       "      <th>taxacount</th>\n",
       "      <th>missing_taxa_id</th>\n",
       "      <th>missing_taxa_docu</th>\n",
       "      <th>docu</th>\n",
       "      <th>label_c</th>\n",
       "      <th>label_background_c</th>\n",
       "      <th>edge_c</th>\n",
       "      <th>lab_c</th>\n",
       "      <th>lab_bg</th>\n",
       "      <th>edg_c</th>\n",
       "    </tr>\n",
       "  </thead>\n",
       "  <tbody>\n",
       "    <tr>\n",
       "      <th>0</th>\n",
       "      <td>[1, size=1]</td>\n",
       "      <td>1,</td>\n",
       "      <td>1</td>\n",
       "      <td>1</td>\n",
       "      <td>1</td>\n",
       "      <td>UIGHUR_XINJIANG_YILI_YINING</td>\n",
       "      <td>UIGHUR_XINJIANG_YILI_YINING</td>\n",
       "      <td>1</td>\n",
       "      <td>0</td>\n",
       "      <td>1</td>\n",
       "      <td>0x00ffffff</td>\n",
       "      <td>0x00000000</td>\n",
       "      <td>0x00ffffff</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>1</th>\n",
       "      <td>[6, size=1]</td>\n",
       "      <td>1,2,3,4,6,7,8,9,10,11,12,</td>\n",
       "      <td>6</td>\n",
       "      <td>11</td>\n",
       "      <td>5</td>\n",
       "      <td>UZBEK_XINJIANG_ULUMUQI</td>\n",
       "      <td>UZBEK_XINJIANG_ULUMUQI</td>\n",
       "      <td>2</td>\n",
       "      <td>0</td>\n",
       "      <td>2</td>\n",
       "      <td>0x0000ffff</td>\n",
       "      <td>0x00000000</td>\n",
       "      <td>0x0000ffff</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>2</th>\n",
       "      <td>[7, size=1]</td>\n",
       "      <td>1,2,3,4,5,6,7,8,9,10,12,</td>\n",
       "      <td>7</td>\n",
       "      <td>11</td>\n",
       "      <td>11</td>\n",
       "      <td>UYGHUR</td>\n",
       "      <td>UYGHUR</td>\n",
       "      <td>4</td>\n",
       "      <td>0</td>\n",
       "      <td>4</td>\n",
       "      <td>0x00ff00ff</td>\n",
       "      <td>0x00000000</td>\n",
       "      <td>0x00ff00ff</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>3</th>\n",
       "      <td>[11, size=1]</td>\n",
       "      <td>1,2,3,4,5,6,7,8,9,11,12,</td>\n",
       "      <td>11</td>\n",
       "      <td>11</td>\n",
       "      <td>10</td>\n",
       "      <td>AYNU</td>\n",
       "      <td>AYNU</td>\n",
       "      <td>3</td>\n",
       "      <td>0</td>\n",
       "      <td>3</td>\n",
       "      <td>0xff0000ff</td>\n",
       "      <td>0x00000000</td>\n",
       "      <td>0xff0000ff</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>4</th>\n",
       "      <td>[16, size=1]</td>\n",
       "      <td>1,2,3,4,5,6,8,9,10,11,12,</td>\n",
       "      <td>16</td>\n",
       "      <td>11</td>\n",
       "      <td>7</td>\n",
       "      <td>TUVA_XINJIANG_ALETAI_BUERJIN</td>\n",
       "      <td>TUVA_XINJIANG_ALETAI_BUERJIN</td>\n",
       "      <td>2</td>\n",
       "      <td>0</td>\n",
       "      <td>2</td>\n",
       "      <td>0x0000ffff</td>\n",
       "      <td>0x00000000</td>\n",
       "      <td>0x0000ffff</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>5</th>\n",
       "      <td>[19, size=1]</td>\n",
       "      <td>1,2,3,4,5,6,7,9,10,11,12,</td>\n",
       "      <td>19</td>\n",
       "      <td>11</td>\n",
       "      <td>8</td>\n",
       "      <td>SALAER_QINGHAI_HAIDONG_XUNHUA</td>\n",
       "      <td>SALAER_QINGHAI_HAIDONG_XUNHUA</td>\n",
       "      <td>3</td>\n",
       "      <td>0</td>\n",
       "      <td>3</td>\n",
       "      <td>0xff0000ff</td>\n",
       "      <td>0x00000000</td>\n",
       "      <td>0xff0000ff</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>6</th>\n",
       "      <td>[21, size=1]</td>\n",
       "      <td>1,2,3,4,5,6,7,8,10,11,12,</td>\n",
       "      <td>21</td>\n",
       "      <td>11</td>\n",
       "      <td>9</td>\n",
       "      <td>YUGU_GANSU_SUNAN_LIANHUA</td>\n",
       "      <td>YUGU_GANSU_SUNAN_LIANHUA</td>\n",
       "      <td>3</td>\n",
       "      <td>0</td>\n",
       "      <td>3</td>\n",
       "      <td>0xff0000ff</td>\n",
       "      <td>0x00000000</td>\n",
       "      <td>0xff0000ff</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>7</th>\n",
       "      <td>[24, size=1]</td>\n",
       "      <td>1,2,3,4,5,6,7,8,9,10,11,</td>\n",
       "      <td>24</td>\n",
       "      <td>11</td>\n",
       "      <td>12</td>\n",
       "      <td>WEST_YUGUR</td>\n",
       "      <td>WEST_YUGUR</td>\n",
       "      <td>4</td>\n",
       "      <td>0</td>\n",
       "      <td>4</td>\n",
       "      <td>0x00ff00ff</td>\n",
       "      <td>0x00000000</td>\n",
       "      <td>0x00ff00ff</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>8</th>\n",
       "      <td>[25, size=1]</td>\n",
       "      <td>1,3,4,5,6,7,8,9,10,11,12,</td>\n",
       "      <td>25</td>\n",
       "      <td>11</td>\n",
       "      <td>2</td>\n",
       "      <td>KAZAKH_XINJIANG_HABAHE</td>\n",
       "      <td>KAZAKH_XINJIANG_HABAHE</td>\n",
       "      <td>1</td>\n",
       "      <td>0</td>\n",
       "      <td>1</td>\n",
       "      <td>0x00ffffff</td>\n",
       "      <td>0x00000000</td>\n",
       "      <td>0x00ffffff</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>9</th>\n",
       "      <td>[29, size=1]</td>\n",
       "      <td>1,2,4,5,6,7,8,9,10,11,12,</td>\n",
       "      <td>29</td>\n",
       "      <td>11</td>\n",
       "      <td>3</td>\n",
       "      <td>KAZAKH_XINJIANG_CHABUCHAER</td>\n",
       "      <td>KAZAKH_XINJIANG_CHABUCHAER</td>\n",
       "      <td>1</td>\n",
       "      <td>0</td>\n",
       "      <td>1</td>\n",
       "      <td>0x00ffffff</td>\n",
       "      <td>0x00000000</td>\n",
       "      <td>0x00ffffff</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>10</th>\n",
       "      <td>[31, size=1]</td>\n",
       "      <td>1,2,3,5,6,7,8,9,10,11,12,</td>\n",
       "      <td>31</td>\n",
       "      <td>11</td>\n",
       "      <td>4</td>\n",
       "      <td>KIRGHIZ_XINJIANG_KEZILESU</td>\n",
       "      <td>KIRGHIZ_XINJIANG_KEZILESU</td>\n",
       "      <td>2</td>\n",
       "      <td>0</td>\n",
       "      <td>2</td>\n",
       "      <td>0x0000ffff</td>\n",
       "      <td>0x00000000</td>\n",
       "      <td>0x0000ffff</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>11</th>\n",
       "      <td>[32, size=1]</td>\n",
       "      <td>1,2,3,4,5,7,8,9,10,11,12,</td>\n",
       "      <td>32</td>\n",
       "      <td>11</td>\n",
       "      <td>6</td>\n",
       "      <td>TATAR_XINJIANG_ULUMUQI</td>\n",
       "      <td>TATAR_XINJIANG_ULUMUQI</td>\n",
       "      <td>2</td>\n",
       "      <td>0</td>\n",
       "      <td>2</td>\n",
       "      <td>0x0000ffff</td>\n",
       "      <td>0x00000000</td>\n",
       "      <td>0x0000ffff</td>\n",
       "    </tr>\n",
       "  </tbody>\n",
       "</table>\n",
       "</div>"
      ],
      "text/plain": [
       "                0                   taxa_ids nsplits_id  taxacount  \\\n",
       "0    [1, size=1]                          1,          1          1   \n",
       "1    [6, size=1]   1,2,3,4,6,7,8,9,10,11,12,          6         11   \n",
       "2    [7, size=1]    1,2,3,4,5,6,7,8,9,10,12,          7         11   \n",
       "3   [11, size=1]    1,2,3,4,5,6,7,8,9,11,12,         11         11   \n",
       "4   [16, size=1]   1,2,3,4,5,6,8,9,10,11,12,         16         11   \n",
       "5   [19, size=1]   1,2,3,4,5,6,7,9,10,11,12,         19         11   \n",
       "6   [21, size=1]   1,2,3,4,5,6,7,8,10,11,12,         21         11   \n",
       "7   [24, size=1]    1,2,3,4,5,6,7,8,9,10,11,         24         11   \n",
       "8   [25, size=1]   1,3,4,5,6,7,8,9,10,11,12,         25         11   \n",
       "9   [29, size=1]   1,2,4,5,6,7,8,9,10,11,12,         29         11   \n",
       "10  [31, size=1]   1,2,3,5,6,7,8,9,10,11,12,         31         11   \n",
       "11  [32, size=1]   1,2,3,4,5,7,8,9,10,11,12,         32         11   \n",
       "\n",
       "   missing_taxa_id              missing_taxa_docu  \\\n",
       "0                1    UIGHUR_XINJIANG_YILI_YINING   \n",
       "1                5         UZBEK_XINJIANG_ULUMUQI   \n",
       "2               11                         UYGHUR   \n",
       "3               10                           AYNU   \n",
       "4                7   TUVA_XINJIANG_ALETAI_BUERJIN   \n",
       "5                8  SALAER_QINGHAI_HAIDONG_XUNHUA   \n",
       "6                9       YUGU_GANSU_SUNAN_LIANHUA   \n",
       "7               12                     WEST_YUGUR   \n",
       "8                2         KAZAKH_XINJIANG_HABAHE   \n",
       "9                3     KAZAKH_XINJIANG_CHABUCHAER   \n",
       "10               4      KIRGHIZ_XINJIANG_KEZILESU   \n",
       "11               6         TATAR_XINJIANG_ULUMUQI   \n",
       "\n",
       "                             docu  label_c  label_background_c  edge_c  \\\n",
       "0     UIGHUR_XINJIANG_YILI_YINING        1                   0       1   \n",
       "1          UZBEK_XINJIANG_ULUMUQI        2                   0       2   \n",
       "2                          UYGHUR        4                   0       4   \n",
       "3                            AYNU        3                   0       3   \n",
       "4    TUVA_XINJIANG_ALETAI_BUERJIN        2                   0       2   \n",
       "5   SALAER_QINGHAI_HAIDONG_XUNHUA        3                   0       3   \n",
       "6        YUGU_GANSU_SUNAN_LIANHUA        3                   0       3   \n",
       "7                      WEST_YUGUR        4                   0       4   \n",
       "8          KAZAKH_XINJIANG_HABAHE        1                   0       1   \n",
       "9      KAZAKH_XINJIANG_CHABUCHAER        1                   0       1   \n",
       "10      KIRGHIZ_XINJIANG_KEZILESU        2                   0       2   \n",
       "11         TATAR_XINJIANG_ULUMUQI        2                   0       2   \n",
       "\n",
       "         lab_c      lab_bg       edg_c  \n",
       "0   0x00ffffff  0x00000000  0x00ffffff  \n",
       "1   0x0000ffff  0x00000000  0x0000ffff  \n",
       "2   0x00ff00ff  0x00000000  0x00ff00ff  \n",
       "3   0xff0000ff  0x00000000  0xff0000ff  \n",
       "4   0x0000ffff  0x00000000  0x0000ffff  \n",
       "5   0xff0000ff  0x00000000  0xff0000ff  \n",
       "6   0xff0000ff  0x00000000  0xff0000ff  \n",
       "7   0x00ff00ff  0x00000000  0x00ff00ff  \n",
       "8   0x00ffffff  0x00000000  0x00ffffff  \n",
       "9   0x00ffffff  0x00000000  0x00ffffff  \n",
       "10  0x0000ffff  0x00000000  0x0000ffff  \n",
       "11  0x0000ffff  0x00000000  0x0000ffff  "
      ]
     },
     "execution_count": 11,
     "metadata": {},
     "output_type": "execute_result"
    }
   ],
   "source": [
    "# 将 df_docu_color 合并到 df_matrix 中，使用 missing_taxa_docu 列作为合并的键\n",
    "df_matrix = df_matrix.merge(df_docu_color, how='left', left_on='missing_taxa_docu', right_on='docu')\n",
    "\n",
    "# 打印\n",
    "df_matrix\n"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 12,
   "id": "6d1ad554-321c-4050-bb92-be5971920781",
   "metadata": {
    "execution": {
     "iopub.execute_input": "2024-03-04T02:40:44.174809Z",
     "iopub.status.busy": "2024-03-04T02:40:44.174349Z",
     "iopub.status.idle": "2024-03-04T02:40:44.193680Z",
     "shell.execute_reply": "2024-03-04T02:40:44.193083Z",
     "shell.execute_reply.started": "2024-03-04T02:40:44.174780Z"
    }
   },
   "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>0</th>\n",
       "      <th>taxa_ids</th>\n",
       "      <th>nsplits_id</th>\n",
       "      <th>taxacount</th>\n",
       "      <th>missing_taxa_id</th>\n",
       "      <th>missing_taxa_docu</th>\n",
       "      <th>docu</th>\n",
       "      <th>label_c</th>\n",
       "      <th>label_background_c</th>\n",
       "      <th>edge_c</th>\n",
       "      <th>lab_c</th>\n",
       "      <th>lab_bg</th>\n",
       "      <th>edg_c</th>\n",
       "      <th>labcolor</th>\n",
       "      <th>edgcolor</th>\n",
       "    </tr>\n",
       "  </thead>\n",
       "  <tbody>\n",
       "    <tr>\n",
       "      <th>0</th>\n",
       "      <td>[1, size=1]</td>\n",
       "      <td>1,</td>\n",
       "      <td>1</td>\n",
       "      <td>1</td>\n",
       "      <td>1</td>\n",
       "      <td>UIGHUR_XINJIANG_YILI_YINING</td>\n",
       "      <td>UIGHUR_XINJIANG_YILI_YINING</td>\n",
       "      <td>1</td>\n",
       "      <td>0</td>\n",
       "      <td>1</td>\n",
       "      <td>0x00ffffff</td>\n",
       "      <td>0x00000000</td>\n",
       "      <td>0x00ffffff</td>\n",
       "      <td>\\t[1] '&lt;c 0x00ffffff&gt;&lt;bg 0x00000000&gt;UIGHUR_XIN...</td>\n",
       "      <td>'c:1:0x00ffffff'</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>8</th>\n",
       "      <td>[25, size=1]</td>\n",
       "      <td>1,3,4,5,6,7,8,9,10,11,12,</td>\n",
       "      <td>25</td>\n",
       "      <td>11</td>\n",
       "      <td>2</td>\n",
       "      <td>KAZAKH_XINJIANG_HABAHE</td>\n",
       "      <td>KAZAKH_XINJIANG_HABAHE</td>\n",
       "      <td>1</td>\n",
       "      <td>0</td>\n",
       "      <td>1</td>\n",
       "      <td>0x00ffffff</td>\n",
       "      <td>0x00000000</td>\n",
       "      <td>0x00ffffff</td>\n",
       "      <td>\\t[2] '&lt;c 0x00ffffff&gt;&lt;bg 0x00000000&gt;KAZAKH_XIN...</td>\n",
       "      <td>'c:25:0x00ffffff'</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>9</th>\n",
       "      <td>[29, size=1]</td>\n",
       "      <td>1,2,4,5,6,7,8,9,10,11,12,</td>\n",
       "      <td>29</td>\n",
       "      <td>11</td>\n",
       "      <td>3</td>\n",
       "      <td>KAZAKH_XINJIANG_CHABUCHAER</td>\n",
       "      <td>KAZAKH_XINJIANG_CHABUCHAER</td>\n",
       "      <td>1</td>\n",
       "      <td>0</td>\n",
       "      <td>1</td>\n",
       "      <td>0x00ffffff</td>\n",
       "      <td>0x00000000</td>\n",
       "      <td>0x00ffffff</td>\n",
       "      <td>\\t[3] '&lt;c 0x00ffffff&gt;&lt;bg 0x00000000&gt;KAZAKH_XIN...</td>\n",
       "      <td>'c:29:0x00ffffff'</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>10</th>\n",
       "      <td>[31, size=1]</td>\n",
       "      <td>1,2,3,5,6,7,8,9,10,11,12,</td>\n",
       "      <td>31</td>\n",
       "      <td>11</td>\n",
       "      <td>4</td>\n",
       "      <td>KIRGHIZ_XINJIANG_KEZILESU</td>\n",
       "      <td>KIRGHIZ_XINJIANG_KEZILESU</td>\n",
       "      <td>2</td>\n",
       "      <td>0</td>\n",
       "      <td>2</td>\n",
       "      <td>0x0000ffff</td>\n",
       "      <td>0x00000000</td>\n",
       "      <td>0x0000ffff</td>\n",
       "      <td>\\t[4] '&lt;c 0x0000ffff&gt;&lt;bg 0x00000000&gt;KIRGHIZ_XI...</td>\n",
       "      <td>'c:31:0x0000ffff'</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>1</th>\n",
       "      <td>[6, size=1]</td>\n",
       "      <td>1,2,3,4,6,7,8,9,10,11,12,</td>\n",
       "      <td>6</td>\n",
       "      <td>11</td>\n",
       "      <td>5</td>\n",
       "      <td>UZBEK_XINJIANG_ULUMUQI</td>\n",
       "      <td>UZBEK_XINJIANG_ULUMUQI</td>\n",
       "      <td>2</td>\n",
       "      <td>0</td>\n",
       "      <td>2</td>\n",
       "      <td>0x0000ffff</td>\n",
       "      <td>0x00000000</td>\n",
       "      <td>0x0000ffff</td>\n",
       "      <td>\\t[5] '&lt;c 0x0000ffff&gt;&lt;bg 0x00000000&gt;UZBEK_XINJ...</td>\n",
       "      <td>'c:6:0x0000ffff'</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>11</th>\n",
       "      <td>[32, size=1]</td>\n",
       "      <td>1,2,3,4,5,7,8,9,10,11,12,</td>\n",
       "      <td>32</td>\n",
       "      <td>11</td>\n",
       "      <td>6</td>\n",
       "      <td>TATAR_XINJIANG_ULUMUQI</td>\n",
       "      <td>TATAR_XINJIANG_ULUMUQI</td>\n",
       "      <td>2</td>\n",
       "      <td>0</td>\n",
       "      <td>2</td>\n",
       "      <td>0x0000ffff</td>\n",
       "      <td>0x00000000</td>\n",
       "      <td>0x0000ffff</td>\n",
       "      <td>\\t[6] '&lt;c 0x0000ffff&gt;&lt;bg 0x00000000&gt;TATAR_XINJ...</td>\n",
       "      <td>'c:32:0x0000ffff'</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>4</th>\n",
       "      <td>[16, size=1]</td>\n",
       "      <td>1,2,3,4,5,6,8,9,10,11,12,</td>\n",
       "      <td>16</td>\n",
       "      <td>11</td>\n",
       "      <td>7</td>\n",
       "      <td>TUVA_XINJIANG_ALETAI_BUERJIN</td>\n",
       "      <td>TUVA_XINJIANG_ALETAI_BUERJIN</td>\n",
       "      <td>2</td>\n",
       "      <td>0</td>\n",
       "      <td>2</td>\n",
       "      <td>0x0000ffff</td>\n",
       "      <td>0x00000000</td>\n",
       "      <td>0x0000ffff</td>\n",
       "      <td>\\t[7] '&lt;c 0x0000ffff&gt;&lt;bg 0x00000000&gt;TUVA_XINJI...</td>\n",
       "      <td>'c:16:0x0000ffff'</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>5</th>\n",
       "      <td>[19, size=1]</td>\n",
       "      <td>1,2,3,4,5,6,7,9,10,11,12,</td>\n",
       "      <td>19</td>\n",
       "      <td>11</td>\n",
       "      <td>8</td>\n",
       "      <td>SALAER_QINGHAI_HAIDONG_XUNHUA</td>\n",
       "      <td>SALAER_QINGHAI_HAIDONG_XUNHUA</td>\n",
       "      <td>3</td>\n",
       "      <td>0</td>\n",
       "      <td>3</td>\n",
       "      <td>0xff0000ff</td>\n",
       "      <td>0x00000000</td>\n",
       "      <td>0xff0000ff</td>\n",
       "      <td>\\t[8] '&lt;c 0xff0000ff&gt;&lt;bg 0x00000000&gt;SALAER_QIN...</td>\n",
       "      <td>'c:19:0xff0000ff'</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>6</th>\n",
       "      <td>[21, size=1]</td>\n",
       "      <td>1,2,3,4,5,6,7,8,10,11,12,</td>\n",
       "      <td>21</td>\n",
       "      <td>11</td>\n",
       "      <td>9</td>\n",
       "      <td>YUGU_GANSU_SUNAN_LIANHUA</td>\n",
       "      <td>YUGU_GANSU_SUNAN_LIANHUA</td>\n",
       "      <td>3</td>\n",
       "      <td>0</td>\n",
       "      <td>3</td>\n",
       "      <td>0xff0000ff</td>\n",
       "      <td>0x00000000</td>\n",
       "      <td>0xff0000ff</td>\n",
       "      <td>\\t[9] '&lt;c 0xff0000ff&gt;&lt;bg 0x00000000&gt;YUGU_GANSU...</td>\n",
       "      <td>'c:21:0xff0000ff'</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>3</th>\n",
       "      <td>[11, size=1]</td>\n",
       "      <td>1,2,3,4,5,6,7,8,9,11,12,</td>\n",
       "      <td>11</td>\n",
       "      <td>11</td>\n",
       "      <td>10</td>\n",
       "      <td>AYNU</td>\n",
       "      <td>AYNU</td>\n",
       "      <td>3</td>\n",
       "      <td>0</td>\n",
       "      <td>3</td>\n",
       "      <td>0xff0000ff</td>\n",
       "      <td>0x00000000</td>\n",
       "      <td>0xff0000ff</td>\n",
       "      <td>\\t[10] '&lt;c 0xff0000ff&gt;&lt;bg 0x00000000&gt;AYNU'</td>\n",
       "      <td>'c:11:0xff0000ff'</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>2</th>\n",
       "      <td>[7, size=1]</td>\n",
       "      <td>1,2,3,4,5,6,7,8,9,10,12,</td>\n",
       "      <td>7</td>\n",
       "      <td>11</td>\n",
       "      <td>11</td>\n",
       "      <td>UYGHUR</td>\n",
       "      <td>UYGHUR</td>\n",
       "      <td>4</td>\n",
       "      <td>0</td>\n",
       "      <td>4</td>\n",
       "      <td>0x00ff00ff</td>\n",
       "      <td>0x00000000</td>\n",
       "      <td>0x00ff00ff</td>\n",
       "      <td>\\t[11] '&lt;c 0x00ff00ff&gt;&lt;bg 0x00000000&gt;UYGHUR'</td>\n",
       "      <td>'c:7:0x00ff00ff'</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>7</th>\n",
       "      <td>[24, size=1]</td>\n",
       "      <td>1,2,3,4,5,6,7,8,9,10,11,</td>\n",
       "      <td>24</td>\n",
       "      <td>11</td>\n",
       "      <td>12</td>\n",
       "      <td>WEST_YUGUR</td>\n",
       "      <td>WEST_YUGUR</td>\n",
       "      <td>4</td>\n",
       "      <td>0</td>\n",
       "      <td>4</td>\n",
       "      <td>0x00ff00ff</td>\n",
       "      <td>0x00000000</td>\n",
       "      <td>0x00ff00ff</td>\n",
       "      <td>\\t[12] '&lt;c 0x00ff00ff&gt;&lt;bg 0x00000000&gt;WEST_YUGUR'</td>\n",
       "      <td>'c:24:0x00ff00ff'</td>\n",
       "    </tr>\n",
       "  </tbody>\n",
       "</table>\n",
       "</div>"
      ],
      "text/plain": [
       "                0                   taxa_ids nsplits_id  taxacount  \\\n",
       "0    [1, size=1]                          1,          1          1   \n",
       "8   [25, size=1]   1,3,4,5,6,7,8,9,10,11,12,         25         11   \n",
       "9   [29, size=1]   1,2,4,5,6,7,8,9,10,11,12,         29         11   \n",
       "10  [31, size=1]   1,2,3,5,6,7,8,9,10,11,12,         31         11   \n",
       "1    [6, size=1]   1,2,3,4,6,7,8,9,10,11,12,          6         11   \n",
       "11  [32, size=1]   1,2,3,4,5,7,8,9,10,11,12,         32         11   \n",
       "4   [16, size=1]   1,2,3,4,5,6,8,9,10,11,12,         16         11   \n",
       "5   [19, size=1]   1,2,3,4,5,6,7,9,10,11,12,         19         11   \n",
       "6   [21, size=1]   1,2,3,4,5,6,7,8,10,11,12,         21         11   \n",
       "3   [11, size=1]    1,2,3,4,5,6,7,8,9,11,12,         11         11   \n",
       "2    [7, size=1]    1,2,3,4,5,6,7,8,9,10,12,          7         11   \n",
       "7   [24, size=1]    1,2,3,4,5,6,7,8,9,10,11,         24         11   \n",
       "\n",
       "    missing_taxa_id              missing_taxa_docu  \\\n",
       "0                 1    UIGHUR_XINJIANG_YILI_YINING   \n",
       "8                 2         KAZAKH_XINJIANG_HABAHE   \n",
       "9                 3     KAZAKH_XINJIANG_CHABUCHAER   \n",
       "10                4      KIRGHIZ_XINJIANG_KEZILESU   \n",
       "1                 5         UZBEK_XINJIANG_ULUMUQI   \n",
       "11                6         TATAR_XINJIANG_ULUMUQI   \n",
       "4                 7   TUVA_XINJIANG_ALETAI_BUERJIN   \n",
       "5                 8  SALAER_QINGHAI_HAIDONG_XUNHUA   \n",
       "6                 9       YUGU_GANSU_SUNAN_LIANHUA   \n",
       "3                10                           AYNU   \n",
       "2                11                         UYGHUR   \n",
       "7                12                     WEST_YUGUR   \n",
       "\n",
       "                             docu  label_c  label_background_c  edge_c  \\\n",
       "0     UIGHUR_XINJIANG_YILI_YINING        1                   0       1   \n",
       "8          KAZAKH_XINJIANG_HABAHE        1                   0       1   \n",
       "9      KAZAKH_XINJIANG_CHABUCHAER        1                   0       1   \n",
       "10      KIRGHIZ_XINJIANG_KEZILESU        2                   0       2   \n",
       "1          UZBEK_XINJIANG_ULUMUQI        2                   0       2   \n",
       "11         TATAR_XINJIANG_ULUMUQI        2                   0       2   \n",
       "4    TUVA_XINJIANG_ALETAI_BUERJIN        2                   0       2   \n",
       "5   SALAER_QINGHAI_HAIDONG_XUNHUA        3                   0       3   \n",
       "6        YUGU_GANSU_SUNAN_LIANHUA        3                   0       3   \n",
       "3                            AYNU        3                   0       3   \n",
       "2                          UYGHUR        4                   0       4   \n",
       "7                      WEST_YUGUR        4                   0       4   \n",
       "\n",
       "         lab_c      lab_bg       edg_c  \\\n",
       "0   0x00ffffff  0x00000000  0x00ffffff   \n",
       "8   0x00ffffff  0x00000000  0x00ffffff   \n",
       "9   0x00ffffff  0x00000000  0x00ffffff   \n",
       "10  0x0000ffff  0x00000000  0x0000ffff   \n",
       "1   0x0000ffff  0x00000000  0x0000ffff   \n",
       "11  0x0000ffff  0x00000000  0x0000ffff   \n",
       "4   0x0000ffff  0x00000000  0x0000ffff   \n",
       "5   0xff0000ff  0x00000000  0xff0000ff   \n",
       "6   0xff0000ff  0x00000000  0xff0000ff   \n",
       "3   0xff0000ff  0x00000000  0xff0000ff   \n",
       "2   0x00ff00ff  0x00000000  0x00ff00ff   \n",
       "7   0x00ff00ff  0x00000000  0x00ff00ff   \n",
       "\n",
       "                                             labcolor           edgcolor  \n",
       "0   \\t[1] '<c 0x00ffffff><bg 0x00000000>UIGHUR_XIN...   'c:1:0x00ffffff'  \n",
       "8   \\t[2] '<c 0x00ffffff><bg 0x00000000>KAZAKH_XIN...  'c:25:0x00ffffff'  \n",
       "9   \\t[3] '<c 0x00ffffff><bg 0x00000000>KAZAKH_XIN...  'c:29:0x00ffffff'  \n",
       "10  \\t[4] '<c 0x0000ffff><bg 0x00000000>KIRGHIZ_XI...  'c:31:0x0000ffff'  \n",
       "1   \\t[5] '<c 0x0000ffff><bg 0x00000000>UZBEK_XINJ...   'c:6:0x0000ffff'  \n",
       "11  \\t[6] '<c 0x0000ffff><bg 0x00000000>TATAR_XINJ...  'c:32:0x0000ffff'  \n",
       "4   \\t[7] '<c 0x0000ffff><bg 0x00000000>TUVA_XINJI...  'c:16:0x0000ffff'  \n",
       "5   \\t[8] '<c 0xff0000ff><bg 0x00000000>SALAER_QIN...  'c:19:0xff0000ff'  \n",
       "6   \\t[9] '<c 0xff0000ff><bg 0x00000000>YUGU_GANSU...  'c:21:0xff0000ff'  \n",
       "3          \\t[10] '<c 0xff0000ff><bg 0x00000000>AYNU'  'c:11:0xff0000ff'  \n",
       "2        \\t[11] '<c 0x00ff00ff><bg 0x00000000>UYGHUR'   'c:7:0x00ff00ff'  \n",
       "7    \\t[12] '<c 0x00ff00ff><bg 0x00000000>WEST_YUGUR'  'c:24:0x00ff00ff'  "
      ]
     },
     "execution_count": 12,
     "metadata": {},
     "output_type": "execute_result"
    }
   ],
   "source": [
    "# 生成新的 labcolor 列\n",
    "df_matrix['labcolor'] = df_matrix.apply(lambda row: f\"\t[{row['missing_taxa_id']}] '<c {row['lab_c']}><bg {row['lab_bg']}>{row['missing_taxa_docu']}'\" if pd.notnull(row['missing_taxa_id']) else '', axis=1)\n",
    "df_matrix['edgcolor'] = df_matrix.apply(lambda row: f\"'c:{row['nsplits_id']}:{row['edg_c']}'\" if pd.notnull(row['missing_taxa_id']) else '', axis=1)\n",
    "\n",
    "# 将 missing_taxa_id 转换为数字，实现升序排列\n",
    "df_matrix['missing_taxa_id'] = df_matrix['missing_taxa_id'].astype(int)\n",
    "df_matrix = df_matrix.sort_values(by='missing_taxa_id', ascending=True)\n",
    "\n",
    "# 打印\n",
    "df_matrix"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 13,
   "id": "e391cd17-68ba-4bf8-9a73-517a0cfe3051",
   "metadata": {
    "execution": {
     "iopub.execute_input": "2024-03-04T02:40:44.365777Z",
     "iopub.status.busy": "2024-03-04T02:40:44.365105Z",
     "iopub.status.idle": "2024-03-04T02:40:44.375453Z",
     "shell.execute_reply": "2024-03-04T02:40:44.374819Z",
     "shell.execute_reply.started": "2024-03-04T02:40:44.365731Z"
    }
   },
   "outputs": [
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "                                            edgcolor\n",
      "0  'c:1:0x00ffffff' 'c:25:0x00ffffff' 'c:29:0x00f...\n"
     ]
    }
   ],
   "source": [
    "# 连接 edgcolor 列的内容，用空格分隔\n",
    "df_edgc = ' '.join(df_matrix['edgcolor'].astype(str))\n",
    "\n",
    "# 保存为 DataFrame\n",
    "df_edgc = pd.DataFrame({'edgcolor': [df_edgc]})\n",
    "\n",
    "df_edgc.to_csv('df_edgcolor.txt', index=False, header=False)\n",
    "\n",
    "# 打印结果\n",
    "print(df_edgc)\n"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 15,
   "id": "7f2f2295-7938-45b9-8b21-f2b46d4b2bee",
   "metadata": {
    "execution": {
     "iopub.execute_input": "2024-03-04T04:02:08.439783Z",
     "iopub.status.busy": "2024-03-04T04:02:08.439083Z",
     "iopub.status.idle": "2024-03-04T04:02:08.465974Z",
     "shell.execute_reply": "2024-03-04T04:02:08.465143Z",
     "shell.execute_reply.started": "2024-03-04T04:02:08.439706Z"
    }
   },
   "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>labcolor</th>\n",
       "    </tr>\n",
       "  </thead>\n",
       "  <tbody>\n",
       "    <tr>\n",
       "      <th>0</th>\n",
       "      <td>\\t[1] '&lt;c 0x00ffffff&gt;&lt;bg 0x00000000&gt;UIGHUR_XIN...</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>8</th>\n",
       "      <td>\\t[2] '&lt;c 0x00ffffff&gt;&lt;bg 0x00000000&gt;KAZAKH_XIN...</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>9</th>\n",
       "      <td>\\t[3] '&lt;c 0x00ffffff&gt;&lt;bg 0x00000000&gt;KAZAKH_XIN...</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>10</th>\n",
       "      <td>\\t[4] '&lt;c 0x0000ffff&gt;&lt;bg 0x00000000&gt;KIRGHIZ_XI...</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>1</th>\n",
       "      <td>\\t[5] '&lt;c 0x0000ffff&gt;&lt;bg 0x00000000&gt;UZBEK_XINJ...</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>11</th>\n",
       "      <td>\\t[6] '&lt;c 0x0000ffff&gt;&lt;bg 0x00000000&gt;TATAR_XINJ...</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>4</th>\n",
       "      <td>\\t[7] '&lt;c 0x0000ffff&gt;&lt;bg 0x00000000&gt;TUVA_XINJI...</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>5</th>\n",
       "      <td>\\t[8] '&lt;c 0xff0000ff&gt;&lt;bg 0x00000000&gt;SALAER_QIN...</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>6</th>\n",
       "      <td>\\t[9] '&lt;c 0xff0000ff&gt;&lt;bg 0x00000000&gt;YUGU_GANSU...</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>3</th>\n",
       "      <td>\\t[10] '&lt;c 0xff0000ff&gt;&lt;bg 0x00000000&gt;AYNU'</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>2</th>\n",
       "      <td>\\t[11] '&lt;c 0x00ff00ff&gt;&lt;bg 0x00000000&gt;UYGHUR'</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>7</th>\n",
       "      <td>\\t[12] '&lt;c 0x00ff00ff&gt;&lt;bg 0x00000000&gt;WEST_YUGUR'</td>\n",
       "    </tr>\n",
       "  </tbody>\n",
       "</table>\n",
       "</div>"
      ],
      "text/plain": [
       "                                             labcolor\n",
       "0   \\t[1] '<c 0x00ffffff><bg 0x00000000>UIGHUR_XIN...\n",
       "8   \\t[2] '<c 0x00ffffff><bg 0x00000000>KAZAKH_XIN...\n",
       "9   \\t[3] '<c 0x00ffffff><bg 0x00000000>KAZAKH_XIN...\n",
       "10  \\t[4] '<c 0x0000ffff><bg 0x00000000>KIRGHIZ_XI...\n",
       "1   \\t[5] '<c 0x0000ffff><bg 0x00000000>UZBEK_XINJ...\n",
       "11  \\t[6] '<c 0x0000ffff><bg 0x00000000>TATAR_XINJ...\n",
       "4   \\t[7] '<c 0x0000ffff><bg 0x00000000>TUVA_XINJI...\n",
       "5   \\t[8] '<c 0xff0000ff><bg 0x00000000>SALAER_QIN...\n",
       "6   \\t[9] '<c 0xff0000ff><bg 0x00000000>YUGU_GANSU...\n",
       "3          \\t[10] '<c 0xff0000ff><bg 0x00000000>AYNU'\n",
       "2        \\t[11] '<c 0x00ff00ff><bg 0x00000000>UYGHUR'\n",
       "7    \\t[12] '<c 0x00ff00ff><bg 0x00000000>WEST_YUGUR'"
      ]
     },
     "execution_count": 15,
     "metadata": {},
     "output_type": "execute_result"
    }
   ],
   "source": [
    "# 从 df_matrix 中提取 labcolor 列并保存到 df_color 中\n",
    "df_color = df_matrix[['labcolor']]\n",
    "\n",
    "# 保存 df_color 为 tab 分隔的 txt 文件\n",
    "df_color.to_csv('df_color.txt', index=False, header=False)\n",
    "\n",
    "# 打印\n",
    "df_color"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": null,
   "id": "ea6d2cdd-a5a4-484c-b6b0-601f5ec84a27",
   "metadata": {},
   "outputs": [],
   "source": []
  },
  {
   "cell_type": "code",
   "execution_count": null,
   "id": "790fe0a7-5066-4d11-b302-0ac14ce65c9d",
   "metadata": {},
   "outputs": [],
   "source": []
  },
  {
   "cell_type": "code",
   "execution_count": null,
   "id": "ac3ecbd4-ea9a-42c3-b6e7-8395449cf8de",
   "metadata": {},
   "outputs": [],
   "source": []
  }
 ],
 "metadata": {
  "kernelspec": {
   "display_name": "Python [conda env:py38data]",
   "language": "python",
   "name": "conda-env-py38data-py"
  },
  "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.8.17"
  }
 },
 "nbformat": 4,
 "nbformat_minor": 5
}