Skip to content

Instantly share code, notes, and snippets.

@alonsosilvaallende

alonsosilvaallende/kuzu-add-nodes-and-edges-fancy.ipynb

Created June 29, 2024 14:36
Show Gist options
  • Save alonsosilvaallende/3340d10e8ceefed56cfb384ae9097e73 to your computer and use it in GitHub Desktop.
Save alonsosilvaallende/3340d10e8ceefed56cfb384ae9097e73 to your computer and use it in GitHub Desktop.
Download ZIP
Kuzu-add-nodes-and-edges-fancy.ipynb
Display the source blob
Display the rendered blob
Raw
kuzu-add-nodes-and-edges-fancy.ipynb
{
"nbformat": 4,
"nbformat_minor": 0,
"metadata": {
"colab": {
"provenance": [],
"authorship_tag": "ABX9TyMJ9cZ9x+WS+T7Jo0njOIwp",
"include_colab_link": true
},
"kernelspec": {
"name": "python3",
"display_name": "Python 3"
},
"language_info": {
"name": "python"
}
},
"cells": [
{
"cell_type": "markdown",
"metadata": {
"id": "view-in-github",
"colab_type": "text"
},
"source": [
"<a href=\"https://colab.research.google.com/gist/alonsosilvaallende/3340d10e8ceefed56cfb384ae9097e73/kuzu-add-nodes-and-edges-fancy.ipynb\" target=\"_parent\"><img src=\"https://colab.research.google.com/assets/colab-badge.svg\" alt=\"Open In Colab\"/></a>"
]
},
{
"cell_type": "code",
"execution_count": 1,
"metadata": {
"id": "y71JLerQJVKC"
},
"outputs": [],
"source": [
"%pip install --upgrade --quiet kuzu"
]
},
{
"cell_type": "code",
"source": [
"import kuzu"
],
"metadata": {
"id": "dwXoLi9IJX-s"
},
"execution_count": 2,
"outputs": []
},
{
"cell_type": "code",
"source": [
"import shutil\n",
"\n",
"shutil.rmtree(\"./test\", ignore_errors=True)"
],
"metadata": {
"id": "-dusTS7oJawa"
},
"execution_count": 3,
"outputs": []
},
{
"cell_type": "code",
"source": [
"# Initialize database\n",
"db = kuzu.Database(\"./test\")\n",
"conn = kuzu.Connection(db)"
],
"metadata": {
"id": "pa_CjCzaJd64"
},
"execution_count": 4,
"outputs": []
},
{
"cell_type": "code",
"source": [
"# Create schema\n",
"conn.execute(\"CREATE NODE TABLE Actor(name STRING, PRIMARY KEY (name))\")\n",
"conn.execute(\"CREATE NODE TABLE Movie(title STRING, PRIMARY KEY (title))\")\n",
"conn.execute(\"CREATE REL TABLE ActsIn(FROM Actor TO Movie, role STRING)\");"
],
"metadata": {
"id": "uvClkZW2Jhgb"
},
"execution_count": 5,
"outputs": []
},
{
"cell_type": "code",
"source": [
"# Insert data\n",
"actors = [\"Keanu Reeves\", \"Carrie-Anne Moss\", \"Edward Norton\", \"Brad Pitt\", \"Helena Bonham Carter\"]\n",
"roles = [\"Neo\", \"Trinity\", \"Narrator\", \"Tyler Durden\", \"Marla Singer\"]\n",
"movies = [\"The Matrix\", \"Fight Club\"]"
],
"metadata": {
"id": "r0TYn3mXVXp6"
},
"execution_count": 6,
"outputs": []
},
{
"cell_type": "code",
"source": [
"for actor in actors:\n",
" conn.execute(f\"CREATE (actor:Actor {{name: '{actor}'}})\")"
],
"metadata": {
"id": "WBkx5KieVdbU"
},
"execution_count": 7,
"outputs": []
},
{
"cell_type": "code",
"source": [
"for movie in movies:\n",
" conn.execute(f\"CREATE (movie:Movie {{title: '{movie}'}})\")"
],
"metadata": {
"id": "QCTheZqWVh0-"
},
"execution_count": 8,
"outputs": []
},
{
"cell_type": "code",
"source": [
"for i, zipped in enumerate(zip(actors, roles)):\n",
" actor, role = zipped\n",
" if i < 3:\n",
" conn.execute(\n",
" f\"\"\"\n",
" MATCH (a:Actor), (m:Movie) WHERE a.name = '{actor}' AND m.title = 'The Matrix'\n",
" CREATE (a)-[:ActsIn {{role: '{role}'}}]->(m)\n",
" \"\"\"\n",
" )\n",
" else:\n",
" conn.execute(\n",
" f\"\"\"\n",
" MATCH (a:Actor), (m:Movie) WHERE a.name = '{actor}' AND m.title = 'Fight Club'\n",
" CREATE (a)-[:ActsIn {{role: '{role}'}}]->(m)\n",
" \"\"\"\n",
" )"
],
"metadata": {
"id": "gI1sHKIGVmHS"
},
"execution_count": 9,
"outputs": []
},
{
"cell_type": "code",
"source": [
"# Execute Cypher query\n",
"response = conn.execute(\n",
" \"\"\"\n",
" MATCH (a:Actor)-[r:ActsIn]->(m:Movie)\n",
" RETURN a.name, m.title, r.role;\n",
" \"\"\"\n",
")"
],
"metadata": {
"id": "urbieh3eVqOt"
},
"execution_count": 10,
"outputs": []
},
{
"cell_type": "code",
"source": [
"# Show response\n",
"while response.has_next():\n",
" print(response.get_next())"
],
"metadata": {
"colab": {
"base_uri": "https://localhost:8080/"
},
"id": "8FaIf1teVuU4",
"outputId": "0760f1c4-7e63-49cf-fd29-7c8e18734780"
},
"execution_count": 11,
"outputs": [
{
"output_type": "stream",
"name": "stdout",
"text": [
"['Keanu Reeves', 'The Matrix', 'Neo']\n",
"['Carrie-Anne Moss', 'The Matrix', 'Trinity']\n",
"['Edward Norton', 'The Matrix', 'Narrator']\n",
"['Brad Pitt', 'Fight Club', 'Tyler Durden']\n",
"['Helena Bonham Carter', 'Fight Club', 'Marla Singer']\n"
]
}
]
},
{
"cell_type": "code",
"source": [
"# Get response as a pandas dataframe\n",
"df_response = response.get_as_df()\n",
"df_response"
],
"metadata": {
"colab": {
"base_uri": "https://localhost:8080/",
"height": 206
},
"id": "avaSOFxhV3Qs",
"outputId": "5ff9e375-1af3-4e29-e389-dced4ac3d74c"
},
"execution_count": 12,
"outputs": [
{
"output_type": "execute_result",
"data": {
"text/plain": [
" a.name m.title r.role\n",
"0 Keanu Reeves The Matrix Neo\n",
"1 Carrie-Anne Moss The Matrix Trinity\n",
"2 Edward Norton The Matrix Narrator\n",
"3 Brad Pitt Fight Club Tyler Durden\n",
"4 Helena Bonham Carter Fight Club Marla Singer"
],
"text/html": [
"\n",
" <div id=\"df-2b39789c-4bd2-4409-be1d-e6e3cbe5802a\" class=\"colab-df-container\">\n",
" <div>\n",
"<style scoped>\n",
" .dataframe tbody tr th:only-of-type {\n",
" vertical-align: middle;\n",
" }\n",
"\n",
" .dataframe tbody tr th {\n",
" vertical-align: top;\n",
" }\n",
"\n",
" .dataframe thead th {\n",
" text-align: right;\n",
" }\n",
"</style>\n",
"<table border=\"1\" class=\"dataframe\">\n",
" <thead>\n",
" <tr style=\"text-align: right;\">\n",
" <th></th>\n",
" <th>a.name</th>\n",
" <th>m.title</th>\n",
" <th>r.role</th>\n",
" </tr>\n",
" </thead>\n",
" <tbody>\n",
" <tr>\n",
" <th>0</th>\n",
" <td>Keanu Reeves</td>\n",
" <td>The Matrix</td>\n",
" <td>Neo</td>\n",
" </tr>\n",
" <tr>\n",
" <th>1</th>\n",
" <td>Carrie-Anne Moss</td>\n",
" <td>The Matrix</td>\n",
" <td>Trinity</td>\n",
" </tr>\n",
" <tr>\n",
" <th>2</th>\n",
" <td>Edward Norton</td>\n",
" <td>The Matrix</td>\n",
" <td>Narrator</td>\n",
" </tr>\n",
" <tr>\n",
" <th>3</th>\n",
" <td>Brad Pitt</td>\n",
" <td>Fight Club</td>\n",
" <td>Tyler Durden</td>\n",
" </tr>\n",
" <tr>\n",
" <th>4</th>\n",
" <td>Helena Bonham Carter</td>\n",
" <td>Fight Club</td>\n",
" <td>Marla Singer</td>\n",
" </tr>\n",
" </tbody>\n",
"</table>\n",
"</div>\n",
" <div class=\"colab-df-buttons\">\n",
"\n",
" <div class=\"colab-df-container\">\n",
" <button class=\"colab-df-convert\" onclick=\"convertToInteractive('df-2b39789c-4bd2-4409-be1d-e6e3cbe5802a')\"\n",
" title=\"Convert this dataframe to an interactive table.\"\n",
" style=\"display:none;\">\n",
"\n",
" <svg xmlns=\"http://www.w3.org/2000/svg\" height=\"24px\" viewBox=\"0 -960 960 960\">\n",
" <path d=\"M120-120v-720h720v720H120Zm60-500h600v-160H180v160Zm220 220h160v-160H400v160Zm0 220h160v-160H400v160ZM180-400h160v-160H180v160Zm440 0h160v-160H620v160ZM180-180h160v-160H180v160Zm440 0h160v-160H620v160Z\"/>\n",
" </svg>\n",
" </button>\n",
"\n",
" <style>\n",
" .colab-df-container {\n",
" display:flex;\n",
" gap: 12px;\n",
" }\n",
"\n",
" .colab-df-convert {\n",
" background-color: #E8F0FE;\n",
" border: none;\n",
" border-radius: 50%;\n",
" cursor: pointer;\n",
" display: none;\n",
" fill: #1967D2;\n",
" height: 32px;\n",
" padding: 0 0 0 0;\n",
" width: 32px;\n",
" }\n",
"\n",
" .colab-df-convert:hover {\n",
" background-color: #E2EBFA;\n",
" box-shadow: 0px 1px 2px rgba(60, 64, 67, 0.3), 0px 1px 3px 1px rgba(60, 64, 67, 0.15);\n",
" fill: #174EA6;\n",
" }\n",
"\n",
" .colab-df-buttons div {\n",
" margin-bottom: 4px;\n",
" }\n",
"\n",
" [theme=dark] .colab-df-convert {\n",
" background-color: #3B4455;\n",
" fill: #D2E3FC;\n",
" }\n",
"\n",
" [theme=dark] .colab-df-convert:hover {\n",
" background-color: #434B5C;\n",
" box-shadow: 0px 1px 3px 1px rgba(0, 0, 0, 0.15);\n",
" filter: drop-shadow(0px 1px 2px rgba(0, 0, 0, 0.3));\n",
" fill: #FFFFFF;\n",
" }\n",
" </style>\n",
"\n",
" <script>\n",
" const buttonEl =\n",
" document.querySelector('#df-2b39789c-4bd2-4409-be1d-e6e3cbe5802a button.colab-df-convert');\n",
" buttonEl.style.display =\n",
" google.colab.kernel.accessAllowed ? 'block' : 'none';\n",
"\n",
" async function convertToInteractive(key) {\n",
" const element = document.querySelector('#df-2b39789c-4bd2-4409-be1d-e6e3cbe5802a');\n",
" const dataTable =\n",
" await google.colab.kernel.invokeFunction('convertToInteractive',\n",
" [key], {});\n",
" if (!dataTable) return;\n",
"\n",
" const docLinkHtml = 'Like what you see? Visit the ' +\n",
" '<a target=\"_blank\" href=https://colab.research.google.com/notebooks/data_table.ipynb>data table notebook</a>'\n",
" + ' to learn more about interactive tables.';\n",
" element.innerHTML = '';\n",
" dataTable['output_type'] = 'display_data';\n",
" await google.colab.output.renderOutput(dataTable, element);\n",
" const docLink = document.createElement('div');\n",
" docLink.innerHTML = docLinkHtml;\n",
" element.appendChild(docLink);\n",
" }\n",
" </script>\n",
" </div>\n",
"\n",
"\n",
"<div id=\"df-0e6c1a30-fd4b-4ca0-a0f2-1d0c75f2af11\">\n",
" <button class=\"colab-df-quickchart\" onclick=\"quickchart('df-0e6c1a30-fd4b-4ca0-a0f2-1d0c75f2af11')\"\n",
" title=\"Suggest charts\"\n",
" style=\"display:none;\">\n",
"\n",
"<svg xmlns=\"http://www.w3.org/2000/svg\" height=\"24px\"viewBox=\"0 0 24 24\"\n",
" width=\"24px\">\n",
" <g>\n",
" <path d=\"M19 3H5c-1.1 0-2 .9-2 2v14c0 1.1.9 2 2 2h14c1.1 0 2-.9 2-2V5c0-1.1-.9-2-2-2zM9 17H7v-7h2v7zm4 0h-2V7h2v10zm4 0h-2v-4h2v4z\"/>\n",
" </g>\n",
"</svg>\n",
" </button>\n",
"\n",
"<style>\n",
" .colab-df-quickchart {\n",
" --bg-color: #E8F0FE;\n",
" --fill-color: #1967D2;\n",
" --hover-bg-color: #E2EBFA;\n",
" --hover-fill-color: #174EA6;\n",
" --disabled-fill-color: #AAA;\n",
" --disabled-bg-color: #DDD;\n",
" }\n",
"\n",
" [theme=dark] .colab-df-quickchart {\n",
" --bg-color: #3B4455;\n",
" --fill-color: #D2E3FC;\n",
" --hover-bg-color: #434B5C;\n",
" --hover-fill-color: #FFFFFF;\n",
" --disabled-bg-color: #3B4455;\n",
" --disabled-fill-color: #666;\n",
" }\n",
"\n",
" .colab-df-quickchart {\n",
" background-color: var(--bg-color);\n",
" border: none;\n",
" border-radius: 50%;\n",
" cursor: pointer;\n",
" display: none;\n",
" fill: var(--fill-color);\n",
" height: 32px;\n",
" padding: 0;\n",
" width: 32px;\n",
" }\n",
"\n",
" .colab-df-quickchart:hover {\n",
" background-color: var(--hover-bg-color);\n",
" box-shadow: 0 1px 2px rgba(60, 64, 67, 0.3), 0 1px 3px 1px rgba(60, 64, 67, 0.15);\n",
" fill: var(--button-hover-fill-color);\n",
" }\n",
"\n",
" .colab-df-quickchart-complete:disabled,\n",
" .colab-df-quickchart-complete:disabled:hover {\n",
" background-color: var(--disabled-bg-color);\n",
" fill: var(--disabled-fill-color);\n",
" box-shadow: none;\n",
" }\n",
"\n",
" .colab-df-spinner {\n",
" border: 2px solid var(--fill-color);\n",
" border-color: transparent;\n",
" border-bottom-color: var(--fill-color);\n",
" animation:\n",
" spin 1s steps(1) infinite;\n",
" }\n",
"\n",
" @keyframes spin {\n",
" 0% {\n",
" border-color: transparent;\n",
" border-bottom-color: var(--fill-color);\n",
" border-left-color: var(--fill-color);\n",
" }\n",
" 20% {\n",
" border-color: transparent;\n",
" border-left-color: var(--fill-color);\n",
" border-top-color: var(--fill-color);\n",
" }\n",
" 30% {\n",
" border-color: transparent;\n",
" border-left-color: var(--fill-color);\n",
" border-top-color: var(--fill-color);\n",
" border-right-color: var(--fill-color);\n",
" }\n",
" 40% {\n",
" border-color: transparent;\n",
" border-right-color: var(--fill-color);\n",
" border-top-color: var(--fill-color);\n",
" }\n",
" 60% {\n",
" border-color: transparent;\n",
" border-right-color: var(--fill-color);\n",
" }\n",
" 80% {\n",
" border-color: transparent;\n",
" border-right-color: var(--fill-color);\n",
" border-bottom-color: var(--fill-color);\n",
" }\n",
" 90% {\n",
" border-color: transparent;\n",
" border-bottom-color: var(--fill-color);\n",
" }\n",
" }\n",
"</style>\n",
"\n",
" <script>\n",
" async function quickchart(key) {\n",
" const quickchartButtonEl =\n",
" document.querySelector('#' + key + ' button');\n",
" quickchartButtonEl.disabled = true; // To prevent multiple clicks.\n",
" quickchartButtonEl.classList.add('colab-df-spinner');\n",
" try {\n",
" const charts = await google.colab.kernel.invokeFunction(\n",
" 'suggestCharts', [key], {});\n",
" } catch (error) {\n",
" console.error('Error during call to suggestCharts:', error);\n",
" }\n",
" quickchartButtonEl.classList.remove('colab-df-spinner');\n",
" quickchartButtonEl.classList.add('colab-df-quickchart-complete');\n",
" }\n",
" (() => {\n",
" let quickchartButtonEl =\n",
" document.querySelector('#df-0e6c1a30-fd4b-4ca0-a0f2-1d0c75f2af11 button');\n",
" quickchartButtonEl.style.display =\n",
" google.colab.kernel.accessAllowed ? 'block' : 'none';\n",
" })();\n",
" </script>\n",
"</div>\n",
"\n",
" <div id=\"id_6d27c8ac-94d9-4906-9962-1d01fccc18c8\">\n",
" <style>\n",
" .colab-df-generate {\n",
" background-color: #E8F0FE;\n",
" border: none;\n",
" border-radius: 50%;\n",
" cursor: pointer;\n",
" display: none;\n",
" fill: #1967D2;\n",
" height: 32px;\n",
" padding: 0 0 0 0;\n",
" width: 32px;\n",
" }\n",
"\n",
" .colab-df-generate:hover {\n",
" background-color: #E2EBFA;\n",
" box-shadow: 0px 1px 2px rgba(60, 64, 67, 0.3), 0px 1px 3px 1px rgba(60, 64, 67, 0.15);\n",
" fill: #174EA6;\n",
" }\n",
"\n",
" [theme=dark] .colab-df-generate {\n",
" background-color: #3B4455;\n",
" fill: #D2E3FC;\n",
" }\n",
"\n",
" [theme=dark] .colab-df-generate:hover {\n",
" background-color: #434B5C;\n",
" box-shadow: 0px 1px 3px 1px rgba(0, 0, 0, 0.15);\n",
" filter: drop-shadow(0px 1px 2px rgba(0, 0, 0, 0.3));\n",
" fill: #FFFFFF;\n",
" }\n",
" </style>\n",
" <button class=\"colab-df-generate\" onclick=\"generateWithVariable('df_response')\"\n",
" title=\"Generate code using this dataframe.\"\n",
" style=\"display:none;\">\n",
"\n",
" <svg xmlns=\"http://www.w3.org/2000/svg\" height=\"24px\"viewBox=\"0 0 24 24\"\n",
" width=\"24px\">\n",
" <path d=\"M7,19H8.4L18.45,9,17,7.55,7,17.6ZM5,21V16.75L18.45,3.32a2,2,0,0,1,2.83,0l1.4,1.43a1.91,1.91,0,0,1,.58,1.4,1.91,1.91,0,0,1-.58,1.4L9.25,21ZM18.45,9,17,7.55Zm-12,3A5.31,5.31,0,0,0,4.9,8.1,5.31,5.31,0,0,0,1,6.5,5.31,5.31,0,0,0,4.9,4.9,5.31,5.31,0,0,0,6.5,1,5.31,5.31,0,0,0,8.1,4.9,5.31,5.31,0,0,0,12,6.5,5.46,5.46,0,0,0,6.5,12Z\"/>\n",
" </svg>\n",
" </button>\n",
" <script>\n",
" (() => {\n",
" const buttonEl =\n",
" document.querySelector('#id_6d27c8ac-94d9-4906-9962-1d01fccc18c8 button.colab-df-generate');\n",
" buttonEl.style.display =\n",
" google.colab.kernel.accessAllowed ? 'block' : 'none';\n",
"\n",
" buttonEl.onclick = () => {\n",
" google.colab.notebook.generateWithVariable('df_response');\n",
" }\n",
" })();\n",
" </script>\n",
" </div>\n",
"\n",
" </div>\n",
" </div>\n"
],
"application/vnd.google.colaboratory.intrinsic+json": {
"type": "dataframe",
"variable_name": "df_response",
"summary": "{\n \"name\": \"df_response\",\n \"rows\": 5,\n \"fields\": [\n {\n \"column\": \"a.name\",\n \"properties\": {\n \"dtype\": \"string\",\n \"num_unique_values\": 5,\n \"samples\": [\n \"Carrie-Anne Moss\",\n \"Helena Bonham Carter\",\n \"Edward Norton\"\n ],\n \"semantic_type\": \"\",\n \"description\": \"\"\n }\n },\n {\n \"column\": \"m.title\",\n \"properties\": {\n \"dtype\": \"category\",\n \"num_unique_values\": 2,\n \"samples\": [\n \"Fight Club\",\n \"The Matrix\"\n ],\n \"semantic_type\": \"\",\n \"description\": \"\"\n }\n },\n {\n \"column\": \"r.role\",\n \"properties\": {\n \"dtype\": \"string\",\n \"num_unique_values\": 5,\n \"samples\": [\n \"Trinity\",\n \"Marla Singer\"\n ],\n \"semantic_type\": \"\",\n \"description\": \"\"\n }\n }\n ]\n}"
}
},
"metadata": {},
"execution_count": 12
}
]
}
]
}
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment