script for checking whether two columns are functionally dependent

check_func_dependency.py

@@ -0,0 +1,25 @@
+import pandas as pd
+import sys
+
+def check_dep(df, col1, col2):
+    mp = {}
+    for idx, row in df.iterrows():
+        if row[col1] in mp:
+            if mp[row[col1]] != row[col2]:
+                return False
+        else:
+            mp[row[col1]] = row[col2]
+    return True
+
+def main():	
+    filename = sys.argv[1]
+    col1 = sys.argv[2]
+    col2 = sys.argv[3]
+    df = pd.read_csv(filename)
+    if check_dep(df, col1, col2):
+        print("Functional dependency:", col1, "->", col2) 
+    else:
+        print("No functional dependency:", col1, " and", col2) 
+
+if __name__ == "__main__":
+	main()