Added Shape as a different column

Simply Added it to the html with a check that the kernel is python.
I don't know if R has shapes or anything of that sort so I add the column
only if the kernel lang is python.
If an object is not an nd.array it will print "None" which can be fixed by
making the function _getshapeof(x) return "" instead of None.

Author: Tadaboody

src/jupyter_contrib_nbextensions/nbextensions/varInspector/main.js

@@ -129,17 +129,30 @@ function html_table(jsonVars) {
     var kernel_config = cfg.kernels_config[kernelLanguage];
     var varList = JSON.parse(String(jsonVars))
 
+    var shape_str;
+    if(kernelLanguage === "python")
+    {
+        shape_str = '<th >Shape</th>';
+    }
     var beg_table = '<div class=\"inspector\"><table class=\"table fixed table-condensed table-nonfluid \"><col /> \
- <col  /><col /><thead><tr><th >X</th><th >Name</th><th >Type</th><th >Size</th><th >Value</th></tr></thead><tr><td> \
+ <col  /><col /><thead><tr><th >X</th><th >Name</th><th >Type</th><th >Size</th>' + shape_str + '<th >Value</th></tr></thead><tr><td> \
  </td></tr>'
     var nb_vars = varList.length;
+    var shape_col_str;
     for (var i = 0; i < nb_vars; i++) {
+        if(kernelLanguage === "python")
+        {
+            shape_col_str = '</td><td>' + varList[i].varShape + '</td><td>';
+        }else
+        {
+            shape_col_str = '';
+        }
         beg_table = beg_table +
             '<tr><td><a href=\"#\" onClick=\"Jupyter.notebook.kernel.execute(\'' +
             kernel_config.delete_cmd_prefix + varList[i].varName + kernel_config.delete_cmd_postfix + '\'' + '); ' +
             'Jupyter.notebook.events.trigger(\'varRefresh\'); \">x</a></td>' +
             '<td>' + _trunc(varList[i].varName, cfg.cols.lenName) + '</td><td>' + _trunc(varList[i].varType, cfg.cols.lenType) +
-            '</td><td>' + varList[i].varSize + '</td><td>' + _trunc(varList[i].varContent, cfg.cols.lenVar) +
+            '</td><td>' + varList[i].varSize + shape_col_str + _trunc(varList[i].varContent, cfg.cols.lenVar) +
             '</td></tr>'
     }
     var full_table = beg_table + '</table></div>'

src/jupyter_contrib_nbextensions/nbextensions/varInspector/var_list.py

@@ -12,21 +12,26 @@
 def _getsizeof(x):
     # return the size of variable x. Amended version of sys.getsizeof
     # which also supports ndarray, Series and DataFrame
-    if type(x) is np.ndarray: 
-        return x.shape
-    elif type(x).__name__ in ['ndarray', 'Series']:
+    if type(x).__name__ in ['ndarray', 'Series']:
         return x.nbytes
     elif type(x).__name__ == 'DataFrame':
         return x.memory_usage().sum()
     else:
         return getsizeof(x)
 
+def _getshapeof(x):
+    #returns the shape of x if it has one
+    #returns None otherwise - might want to return an empty string for an empty collum
+    try:
+        return x.shape
+    except AttributeError: #x does not have a shape
+        return None
 
 def var_dic_list():
     types_to_exclude = ['module', 'function', 'builtin_function_or_method',
                         'instance', '_Feature', 'type', 'ufunc']
     values = _nms.who_ls()
-    vardic = [{'varName': v, 'varType': type(eval(v)).__name__, 'varSize': str(_getsizeof(eval(v))), 'varContent': str(eval(v))[:200]} # noqa
+    vardic = [{'varName': v, 'varType': type(eval(v)).__name__, 'varSize': str(_getsizeof(eval(v))), 'varShape': str(_getshapeof(eval(v))), 'varContent': str(eval(v))[:200]}  # noqa
     for v in values if (v not in ['_html', '_nms', 'NamespaceMagics', '_Jupyter']) & (type(eval(v)).__name__ not in types_to_exclude)] # noqa 
     return json.dumps(vardic)