demo/Triangle-Counting.ipynb

{
 "cells": [
  {
   "cell_type": "markdown",
   "metadata": {},
   "source": [
    "# Triangle Counting\n",
    "\n",
    "These are translations of the methods for triangle counting from LAGraph_tricount()"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 1,
   "metadata": {},
   "outputs": [],
   "source": [
    "from pygraphblas import *\n",
    "from pygraphblas.gviz import *\n",
    "\n",
    "prob = dict(Matrix.ssget('Newman/karate'))\n",
    "M = prob['karate.mtx'].cast(INT64)"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 2,
   "metadata": {},
   "outputs": [
    {
     "data": {
      "text/plain": [
       "45"
      ]
     },
     "execution_count": 2,
     "metadata": {},
     "output_type": "execute_result"
    }
   ],
   "source": [
    "def cohen(A, U, L):\n",
    "    return L.mxm(U, mask=A).reduce_int() // 2\n",
    "    \n",
    "cohen(M, M.triu(), M.tril())"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 3,
   "metadata": {},
   "outputs": [
    {
     "data": {
      "text/plain": [
       "45"
      ]
     },
     "execution_count": 3,
     "metadata": {},
     "output_type": "execute_result"
    }
   ],
   "source": [
    "def sandia_dot(A, U, L):\n",
    "    return L.mxm(U.transpose(), mask=L).reduce_int()\n",
    "\n",
    "sandia_dot(M, M.triu(), M.tril())"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 8,
   "metadata": {},
   "outputs": [
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "45\n"
     ]
    },
    {
     "data": {
      "image/svg+xml": [
       "<?xml version=\"1.0\" encoding=\"UTF-8\" standalone=\"no\"?>\n",
       "<!DOCTYPE svg PUBLIC \"-//W3C//DTD SVG 1.1//EN\"\n",
       " \"http://www.w3.org/Graphics/SVG/1.1/DTD/svg11.dtd\">\n",
       "<!-- Generated by graphviz version 2.40.1 (20161225.0304)\n",
       " -->\n",
       "<!-- Title: %3 Pages: 1 -->\n",
       "<svg width=\"551pt\" height=\"342pt\"\n",
       " viewBox=\"0.00 0.00 550.70 341.75\" xmlns=\"http://www.w3.org/2000/svg\" xmlns:xlink=\"http://www.w3.org/1999/xlink\">\n",
       "<g id=\"graph0\" class=\"graph\" transform=\"scale(1 1) rotate(0) translate(4 337.7512)\">\n",
       "<title>%3</title>\n",
       "<polygon fill=\"#ffffff\" stroke=\"transparent\" points=\"-4,4 -4,-337.7512 546.7026,-337.7512 546.7026,4 -4,4\"/>\n",
       "<!-- 2 -->\n",
       "<g id=\"node1\" class=\"node\">\n",
       "<title>2</title>\n",
       "<ellipse fill=\"none\" stroke=\"#000000\" cx=\"367.0342\" cy=\"-155.7624\" rx=\"18\" ry=\"18\"/>\n",
       "<text text-anchor=\"middle\" x=\"367.0342\" y=\"-152.0624\" font-family=\"Times,serif\" font-size=\"14.00\" fill=\"#000000\">2</text>\n",
       "</g>\n",
       "<!-- 0 -->\n",
       "<g id=\"node2\" class=\"node\">\n",
       "<title>0</title>\n",
       "<ellipse fill=\"none\" stroke=\"#000000\" cx=\"426.389\" cy=\"-215.8624\" rx=\"18\" ry=\"18\"/>\n",
       "<text text-anchor=\"middle\" x=\"426.389\" y=\"-212.1624\" font-family=\"Times,serif\" font-size=\"14.00\" fill=\"#000000\">0</text>\n",
       "</g>\n",
       "<!-- 2&#45;&#45;0 -->\n",
       "<g id=\"edge1\" class=\"edge\">\n",
       "<title>2&#45;&#45;0</title>\n",
       "<g id=\"a_edge1\"><a xlink:title=\"1\">\n",
       "<path fill=\"none\" stroke=\"#000000\" d=\"M379.9133,-168.8032C389.8691,-178.884 403.5856,-192.7727 413.535,-202.847\"/>\n",
       "</a>\n",
       "</g>\n",
       "<text text-anchor=\"middle\" x=\"393.2242\" y=\"-189.6251\" font-family=\"Times,serif\" font-size=\"14.00\" fill=\"#000000\">1</text>\n",
       "</g>\n",
       "<!-- 3 -->\n",
       "<g id=\"node3\" class=\"node\">\n",
       "<title>3</title>\n",
       "<ellipse fill=\"none\" stroke=\"#000000\" cx=\"497.9165\" cy=\"-141.6187\" rx=\"18\" ry=\"18\"/>\n",
       "<text text-anchor=\"middle\" x=\"497.9165\" y=\"-137.9187\" font-family=\"Times,serif\" font-size=\"14.00\" fill=\"#000000\">3</text>\n",
       "</g>\n",
       "<!-- 3&#45;&#45;0 -->\n",
       "<g id=\"edge2\" class=\"edge\">\n",
       "<title>3&#45;&#45;0</title>\n",
       "<g id=\"a_edge2\"><a xlink:title=\"2\">\n",
       "<path fill=\"none\" stroke=\"#000000\" d=\"M485.1288,-154.892C472.1094,-168.4058 452.0453,-189.2319 439.066,-202.704\"/>\n",
       "</a>\n",
       "</g>\n",
       "<text text-anchor=\"middle\" x=\"458.5974\" y=\"-182.598\" font-family=\"Times,serif\" font-size=\"14.00\" fill=\"#000000\">2</text>\n",
       "</g>\n",
       "<!-- 1 -->\n",
       "<g id=\"node4\" class=\"node\">\n",
       "<title>1</title>\n",
       "<ellipse fill=\"none\" stroke=\"#000000\" cx=\"470.789\" cy=\"-84.1756\" rx=\"18\" ry=\"18\"/>\n",
       "<text text-anchor=\"middle\" x=\"470.789\" y=\"-80.4756\" font-family=\"Times,serif\" font-size=\"14.00\" fill=\"#000000\">1</text>\n",
       "</g>\n",
       "<!-- 3&#45;&#45;1 -->\n",
       "<g id=\"edge3\" class=\"edge\">\n",
       "<title>3&#45;&#45;1</title>\n",
       "<g id=\"a_edge3\"><a xlink:title=\"1\">\n",
       "<path fill=\"none\" stroke=\"#000000\" d=\"M490.2154,-125.3115C486.5813,-117.6162 482.2519,-108.4486 478.6035,-100.7231\"/>\n",
       "</a>\n",
       "</g>\n",
       "<text text-anchor=\"middle\" x=\"487.9095\" y=\"-101.8173\" font-family=\"Times,serif\" font-size=\"14.00\" fill=\"#000000\">1</text>\n",
       "</g>\n",
       "<!-- 6 -->\n",
       "<g id=\"node5\" class=\"node\">\n",
       "<title>6</title>\n",
       "<ellipse fill=\"none\" stroke=\"#000000\" cx=\"470.3934\" cy=\"-257.8615\" rx=\"18\" ry=\"18\"/>\n",
       "<text text-anchor=\"middle\" x=\"470.3934\" y=\"-254.1615\" font-family=\"Times,serif\" font-size=\"14.00\" fill=\"#000000\">6</text>\n",
       "</g>\n",
       "<!-- 6&#45;&#45;0 -->\n",
       "<g id=\"edge4\" class=\"edge\">\n",
       "<title>6&#45;&#45;0</title>\n",
       "<g id=\"a_edge4\"><a xlink:title=\"2\">\n",
       "<path fill=\"none\" stroke=\"#000000\" d=\"M456.9515,-245.0322C451.391,-239.725 444.9663,-233.5931 439.4466,-228.325\"/>\n",
       "</a>\n",
       "</g>\n",
       "<text text-anchor=\"middle\" x=\"444.6991\" y=\"-240.4786\" font-family=\"Times,serif\" font-size=\"14.00\" fill=\"#000000\">2</text>\n",
       "</g>\n",
       "<!-- 7 -->\n",
       "<g id=\"node6\" class=\"node\">\n",
       "<title>7</title>\n",
       "<ellipse fill=\"none\" stroke=\"#000000\" cx=\"410.0218\" cy=\"-111.626\" rx=\"18\" ry=\"18\"/>\n",
       "<text text-anchor=\"middle\" x=\"410.0218\" y=\"-107.926\" font-family=\"Times,serif\" font-size=\"14.00\" fill=\"#000000\">7</text>\n",
       "</g>\n",
       "<!-- 7&#45;&#45;2 -->\n",
       "<g id=\"edge7\" class=\"edge\">\n",
       "<title>7&#45;&#45;2</title>\n",
       "<g id=\"a_edge7\"><a xlink:title=\"1\">\n",
       "<path fill=\"none\" stroke=\"#000000\" d=\"M397.3566,-124.6297C391.809,-130.3255 385.3116,-136.9965 379.7574,-142.6992\"/>\n",
       "</a>\n",
       "</g>\n",
       "<text text-anchor=\"middle\" x=\"385.057\" y=\"-122.4644\" font-family=\"Times,serif\" font-size=\"14.00\" fill=\"#000000\">1</text>\n",
       "</g>\n",
       "<!-- 7&#45;&#45;0 -->\n",
       "<g id=\"edge5\" class=\"edge\">\n",
       "<title>7&#45;&#45;0</title>\n",
       "<g id=\"a_edge5\"><a xlink:title=\"3\">\n",
       "<path fill=\"none\" stroke=\"#000000\" d=\"M412.8728,-129.7826C415.8768,-148.9145 420.579,-178.8608 423.5707,-197.9138\"/>\n",
       "</a>\n",
       "</g>\n",
       "<text text-anchor=\"middle\" x=\"414.7217\" y=\"-167.6482\" font-family=\"Times,serif\" font-size=\"14.00\" fill=\"#000000\">3</text>\n",
       "</g>\n",
       "<!-- 7&#45;&#45;1 -->\n",
       "<g id=\"edge6\" class=\"edge\">\n",
       "<title>7&#45;&#45;1</title>\n",
       "<g id=\"a_edge6\"><a xlink:title=\"2\">\n",
       "<path fill=\"none\" stroke=\"#000000\" d=\"M426.6267,-104.1251C435.201,-100.2518 445.6327,-95.5394 454.2041,-91.6675\"/>\n",
       "</a>\n",
       "</g>\n",
       "<text text-anchor=\"middle\" x=\"436.9154\" y=\"-101.6963\" font-family=\"Times,serif\" font-size=\"14.00\" fill=\"#000000\">2</text>\n",
       "</g>\n",
       "<!-- 8 -->\n",
       "<g id=\"node7\" class=\"node\">\n",
       "<title>8</title>\n",
       "<ellipse fill=\"none\" stroke=\"#000000\" cx=\"274.8165\" cy=\"-215.7337\" rx=\"18\" ry=\"18\"/>\n",
       "<text text-anchor=\"middle\" x=\"274.8165\" y=\"-212.0337\" font-family=\"Times,serif\" font-size=\"14.00\" fill=\"#000000\">8</text>\n",
       "</g>\n",
       "<!-- 8&#45;&#45;0 -->\n",
       "<g id=\"edge8\" class=\"edge\">\n",
       "<title>8&#45;&#45;0</title>\n",
       "<g id=\"a_edge8\"><a xlink:title=\"1\">\n",
       "<path fill=\"none\" stroke=\"#000000\" d=\"M293.0997,-215.7492C322.2326,-215.7739 378.8673,-215.822 408.041,-215.8468\"/>\n",
       "</a>\n",
       "</g>\n",
       "<text text-anchor=\"middle\" x=\"347.0703\" y=\"-219.598\" font-family=\"Times,serif\" font-size=\"14.00\" fill=\"#000000\">1</text>\n",
       "</g>\n",
       "<!-- 10 -->\n",
       "<g id=\"node8\" class=\"node\">\n",
       "<title>10</title>\n",
       "<ellipse fill=\"none\" stroke=\"#000000\" cx=\"371.1258\" cy=\"-263.3056\" rx=\"18\" ry=\"18\"/>\n",
       "<text text-anchor=\"middle\" x=\"371.1258\" y=\"-259.6056\" font-family=\"Times,serif\" font-size=\"14.00\" fill=\"#000000\">10</text>\n",
       "</g>\n",
       "<!-- 10&#45;&#45;0 -->\n",
       "<g id=\"edge9\" class=\"edge\">\n",
       "<title>10&#45;&#45;0</title>\n",
       "<g id=\"a_edge9\"><a xlink:title=\"2\">\n",
       "<path fill=\"none\" stroke=\"#000000\" d=\"M384.7864,-251.578C393.2246,-244.3339 404.0217,-235.0646 412.5017,-227.7845\"/>\n",
       "</a>\n",
       "</g>\n",
       "<text text-anchor=\"middle\" x=\"395.144\" y=\"-243.4813\" font-family=\"Times,serif\" font-size=\"14.00\" fill=\"#000000\">2</text>\n",
       "</g>\n",
       "<!-- 12 -->\n",
       "<g id=\"node9\" class=\"node\">\n",
       "<title>12</title>\n",
       "<ellipse fill=\"none\" stroke=\"#000000\" cx=\"474.5561\" cy=\"-311.4466\" rx=\"18\" ry=\"18\"/>\n",
       "<text text-anchor=\"middle\" x=\"474.5561\" y=\"-307.7466\" font-family=\"Times,serif\" font-size=\"14.00\" fill=\"#000000\">12</text>\n",
       "</g>\n",
       "<!-- 12&#45;&#45;0 -->\n",
       "<g id=\"edge10\" class=\"edge\">\n",
       "<title>12&#45;&#45;0</title>\n",
       "<g id=\"a_edge10\"><a xlink:title=\"1\">\n",
       "<path fill=\"none\" stroke=\"#000000\" d=\"M466.3852,-295.2321C457.5279,-277.6553 443.473,-249.7643 434.6,-232.1565\"/>\n",
       "</a>\n",
       "</g>\n",
       "<text text-anchor=\"middle\" x=\"446.9926\" y=\"-267.4943\" font-family=\"Times,serif\" font-size=\"14.00\" fill=\"#000000\">1</text>\n",
       "</g>\n",
       "<!-- 13 -->\n",
       "<g id=\"node10\" class=\"node\">\n",
       "<title>13</title>\n",
       "<ellipse fill=\"none\" stroke=\"#000000\" cx=\"445.3277\" cy=\"-155.6617\" rx=\"18\" ry=\"18\"/>\n",
       "<text text-anchor=\"middle\" x=\"445.3277\" y=\"-151.9617\" font-family=\"Times,serif\" font-size=\"14.00\" fill=\"#000000\">13</text>\n",
       "</g>\n",
       "<!-- 13&#45;&#45;2 -->\n",
       "<g id=\"edge13\" class=\"edge\">\n",
       "<title>13&#45;&#45;2</title>\n",
       "<g id=\"a_edge13\"><a xlink:title=\"1\">\n",
       "<path fill=\"none\" stroke=\"#000000\" d=\"M427.1688,-155.6851C414.5203,-155.7014 397.7215,-155.723 385.0946,-155.7392\"/>\n",
       "</a>\n",
       "</g>\n",
       "<text text-anchor=\"middle\" x=\"402.6317\" y=\"-159.5121\" font-family=\"Times,serif\" font-size=\"14.00\" fill=\"#000000\">1</text>\n",
       "</g>\n",
       "<!-- 13&#45;&#45;0 -->\n",
       "<g id=\"edge11\" class=\"edge\">\n",
       "<title>13&#45;&#45;0</title>\n",
       "<g id=\"a_edge11\"><a xlink:title=\"3\">\n",
       "<path fill=\"none\" stroke=\"#000000\" d=\"M439.8498,-173.0742C437.3591,-180.9917 434.4173,-190.3427 431.9202,-198.2804\"/>\n",
       "</a>\n",
       "</g>\n",
       "<text text-anchor=\"middle\" x=\"432.385\" y=\"-189.4773\" font-family=\"Times,serif\" font-size=\"14.00\" fill=\"#000000\">3</text>\n",
       "</g>\n",
       "<!-- 13&#45;&#45;1 -->\n",
       "<g id=\"edge12\" class=\"edge\">\n",
       "<title>13&#45;&#45;1</title>\n",
       "<g id=\"a_edge12\"><a xlink:title=\"2\">\n",
       "<path fill=\"none\" stroke=\"#000000\" d=\"M451.4912,-138.3569C455.4943,-127.1175 460.6905,-112.5286 464.6823,-101.321\"/>\n",
       "</a>\n",
       "</g>\n",
       "<text text-anchor=\"middle\" x=\"454.5867\" y=\"-123.639\" font-family=\"Times,serif\" font-size=\"14.00\" fill=\"#000000\">2</text>\n",
       "</g>\n",
       "<!-- 16 -->\n",
       "<g id=\"node11\" class=\"node\">\n",
       "<title>16</title>\n",
       "<ellipse fill=\"none\" stroke=\"#000000\" cx=\"334.0044\" cy=\"-52\" rx=\"18\" ry=\"18\"/>\n",
       "<text text-anchor=\"middle\" x=\"334.0044\" y=\"-48.3\" font-family=\"Times,serif\" font-size=\"14.00\" fill=\"#000000\">16</text>\n",
       "</g>\n",
       "<!-- 5 -->\n",
       "<g id=\"node12\" class=\"node\">\n",
       "<title>5</title>\n",
       "<ellipse fill=\"none\" stroke=\"#000000\" cx=\"290\" cy=\"-52\" rx=\"18\" ry=\"18\"/>\n",
       "<text text-anchor=\"middle\" x=\"290\" y=\"-48.3\" font-family=\"Times,serif\" font-size=\"14.00\" fill=\"#000000\">5</text>\n",
       "</g>\n",
       "<!-- 16&#45;&#45;5 -->\n",
       "<g id=\"edge14\" class=\"edge\">\n",
       "<title>16&#45;&#45;5</title>\n",
       "<g id=\"a_edge14\"><a xlink:title=\"1\">\n",
       "<path fill=\"none\" stroke=\"#000000\" d=\"M315.8521,-52C313.2952,-52 310.6667,-52 308.1106,-52\"/>\n",
       "</a>\n",
       "</g>\n",
       "<text text-anchor=\"middle\" x=\"311.9813\" y=\"-40.8\" font-family=\"Times,serif\" font-size=\"14.00\" fill=\"#000000\">1</text>\n",
       "</g>\n",
       "<!-- 17 -->\n",
       "<g id=\"node13\" class=\"node\">\n",
       "<title>17</title>\n",
       "<ellipse fill=\"none\" stroke=\"#000000\" cx=\"411.6893\" cy=\"-308.9474\" rx=\"18\" ry=\"18\"/>\n",
       "<text text-anchor=\"middle\" x=\"411.6893\" y=\"-305.2474\" font-family=\"Times,serif\" font-size=\"14.00\" fill=\"#000000\">17</text>\n",
       "</g>\n",
       "<!-- 17&#45;&#45;0 -->\n",
       "<g id=\"edge15\" class=\"edge\">\n",
       "<title>17&#45;&#45;0</title>\n",
       "<g id=\"a_edge15\"><a xlink:title=\"1\">\n",
       "<path fill=\"none\" stroke=\"#000000\" d=\"M414.5235,-290.9997C417.1265,-274.5164 420.9632,-250.2208 423.5635,-233.755\"/>\n",
       "</a>\n",
       "</g>\n",
       "<text text-anchor=\"middle\" x=\"415.5435\" y=\"-266.1773\" font-family=\"Times,serif\" font-size=\"14.00\" fill=\"#000000\">1</text>\n",
       "</g>\n",
       "<!-- 19 -->\n",
       "<g id=\"node14\" class=\"node\">\n",
       "<title>19</title>\n",
       "<ellipse fill=\"none\" stroke=\"#000000\" cx=\"524.7026\" cy=\"-253.8635\" rx=\"18\" ry=\"18\"/>\n",
       "<text text-anchor=\"middle\" x=\"524.7026\" y=\"-250.1635\" font-family=\"Times,serif\" font-size=\"14.00\" fill=\"#000000\">19</text>\n",
       "</g>\n",
       "<!-- 19&#45;&#45;0 -->\n",
       "<g id=\"edge16\" class=\"edge\">\n",
       "<title>19&#45;&#45;0</title>\n",
       "<g id=\"a_edge16\"><a xlink:title=\"1\">\n",
       "<path fill=\"none\" stroke=\"#000000\" d=\"M507.5777,-247.2442C489.5329,-240.2694 461.2881,-229.3519 443.3177,-222.4058\"/>\n",
       "</a>\n",
       "</g>\n",
       "<text text-anchor=\"middle\" x=\"471.9477\" y=\"-223.625\" font-family=\"Times,serif\" font-size=\"14.00\" fill=\"#000000\">1</text>\n",
       "</g>\n",
       "<!-- 21 -->\n",
       "<g id=\"node15\" class=\"node\">\n",
       "<title>21</title>\n",
       "<ellipse fill=\"none\" stroke=\"#000000\" cx=\"519.5715\" cy=\"-197.5063\" rx=\"18\" ry=\"18\"/>\n",
       "<text text-anchor=\"middle\" x=\"519.5715\" y=\"-193.8063\" font-family=\"Times,serif\" font-size=\"14.00\" fill=\"#000000\">21</text>\n",
       "</g>\n",
       "<!-- 21&#45;&#45;0 -->\n",
       "<g id=\"edge17\" class=\"edge\">\n",
       "<title>21&#45;&#45;0</title>\n",
       "<g id=\"a_edge17\"><a xlink:title=\"1\">\n",
       "<path fill=\"none\" stroke=\"#000000\" d=\"M501.605,-201.0455C485.1045,-204.296 460.7834,-209.087 444.3004,-212.334\"/>\n",
       "</a>\n",
       "</g>\n",
       "<text text-anchor=\"middle\" x=\"469.4527\" y=\"-195.4898\" font-family=\"Times,serif\" font-size=\"14.00\" fill=\"#000000\">1</text>\n",
       "</g>\n",
       "<!-- 31 -->\n",
       "<g id=\"node16\" class=\"node\">\n",
       "<title>31</title>\n",
       "<ellipse fill=\"none\" stroke=\"#000000\" cx=\"108.1848\" cy=\"-97.5901\" rx=\"18\" ry=\"18\"/>\n",
       "<text text-anchor=\"middle\" x=\"108.1848\" y=\"-93.8901\" font-family=\"Times,serif\" font-size=\"14.00\" fill=\"#000000\">31</text>\n",
       "</g>\n",
       "<!-- 24 -->\n",
       "<g id=\"node17\" class=\"node\">\n",
       "<title>24</title>\n",
       "<ellipse fill=\"none\" stroke=\"#000000\" cx=\"103.9718\" cy=\"-18\" rx=\"18\" ry=\"18\"/>\n",
       "<text text-anchor=\"middle\" x=\"103.9718\" y=\"-14.3\" font-family=\"Times,serif\" font-size=\"14.00\" fill=\"#000000\">24</text>\n",
       "</g>\n",
       "<!-- 31&#45;&#45;24 -->\n",
       "<g id=\"edge18\" class=\"edge\">\n",
       "<title>31&#45;&#45;24</title>\n",
       "<g id=\"a_edge18\"><a xlink:title=\"1\">\n",
       "<path fill=\"none\" stroke=\"#000000\" d=\"M107.2288,-79.5299C106.5393,-66.5045 105.6126,-48.9977 104.924,-35.9888\"/>\n",
       "</a>\n",
       "</g>\n",
       "<text text-anchor=\"middle\" x=\"102.5764\" y=\"-61.5593\" font-family=\"Times,serif\" font-size=\"14.00\" fill=\"#000000\">1</text>\n",
       "</g>\n",
       "<!-- 32 -->\n",
       "<g id=\"node18\" class=\"node\">\n",
       "<title>32</title>\n",
       "<ellipse fill=\"none\" stroke=\"#000000\" cx=\"285.2043\" cy=\"-154.0791\" rx=\"18\" ry=\"18\"/>\n",
       "<text text-anchor=\"middle\" x=\"285.2043\" y=\"-150.3791\" font-family=\"Times,serif\" font-size=\"14.00\" fill=\"#000000\">32</text>\n",
       "</g>\n",
       "<!-- 32&#45;&#45;2 -->\n",
       "<g id=\"edge19\" class=\"edge\">\n",
       "<title>32&#45;&#45;2</title>\n",
       "<g id=\"a_edge19\"><a xlink:title=\"1\">\n",
       "<path fill=\"none\" stroke=\"#000000\" d=\"M303.365,-154.4527C316.9256,-154.7317 335.3759,-155.1112 348.9242,-155.3899\"/>\n",
       "</a>\n",
       "</g>\n",
       "<text text-anchor=\"middle\" x=\"322.6446\" y=\"-158.7213\" font-family=\"Times,serif\" font-size=\"14.00\" fill=\"#000000\">1</text>\n",
       "</g>\n",
       "<!-- 32&#45;&#45;8 -->\n",
       "<g id=\"edge20\" class=\"edge\">\n",
       "<title>32&#45;&#45;8</title>\n",
       "<g id=\"a_edge20\"><a xlink:title=\"1\">\n",
       "<path fill=\"none\" stroke=\"#000000\" d=\"M282.1997,-171.9122C280.8335,-180.0208 279.22,-189.5977 277.8503,-197.7271\"/>\n",
       "</a>\n",
       "</g>\n",
       "<text text-anchor=\"middle\" x=\"276.525\" y=\"-188.6196\" font-family=\"Times,serif\" font-size=\"14.00\" fill=\"#000000\">1</text>\n",
       "</g>\n",
       "<!-- 23 -->\n",
       "<g id=\"node19\" class=\"node\">\n",
       "<title>23</title>\n",
       "<ellipse fill=\"none\" stroke=\"#000000\" cx=\"204.5514\" cy=\"-169.8454\" rx=\"18\" ry=\"18\"/>\n",
       "<text text-anchor=\"middle\" x=\"204.5514\" y=\"-166.1454\" font-family=\"Times,serif\" font-size=\"14.00\" fill=\"#000000\">23</text>\n",
       "</g>\n",
       "<!-- 32&#45;&#45;23 -->\n",
       "<g id=\"edge21\" class=\"edge\">\n",
       "<title>32&#45;&#45;23</title>\n",
       "<g id=\"a_edge21\"><a xlink:title=\"1\">\n",
       "<path fill=\"none\" stroke=\"#000000\" d=\"M267.3048,-157.5782C253.9393,-160.1909 235.7544,-163.7457 222.4009,-166.3561\"/>\n",
       "</a>\n",
       "</g>\n",
       "<text text-anchor=\"middle\" x=\"241.3528\" y=\"-165.7672\" font-family=\"Times,serif\" font-size=\"14.00\" fill=\"#000000\">1</text>\n",
       "</g>\n",
       "<!-- 33 -->\n",
       "<g id=\"node20\" class=\"node\">\n",
       "<title>33</title>\n",
       "<ellipse fill=\"none\" stroke=\"#000000\" cx=\"122.4841\" cy=\"-214.3362\" rx=\"18\" ry=\"18\"/>\n",
       "<text text-anchor=\"middle\" x=\"122.4841\" y=\"-210.6362\" font-family=\"Times,serif\" font-size=\"14.00\" fill=\"#000000\">33</text>\n",
       "</g>\n",
       "<!-- 33&#45;&#45;8 -->\n",
       "<g id=\"edge22\" class=\"edge\">\n",
       "<title>33&#45;&#45;8</title>\n",
       "<g id=\"a_edge22\"><a xlink:title=\"2\">\n",
       "<path fill=\"none\" stroke=\"#000000\" d=\"M140.5568,-214.502C169.8745,-214.771 227.4647,-215.2993 256.7673,-215.5681\"/>\n",
       "</a>\n",
       "</g>\n",
       "<text text-anchor=\"middle\" x=\"195.1621\" y=\"-218.8351\" font-family=\"Times,serif\" font-size=\"14.00\" fill=\"#000000\">2</text>\n",
       "</g>\n",
       "<!-- 33&#45;&#45;31 -->\n",
       "<g id=\"edge33\" class=\"edge\">\n",
       "<title>33&#45;&#45;31</title>\n",
       "<g id=\"a_edge33\"><a xlink:title=\"1\">\n",
       "<path fill=\"none\" stroke=\"#000000\" d=\"M120.2812,-196.3505C117.6019,-174.4757 113.1048,-137.7592 110.4138,-115.7887\"/>\n",
       "</a>\n",
       "</g>\n",
       "<text text-anchor=\"middle\" x=\"111.8475\" y=\"-159.8696\" font-family=\"Times,serif\" font-size=\"14.00\" fill=\"#000000\">1</text>\n",
       "</g>\n",
       "<!-- 33&#45;&#45;23 -->\n",
       "<g id=\"edge28\" class=\"edge\">\n",
       "<title>33&#45;&#45;23</title>\n",
       "<g id=\"a_edge28\"><a xlink:title=\"3\">\n",
       "<path fill=\"none\" stroke=\"#000000\" d=\"M138.698,-205.5463C153.1789,-197.6958 174.2657,-186.2641 188.6505,-178.4657\"/>\n",
       "</a>\n",
       "</g>\n",
       "<text text-anchor=\"middle\" x=\"160.1742\" y=\"-195.806\" font-family=\"Times,serif\" font-size=\"14.00\" fill=\"#000000\">3</text>\n",
       "</g>\n",
       "<!-- 14 -->\n",
       "<g id=\"node21\" class=\"node\">\n",
       "<title>14</title>\n",
       "<ellipse fill=\"none\" stroke=\"#000000\" cx=\"151.6985\" cy=\"-305.475\" rx=\"18\" ry=\"18\"/>\n",
       "<text text-anchor=\"middle\" x=\"151.6985\" y=\"-301.775\" font-family=\"Times,serif\" font-size=\"14.00\" fill=\"#000000\">14</text>\n",
       "</g>\n",
       "<!-- 33&#45;&#45;14 -->\n",
       "<g id=\"edge23\" class=\"edge\">\n",
       "<title>33&#45;&#45;14</title>\n",
       "<g id=\"a_edge23\"><a xlink:title=\"1\">\n",
       "<path fill=\"none\" stroke=\"#000000\" d=\"M127.9791,-231.4787C133.1691,-247.6698 140.9144,-271.8325 146.1288,-288.0995\"/>\n",
       "</a>\n",
       "</g>\n",
       "<text text-anchor=\"middle\" x=\"133.554\" y=\"-263.5891\" font-family=\"Times,serif\" font-size=\"14.00\" fill=\"#000000\">1</text>\n",
       "</g>\n",
       "<!-- 15 -->\n",
       "<g id=\"node22\" class=\"node\">\n",
       "<title>15</title>\n",
       "<ellipse fill=\"none\" stroke=\"#000000\" cx=\"178.6274\" cy=\"-253.4403\" rx=\"18\" ry=\"18\"/>\n",
       "<text text-anchor=\"middle\" x=\"178.6274\" y=\"-249.7403\" font-family=\"Times,serif\" font-size=\"14.00\" fill=\"#000000\">15</text>\n",
       "</g>\n",
       "<!-- 33&#45;&#45;15 -->\n",
       "<g id=\"edge24\" class=\"edge\">\n",
       "<title>33&#45;&#45;15</title>\n",
       "<g id=\"a_edge24\"><a xlink:title=\"1\">\n",
       "<path fill=\"none\" stroke=\"#000000\" d=\"M137.5295,-224.8154C145.5848,-230.426 155.4789,-237.3172 163.5405,-242.9322\"/>\n",
       "</a>\n",
       "</g>\n",
       "<text text-anchor=\"middle\" x=\"147.035\" y=\"-237.6738\" font-family=\"Times,serif\" font-size=\"14.00\" fill=\"#000000\">1</text>\n",
       "</g>\n",
       "<!-- 18 -->\n",
       "<g id=\"node23\" class=\"node\">\n",
       "<title>18</title>\n",
       "<ellipse fill=\"none\" stroke=\"#000000\" cx=\"78.448\" cy=\"-253.8956\" rx=\"18\" ry=\"18\"/>\n",
       "<text text-anchor=\"middle\" x=\"78.448\" y=\"-250.1956\" font-family=\"Times,serif\" font-size=\"14.00\" fill=\"#000000\">18</text>\n",
       "</g>\n",
       "<!-- 33&#45;&#45;18 -->\n",
       "<g id=\"edge25\" class=\"edge\">\n",
       "<title>33&#45;&#45;18</title>\n",
       "<g id=\"a_edge25\"><a xlink:title=\"1\">\n",
       "<path fill=\"none\" stroke=\"#000000\" d=\"M109.0325,-226.4203C103.6703,-231.2374 97.505,-236.7759 92.1211,-241.6125\"/>\n",
       "</a>\n",
       "</g>\n",
       "<text text-anchor=\"middle\" x=\"97.0768\" y=\"-222.8164\" font-family=\"Times,serif\" font-size=\"14.00\" fill=\"#000000\">1</text>\n",
       "</g>\n",
       "<!-- 20 -->\n",
       "<g id=\"node24\" class=\"node\">\n",
       "<title>20</title>\n",
       "<ellipse fill=\"none\" stroke=\"#000000\" cx=\"95.1924\" cy=\"-315.7512\" rx=\"18\" ry=\"18\"/>\n",
       "<text text-anchor=\"middle\" x=\"95.1924\" y=\"-312.0512\" font-family=\"Times,serif\" font-size=\"14.00\" fill=\"#000000\">20</text>\n",
       "</g>\n",
       "<!-- 33&#45;&#45;20 -->\n",
       "<g id=\"edge26\" class=\"edge\">\n",
       "<title>33&#45;&#45;20</title>\n",
       "<g id=\"a_edge26\"><a xlink:title=\"1\">\n",
       "<path fill=\"none\" stroke=\"#000000\" d=\"M117.7302,-232.0014C112.721,-250.6154 104.8803,-279.7512 99.8918,-298.2885\"/>\n",
       "</a>\n",
       "</g>\n",
       "<text text-anchor=\"middle\" x=\"105.311\" y=\"-268.9449\" font-family=\"Times,serif\" font-size=\"14.00\" fill=\"#000000\">1</text>\n",
       "</g>\n",
       "<!-- 22 -->\n",
       "<g id=\"node25\" class=\"node\">\n",
       "<title>22</title>\n",
       "<ellipse fill=\"none\" stroke=\"#000000\" cx=\"18\" cy=\"-232.6777\" rx=\"18\" ry=\"18\"/>\n",
       "<text text-anchor=\"middle\" x=\"18\" y=\"-228.9777\" font-family=\"Times,serif\" font-size=\"14.00\" fill=\"#000000\">22</text>\n",
       "</g>\n",
       "<!-- 33&#45;&#45;22 -->\n",
       "<g id=\"edge27\" class=\"edge\">\n",
       "<title>33&#45;&#45;22</title>\n",
       "<g id=\"a_edge27\"><a xlink:title=\"1\">\n",
       "<path fill=\"none\" stroke=\"#000000\" d=\"M104.2844,-217.5311C85.107,-220.8975 55.0895,-226.1669 35.9912,-229.5195\"/>\n",
       "</a>\n",
       "</g>\n",
       "<text text-anchor=\"middle\" x=\"66.6378\" y=\"-212.3253\" font-family=\"Times,serif\" font-size=\"14.00\" fill=\"#000000\">1</text>\n",
       "</g>\n",
       "<!-- 26 -->\n",
       "<g id=\"node26\" class=\"node\">\n",
       "<title>26</title>\n",
       "<ellipse fill=\"none\" stroke=\"#000000\" cx=\"23.3074\" cy=\"-179.6711\" rx=\"18\" ry=\"18\"/>\n",
       "<text text-anchor=\"middle\" x=\"23.3074\" y=\"-175.9711\" font-family=\"Times,serif\" font-size=\"14.00\" fill=\"#000000\">26</text>\n",
       "</g>\n",
       "<!-- 33&#45;&#45;26 -->\n",
       "<g id=\"edge29\" class=\"edge\">\n",
       "<title>33&#45;&#45;26</title>\n",
       "<g id=\"a_edge29\"><a xlink:title=\"1\">\n",
       "<path fill=\"none\" stroke=\"#000000\" d=\"M105.2088,-208.298C87.0056,-201.9355 58.5129,-191.9765 40.3847,-185.6401\"/>\n",
       "</a>\n",
       "</g>\n",
       "<text text-anchor=\"middle\" x=\"69.2967\" y=\"-185.7691\" font-family=\"Times,serif\" font-size=\"14.00\" fill=\"#000000\">1</text>\n",
       "</g>\n",
       "<!-- 28 -->\n",
       "<g id=\"node27\" class=\"node\">\n",
       "<title>28</title>\n",
       "<ellipse fill=\"none\" stroke=\"#000000\" cx=\"34.4113\" cy=\"-292.9936\" rx=\"18\" ry=\"18\"/>\n",
       "<text text-anchor=\"middle\" x=\"34.4113\" y=\"-289.2936\" font-family=\"Times,serif\" font-size=\"14.00\" fill=\"#000000\">28</text>\n",
       "</g>\n",
       "<!-- 33&#45;&#45;28 -->\n",
       "<g id=\"edge30\" class=\"edge\">\n",
       "<title>33&#45;&#45;28</title>\n",
       "<g id=\"a_edge30\"><a xlink:title=\"1\">\n",
       "<path fill=\"none\" stroke=\"#000000\" d=\"M108.7227,-226.6265C92.1768,-241.4035 64.5788,-266.0512 48.076,-280.7897\"/>\n",
       "</a>\n",
       "</g>\n",
       "<text text-anchor=\"middle\" x=\"74.8994\" y=\"-257.5081\" font-family=\"Times,serif\" font-size=\"14.00\" fill=\"#000000\">1</text>\n",
       "</g>\n",
       "<!-- 29 -->\n",
       "<g id=\"node28\" class=\"node\">\n",
       "<title>29</title>\n",
       "<ellipse fill=\"none\" stroke=\"#000000\" cx=\"134.4528\" cy=\"-154.6749\" rx=\"18\" ry=\"18\"/>\n",
       "<text text-anchor=\"middle\" x=\"134.4528\" y=\"-150.9749\" font-family=\"Times,serif\" font-size=\"14.00\" fill=\"#000000\">29</text>\n",
       "</g>\n",
       "<!-- 33&#45;&#45;29 -->\n",
       "<g id=\"edge31\" class=\"edge\">\n",
       "<title>33&#45;&#45;29</title>\n",
       "<g id=\"a_edge31\"><a xlink:title=\"1\">\n",
       "<path fill=\"none\" stroke=\"#000000\" d=\"M126.0751,-196.4359C127.5899,-188.8847 129.3498,-180.1125 130.8643,-172.5627\"/>\n",
       "</a>\n",
       "</g>\n",
       "<text text-anchor=\"middle\" x=\"124.9697\" y=\"-173.2993\" font-family=\"Times,serif\" font-size=\"14.00\" fill=\"#000000\">1</text>\n",
       "</g>\n",
       "<!-- 30 -->\n",
       "<g id=\"node29\" class=\"node\">\n",
       "<title>30</title>\n",
       "<ellipse fill=\"none\" stroke=\"#000000\" cx=\"67.3202\" cy=\"-148.7938\" rx=\"18\" ry=\"18\"/>\n",
       "<text text-anchor=\"middle\" x=\"67.3202\" y=\"-145.0938\" font-family=\"Times,serif\" font-size=\"14.00\" fill=\"#000000\">30</text>\n",
       "</g>\n",
       "<!-- 33&#45;&#45;30 -->\n",
       "<g id=\"edge32\" class=\"edge\">\n",
       "<title>33&#45;&#45;30</title>\n",
       "<g id=\"a_edge32\"><a xlink:title=\"1\">\n",
       "<path fill=\"none\" stroke=\"#000000\" d=\"M110.7853,-200.4364C101.4238,-189.3136 88.3592,-173.7911 79.0022,-162.6736\"/>\n",
       "</a>\n",
       "</g>\n",
       "<text text-anchor=\"middle\" x=\"91.3937\" y=\"-185.355\" font-family=\"Times,serif\" font-size=\"14.00\" fill=\"#000000\">1</text>\n",
       "</g>\n",
       "</g>\n",
       "</svg>\n"
      ],
      "text/plain": [
       "<graphviz.dot.Graph at 0x7ff68467e9a0>"
      ]
     },
     "execution_count": 8,
     "metadata": {},
     "output_type": "execute_result"
    }
   ],
   "source": [
    "def sandia(A, L):\n",
    "    return L.mxm(L, mask=L)\n",
    "\n",
    "S = sandia(M, M.tril())\n",
    "print(S.reduce_int())\n",
    "draw(S, directed=False, graph_attr=dict(layout='sfdp'))"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 5,
   "metadata": {},
   "outputs": [
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "21.1 µs ± 579 ns per loop (mean ± std. dev. of 7 runs, 10000 loops each)\n"
     ]
    }
   ],
   "source": [
    "%timeit sandia(M, M.tril())"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 6,
   "metadata": {},
   "outputs": [
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "39.4 µs ± 1.44 µs per loop (mean ± std. dev. of 7 runs, 10000 loops each)\n"
     ]
    }
   ],
   "source": [
    "%timeit sandia_dot(M, M.triu(), M.tril())"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 9,
   "metadata": {},
   "outputs": [
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "32.9 µs ± 1.01 µs per loop (mean ± std. dev. of 7 runs, 10000 loops each)\n"
     ]
    }
   ],
   "source": [
    "%timeit cohen(M, M.triu(), M.tril())"
   ]
  }
 ],
 "metadata": {
  "kernelspec": {
   "display_name": "Python 3 (ipykernel)",
   "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.8.0"
  }
 },
 "nbformat": 4,
 "nbformat_minor": 4
}