topsis-pranshu-102313009 1.0.1__py3-none-any.whl → 1.0.3__py3-none-any.whl

topsis_pranshu/topsis.py

@@ -3,10 +3,12 @@ import pandas as pd
 import numpy as np
 import os
 
+
 def error(msg):
     print(f"Error: {msg}")
     sys.exit(1)
 
+
 def main():
     if len(sys.argv) != 5:
         error("Usage: topsis <InputFile> <Weights> <Impacts> <OutputFile>")
@@ -16,56 +18,80 @@ def main():
     if not os.path.exists(input_file):
         error("Input file not found")
 
+    # ---------- READ FILE (CSV / XLSX) ----------
     try:
-        df = pd.read_csv(input_file, encoding="utf-8")
-    except UnicodeDecodeError:
-        df = pd.read_csv(input_file, encoding="latin1")
+        if input_file.lower().endswith(".csv"):
+            try:
+                df = pd.read_csv(input_file, encoding="utf-8")
+            except UnicodeDecodeError:
+                df = pd.read_csv(input_file, encoding="latin1")
+
+        elif input_file.lower().endswith(".xlsx"):
+            df = pd.read_excel(input_file)
+
+        else:
+            error("Only .csv or .xlsx files are supported")
 
+    except Exception as e:
+        error(f"Unable to read input file: {e}")
+
+    # ---------- VALIDATIONS ----------
     if df.shape[1] < 3:
-        error("File must contain at least 3 columns")
+        error("Input file must contain at least 3 columns")
 
-    data = df.iloc[:, 1:]
+    # Keep first column as alternative name
+    alternatives = df.iloc[:, 0]
 
-    try:
-        data = data.astype(float)
-    except:
-        error("Non-numeric values found")
+    # Automatically select ONLY numeric columns (criteria)
+    data = df.iloc[:, 1:].select_dtypes(include=[np.number])
+
+    if data.shape[1] < 1:
+        error("No numeric criteria columns found")
 
     weights = list(map(float, weights.split(",")))
     impacts = impacts.split(",")
 
     if len(weights) != len(impacts) or len(weights) != data.shape[1]:
-        error("Weights, impacts and criteria count mismatch")
+        error(
+            f"Weights ({len(weights)}), impacts ({len(impacts)}) "
+            f"and criteria ({data.shape[1]}) count mismatch"
+        )
 
     if not all(i in ['+', '-'] for i in impacts):
         error("Impacts must be + or -")
 
+    # ---------- TOPSIS CALCULATION ----------
     norm = data / np.sqrt((data ** 2).sum())
     weighted = norm * weights
 
-    best = []
-    worst = []
+    ideal_best = []
+    ideal_worst = []
 
     for i, impact in enumerate(impacts):
         if impact == '+':
-            best.append(weighted.iloc[:, i].max())
-            worst.append(weighted.iloc[:, i].min())
+            ideal_best.append(weighted.iloc[:, i].max())
+            ideal_worst.append(weighted.iloc[:, i].min())
         else:
-            best.append(weighted.iloc[:, i].min())
-            worst.append(weighted.iloc[:, i].max())
+            ideal_best.append(weighted.iloc[:, i].min())
+            ideal_worst.append(weighted.iloc[:, i].max())
 
-    best = np.array(best)
-    worst = np.array(worst)
+    ideal_best = np.array(ideal_best)
+    ideal_worst = np.array(ideal_worst)
 
-    d_pos = np.sqrt(((weighted - best) ** 2).sum(axis=1))
-    d_neg = np.sqrt(((weighted - worst) ** 2).sum(axis=1))
+    d_pos = np.sqrt(((weighted - ideal_best) ** 2).sum(axis=1))
+    d_neg = np.sqrt(((weighted - ideal_worst) ** 2).sum(axis=1))
 
     score = d_neg / (d_pos + d_neg)
-    df["Topsis Score"] = score
-    df["Rank"] = score.rank(ascending=False).astype(int)
 
-    df.to_csv(output_file, index=False)
-    print("TOPSIS completed")
+    result = pd.DataFrame({
+        df.columns[0]: alternatives,
+        "Topsis Score": score,
+        "Rank": score.rank(ascending=False).astype(int)
+    })
+
+    result.to_csv(output_file, index=False)
+    print("TOPSIS completed successfully")
+
 
 if __name__ == "__main__":
     main()

{topsis_pranshu_102313009-1.0.1.dist-info → topsis_pranshu_102313009-1.0.3.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: topsis-pranshu-102313009
-Version: 1.0.1
+Version: 1.0.3
 Summary: TOPSIS command-line tool
 Author: Pranshu Goel
 Author-email: your@email.com

topsis_pranshu_102313009-1.0.3.dist-info/RECORD

@@ -0,0 +1,8 @@
+topsis_pranshu/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+topsis_pranshu/topsis.py,sha256=px_u0knbzcvODV6e4RY53tNdIRJYIAvXeCdWqk05k0M,2827
+topsis_pranshu_102313009-1.0.3.dist-info/licenses/LICENSE,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+topsis_pranshu_102313009-1.0.3.dist-info/METADATA,sha256=lUD3Bq2rjW39h-p150c43amkSFF91HUuYywMJFQv3nk,331
+topsis_pranshu_102313009-1.0.3.dist-info/WHEEL,sha256=wUyA8OaulRlbfwMtmQsvNngGrxQHAvkKcvRmdizlJi0,92
+topsis_pranshu_102313009-1.0.3.dist-info/entry_points.txt,sha256=jjw-fXzduRHvZc9NlWHV7aZzaHhK3En131ITtjs63Ag,54
+topsis_pranshu_102313009-1.0.3.dist-info/top_level.txt,sha256=uuGS33veOL0dkp49hZ522LxEG8fHjA3KmvAP3hNVm2M,15
+topsis_pranshu_102313009-1.0.3.dist-info/RECORD,,

topsis_pranshu_102313009-1.0.1.dist-info/RECORD

@@ -1,8 +0,0 @@
-topsis_pranshu/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-topsis_pranshu/topsis.py,sha256=cnIcW3iPZd3eN8qRkuv3QwUawXYhE6Y-QdJl2DeVjY8,1914
-topsis_pranshu_102313009-1.0.1.dist-info/licenses/LICENSE,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-topsis_pranshu_102313009-1.0.1.dist-info/METADATA,sha256=vGz5cqAlkrLHg6hkrHEzCz5BnPbBRfKw50xEx6fm3GY,331
-topsis_pranshu_102313009-1.0.1.dist-info/WHEEL,sha256=wUyA8OaulRlbfwMtmQsvNngGrxQHAvkKcvRmdizlJi0,92
-topsis_pranshu_102313009-1.0.1.dist-info/entry_points.txt,sha256=jjw-fXzduRHvZc9NlWHV7aZzaHhK3En131ITtjs63Ag,54
-topsis_pranshu_102313009-1.0.1.dist-info/top_level.txt,sha256=uuGS33veOL0dkp49hZ522LxEG8fHjA3KmvAP3hNVm2M,15
-topsis_pranshu_102313009-1.0.1.dist-info/RECORD,,