find files

rasbt · rasbt · commit 75c8f46f4aa4 · 2015-01-01T20:14:07.000-05:00
diff --git a/useful_scripts/find_file.py b/useful_scripts/find_file.py
@@ -0,0 +1,18 @@
+# Sebastian Raschka 2014
+#
+# A Python function to find files in a directory based on a substring search.
+
+
+import os
+
+def find_files(substring, path):
+    results = []
+    for f in os.listdir(path):
+        if substring in f:
+            results.append(os.path.join(path, f))
+    return results
+    
+# E.g.
+# find_files('Untitled', '/Users/sebastian/Desktop/')
+# returns
+# ['/Users/sebastian/Desktop/Untitled0.ipynb']
