Merge pull request #30 from andyvanyperenAM/bugfix/ix_iloc

idx is a variable name and not an index
julioasotodv · Nov 7, 2019 · 8ab7d27 · 8ab7d27
2 parents b5f33c0 + 23b55d5
commit 8ab7d27
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/spark_df_profiling/base.py b/spark_df_profiling/base.py
@@ -633,9 +633,9 @@ def format_row(freq, label, extra_class=''):
                 messages.append(templates.messages[col].format(formatted_values, varname = formatters.fmt_varname(idx)))
 
         if row['type'] == 'CAT':
-            formatted_values['minifreqtable'] = freq_table(stats_object['freq'][idx], n_obs, stats_object['variables'].iloc[idx],
+            formatted_values['minifreqtable'] = freq_table(stats_object['freq'][idx], n_obs, stats_object['variables'].loc[idx],
                                                            templates.template('mini_freq_table'), templates.template('mini_freq_table_row'), 3)
-            formatted_values['freqtable'] = freq_table(stats_object['freq'][idx], n_obs, stats_object['variables'].iloc[idx],
+            formatted_values['freqtable'] = freq_table(stats_object['freq'][idx], n_obs, stats_object['variables'].loc[idx],
                                                        templates.template('freq_table'), templates.template('freq_table_row'), 20)
             if row['distinct_count'] > 50:
                 messages.append(templates.messages['HIGH_CARDINALITY'].format(formatted_values, varname = formatters.fmt_varname(idx)))