cityGraph.ipynb

{
 "cells": [
  {
   "cell_type": "code",
   "execution_count": 11,
   "metadata": {},
   "outputs": [
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "Oradea -> {'Sibiu': 151, 'Zerind': 71}\n",
      "Zerind -> {'Oradea': 71, 'Arad': 75}\n",
      "Arad -> {'Sibiu': 140, 'Zerind': 75, 'Timisoara': 118}\n",
      "Timisoara -> {'Arad': 118, 'Lugoj': 111}\n",
      "Lugoj -> {'Timisoara': 111, 'Mehadia': 70}\n",
      "Mehadia -> {'Lugoj': 70, 'Drobeta': 75}\n",
      "Drobeta -> {'Mehadia': 75, 'Craiova': 120}\n",
      "Craiova -> {'Rimnicu_Vilcea': 146, 'Drobeta': 120, 'Pitesti': 138}\n",
      "Sibiu -> {'Rimnicu_Vilcea': 80, 'Oradea': 151, 'Arad': 140, 'Fagaras': 99}\n",
      "Rimnicu_Vilcea -> {'Sibiu': 80, 'Pitesti': 97, 'Craiova': 146}\n",
      "Fagaras -> {'Sibiu': 99, 'Bucharest': 211}\n",
      "Pitesti -> {'Rimnicu_Vilcea': 97, 'Craiova': 138, 'Bucharest': 101}\n",
      "Giurgiu -> {'Bucharest': 90}\n",
      "Bucharest -> {'Fagaras': 211, 'Pitesti': 101, 'Giurgiu': 90, 'Urziceni': 85}\n",
      "Urziceni -> {'Bucharest': 85, 'Hirsova': 98, 'Vaslui': 142}\n",
      "Eforie -> {'Hirsova': 86}\n",
      "Hirsova -> {'Urziceni': 98, 'Eforie': 86, 'Vaslui': 92}\n",
      "Vaslui -> {'Urziceni': 142, 'Hirsova': 92, 'Iasi': 92}\n",
      "Iasi -> {'Vaslui': 92, 'Neamt': 87}\n",
      "Neamt -> {'Iasi': 87}\n"
     ]
    }
   ],
   "source": [
    "class Graph:\n",
    "    def __init__(self):\n",
    "        self.nodes = {}  # Dictionary to store node information\n",
    "        self.edges = {}  # Dictionary to store edge information\n",
    "\n",
    "    def addNode(self, node, coordinates):\n",
    "        if node not in self.nodes:\n",
    "            self.nodes[node] = coordinates\n",
    "            self.edges[node] = {}\n",
    "\n",
    "    def addEdge(self, node1, node2, distance):\n",
    "        if node1 in self.nodes and node2 in self.nodes:\n",
    "            self.edges[node1][node2] = distance\n",
    "            self.edges[node2][node1] = distance  \n",
    "\n",
    "    def delEdge(self, node1, node2):\n",
    "        if node1 in self.edges and node2 in self.edges[node1]:\n",
    "            del self.edges[node1][node2]\n",
    "            del self.edges[node2][node1]\n",
    "\n",
    "    def delNode(self, node):\n",
    "        if node in self.nodes:\n",
    "            del self.nodes[node]\n",
    "            del self.edges[node]\n",
    "            for neighbor in self.edges:\n",
    "                if node in self.edges[neighbor]:\n",
    "                    del self.edges[neighbor][node]\n",
    "\n",
    "    def display(self):\n",
    "        for node, neighbors in self.edges.items():\n",
    "            print(node, \"->\", neighbors)\n",
    "\n",
    "\n",
    "def read_city_coordinates(file_path):\n",
    "    city_coordinates = {}\n",
    "    with open(file_path, 'r') as file:\n",
    "        # Skip the header line\n",
    "        next(file)\n",
    "        for line in file:\n",
    "            parts = line.strip().split()\n",
    "            if len(parts) == 3:\n",
    "                city = parts[0]\n",
    "                try:\n",
    "                    latitude = float(parts[1])\n",
    "                    longitude = float(parts[2])\n",
    "                    city_coordinates[city] = (latitude, longitude)\n",
    "                except ValueError:\n",
    "                    print(f\"Invalid latitude or longitude values in line: {line}\")\n",
    "            else:\n",
    "                print(f\"Invalid line format: {line}\")\n",
    "    return city_coordinates\n",
    "\n",
    "\n",
    "# Initialize the graph\n",
    "file_path = \"city_coordinates.txt\"\n",
    "g = Graph()\n",
    "\n",
    "# Read city coordinates from the file and add nodes to the graph\n",
    "city_coordinates = read_city_coordinates(file_path)\n",
    "for city, coordinates in city_coordinates.items():\n",
    "    g.addNode(city, coordinates)\n",
    "\n",
    "# Display the graph\n",
    "\n",
    "g.addEdge('Rimnicu_Vilcea', 'Sibiu', 80)\n",
    "g.addEdge('Rimnicu_Vilcea', 'Pitesti', 97)\n",
    "g.addEdge('Rimnicu_Vilcea', 'Craiova', 146)\n",
    "\n",
    "g.addEdge('Sibiu', 'Oradea', 151)\n",
    "g.addEdge('Sibiu', 'Arad', 140)\n",
    "g.addEdge('Oradea', 'Zerind', 71)\n",
    "g.addEdge('Oradea', 'Sibiu', 151)\n",
    "\n",
    "g.addEdge('Zerind', 'Arad', 75)\n",
    "\n",
    "g.addEdge('Arad', 'Sibiu', 140)\n",
    "g.addEdge('Arad', 'Timisoara', 118)\n",
    "\n",
    "g.addEdge('Timisoara', 'Lugoj', 111)\n",
    "\n",
    "g.addEdge('Lugoj', 'Mehadia', 70)\n",
    "\n",
    "g.addEdge('Mehadia', 'Drobeta', 75)\n",
    "\n",
    "g.addEdge('Drobeta', 'Craiova', 120)\n",
    "\n",
    "g.addEdge('Craiova', 'Rimnicu_Vilcea', 146)\n",
    "g.addEdge('Craiova', 'Pitesti', 138)\n",
    "\n",
    "g.addEdge('Sibiu', 'Rimnicu_Vilcea', 80)\n",
    "g.addEdge('Sibiu', 'Fagaras', 99)\n",
    "\n",
    "g.addEdge('Rimnicu_Vilcea', 'Pitesti', 97)\n",
    "\n",
    "g.addEdge('Fagaras', 'Bucharest', 211)\n",
    "\n",
    "g.addEdge('Pitesti', 'Bucharest', 101)\n",
    "\n",
    "g.addEdge('Bucharest', 'Giurgiu', 90)\n",
    "g.addEdge('Bucharest', 'Urziceni', 85)\n",
    "\n",
    "g.addEdge('Urziceni', 'Hirsova', 98)\n",
    "g.addEdge('Urziceni', 'Vaslui', 142)\n",
    "\n",
    "g.addEdge('Eforie', 'Hirsova', 86)\n",
    "\n",
    "g.addEdge('Hirsova', 'Vaslui', 92)\n",
    "\n",
    "g.addEdge('Vaslui', 'Iasi', 92)\n",
    "\n",
    "g.addEdge('Iasi', 'Neamt', 87)\n",
    "\n",
    "\n",
    "g.display()\n"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": null,
   "metadata": {},
   "outputs": [],
   "source": []
  }
 ],
 "metadata": {
  "kernelspec": {
   "display_name": "Python 3",
   "language": "python",
   "name": "python3"
  },
  "language_info": {
   "codemirror_mode": {
    "name": "ipython",
    "version": 3
   },
   "file_extension": ".py",
   "mimetype": "text/x-python",
   "name": "python",
   "nbconvert_exporter": "python",
   "pygments_lexer": "ipython3",
   "version": "3.10.12"
  }
 },
 "nbformat": 4,
 "nbformat_minor": 2
}