cost-matrix 0.1.0__py3-none-any.whl → 0.1.1__py3-none-any.whl

cost_matrix/euclidean_matrix.py

@@ -1,5 +1,28 @@
 import numpy as np
 
 
-def euclidean(sources: np.ndarray, destinations: np.ndarray) -> np.ndarray:
-    return np.linalg.norm(sources - destinations, axis=1)
+def euclidean(sources, destinations):
+    """
+    Calculate the Euclidean distance matrix between source and destination points.
+
+    Parameters
+    ----------
+    sources : np.ndarray
+        Array of shape (n, d) containing the source points.
+    destinations : np.ndarray
+        Array of shape (m, d) containing the destination points.
+
+    Returns
+    -------
+    np.ndarray
+        Euclidean distance matrix of shape (n, m).
+    """
+    # Expand dimensions to enable broadcasting
+    sources_expanded = sources[:, np.newaxis, :]
+    destinations_expanded = destinations[np.newaxis, :, :]
+
+    # Compute differences along the last dimension
+    differences = sources_expanded - destinations_expanded
+
+    # Compute Euclidean distances using norm along the last dimension
+    return np.linalg.norm(differences, axis=2)

cost_matrix/manhattan_matrix.py

@@ -2,4 +2,28 @@ import numpy as np
 
 
 def manhattan(sources: np.ndarray, destinations: np.ndarray) -> np.ndarray:
-    return np.sum(np.abs(sources - destinations), axis=1)
+    """
+    Compute the Manhattan distance matrix between source and destination points.
+
+    Parameters
+    ----------
+    sources : np.ndarray
+        Array of shape (n, d) containing the source points.
+    destinations : np.ndarray
+        Array of shape (m, d) containing the destination points.
+
+    Returns
+    -------
+    np.ndarray
+        Manhattan distance matrix of shape (n, m).
+    """
+    # Expand dimensions to enable broadcasting
+    sources_expanded = sources[:, np.newaxis, :]
+    destinations_expanded = destinations[np.newaxis, :, :]
+
+    # Compute absolute differences along the last dimension
+    differences = np.abs(sources_expanded - destinations_expanded)
+
+    # Sum absolute differences along the last dimension to get
+    # Manhattan distance
+    return np.sum(differences, axis=2)

cost_matrix/osrm_matrix.py

@@ -11,7 +11,28 @@ def osrm(
     batch_size: int = 150,
     cost_type: str = "distances",
 ) -> np.ndarray:
-    """Compute the OSRM cost matrix between sources and destinations"""
+    """
+    Compute the OSRM cost matrix between sources and destinations
+
+    Parameters
+    ----------
+    sources : np.ndarray
+        Array of shape (n, d) containing the source points.
+    destinations : np.ndarray
+        Array of shape (m, d) containing the destination points.
+    server_address : str
+        Address of the OSRM server. Default is "http://router.project-osrm.org".
+    batch_size : int
+        Number of points to send in each request. Default is 150.
+    cost_type : str
+        Type of cost to be computed. Default is "distances". Options are
+        "distances" and "durations".
+
+    Returns
+    -------
+    np.ndarray
+        OSRM cost matrix of shape (n, m).
+    """
 
     num_sources = sources.shape[0]
     num_destinations = destinations.shape[0]

cost_matrix/spherical_matrix.py

@@ -4,7 +4,21 @@ EARTH_RADIUS_METERS = 6371000
 
 
 def spherical(sources: np.ndarray, destinations: np.ndarray) -> np.ndarray:
-    """Distance matrix using the Spherical distance"""
+    """
+    Compute the distance matrix using the Spherical distance]
+
+    Parameters
+    ----------
+    sources : np.ndarray
+        Array of shape (n, d) containing the source points.
+    destinations : np.ndarray
+        Array of shape (m, d) containing the destination points.
+
+    Returns
+    -------
+    np.ndarray
+        Spherical distance matrix of shape (n, m).
+    """
 
     sources_rad = np.radians(sources)
     destinations_rad = np.radians(destinations)

{cost_matrix-0.1.0.dist-info → cost_matrix-0.1.1.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: cost-matrix
-Version: 0.1.0
+Version: 0.1.1
 Summary: Designed to simplify the creation of cost matrices for optimization problems.
 Author: Luan
 Author-email: llvdmoraes@gmail.com
@@ -69,4 +69,5 @@ osrm_duration_matrix = cost_matrix.osrm(
     batch_size=250
 )
 print(osrm_duration_matrix)
+```
 

cost_matrix-0.1.1.dist-info/RECORD

@@ -0,0 +1,8 @@
+cost_matrix/__init__.py,sha256=9EaPsXe0SIuym34-CPmzUPSGJuBd2AkQZHqjcwlUywE,202
+cost_matrix/euclidean_matrix.py,sha256=PiOJFjalkvk1p4ktcCvFD7EeXswvsnm8f-oze1gFRlk,844
+cost_matrix/manhattan_matrix.py,sha256=OgOepYRfskarZmNNarl58YqNRw7WXYnjztWV2BbjGBI,907
+cost_matrix/osrm_matrix.py,sha256=5cCNCko-0f2LENXv2HhOD_o9G6JKHLFx19AjVxrpaQI,4953
+cost_matrix/spherical_matrix.py,sha256=9_y5XCpdAEiWcvljPX7kFYSUt0nU3Gq4DOwr7LlbhQY,1271
+cost_matrix-0.1.1.dist-info/METADATA,sha256=88TXrl5feVI1544bxFTYWTPfybVNfpHCM1O7e6RXe2I,2680
+cost_matrix-0.1.1.dist-info/WHEEL,sha256=sP946D7jFCHeNz5Iq4fL4Lu-PrWrFsgfLXbbkciIZwg,88
+cost_matrix-0.1.1.dist-info/RECORD,,

cost_matrix-0.1.0.dist-info/RECORD

@@ -1,8 +0,0 @@
-cost_matrix/__init__.py,sha256=9EaPsXe0SIuym34-CPmzUPSGJuBd2AkQZHqjcwlUywE,202
-cost_matrix/euclidean_matrix.py,sha256=a8uLh0COVb5duaKV6DMfG9HklkJX2dOARKfnnWxLxsc,155
-cost_matrix/manhattan_matrix.py,sha256=bWTTiIgvb0PDWSdfCXuffHEIgb9nNMb4w7Qwhe3EbaQ,155
-cost_matrix/osrm_matrix.py,sha256=VuuoQVRGxMjoN_eUfs9yur9lwINg8JZxB3SX-6cvBFk,4327
-cost_matrix/spherical_matrix.py,sha256=SxaihjkKyrp0den5w-VsimGcjjQEi-q-_cvxeTM3xyY,946
-cost_matrix-0.1.0.dist-info/METADATA,sha256=4TVrFT0TlklV7jcvRAJarKqi-TE2_TPMtSJJtVZTmIQ,2676
-cost_matrix-0.1.0.dist-info/WHEEL,sha256=sP946D7jFCHeNz5Iq4fL4Lu-PrWrFsgfLXbbkciIZwg,88
-cost_matrix-0.1.0.dist-info/RECORD,,