show status when running Pandas apply

bioy_pkg/utils.py

@@ -32,6 +32,22 @@
 log = logging.getLogger(__name__)
 
 
+def apply_df_status(func, df, msg=''):
+    """
+    """
+    tmp_column = 'index_number'
+    row_count = float(len(df))
+    df[tmp_column] = xrange(int(row_count))
+    msg += ' {:.0%}\r'
+
+    def apply_func(item, msg):
+        sys.stderr.write(msg.format(item[tmp_column] / row_count))
+        return func(item)
+
+    df = df.apply(apply_func, args=[msg], axis=1)
+    return df.drop(tmp_column, axis=1)
+
+
 def flattener(iterable):
     """
     Flatten nested iterables (not strings or dict-like objects).