feat: add "id" column to Titanic dataset

Safe-DS · Mar 17, 2023 · cbf46cd · cbf46cd
1 parent 53fc7ec
commit cbf46cd
Show file tree

Hide file tree

Showing 4 changed files with 1,320 additions and 1,314 deletions.
diff --git a/docs/examples/titanic.ipynb b/docs/examples/titanic.ipynb
@@ -116,12 +116,16 @@
    "execution_count": null,
    "outputs": [],
    "source": [
-    "from safeds.data.tabular import Table\n",
     "from safeds.plotting import correlation_heatmap\n",
     "\n",
-    "titanic_correlation = Table.from_columns(\n",
-    "    titanic.list_columns_with_numerical_values()\n",
-    ")\n",
+    "titanic_correlation = titanic.keep_columns([\n",
+    "    \"age\",\n",
+    "    \"siblings_spouses\",\n",
+    "    \"parents_children\",\n",
+    "    \"travel_class\",\n",
+    "    \"fare\",\n",
+    "    \"survived\"\n",
+    "])\n",
     "correlation_heatmap(titanic_correlation)"
    ],
    "metadata": {

diff --git a/src/safeds_examples/tabular/_titanic/_titanic.py b/src/safeds_examples/tabular/_titanic/_titanic.py
@@ -31,6 +31,7 @@ def describe_titanic_columns() -> Table:
 
     return Table(
         [
+            {"Name": "id", "Description": "A unique identifier"},
             {"Name": "name", "Description": "Name of the passenger"},
             {"Name": "sex", "Description": "Sex of the passenger"},
             {"Name": "age", "Description": "Age of the passenger at the time of the accident"},