Changeset 14725 for branches/DataPreprocessing Enhancements/HeuristicLab.DataPreprocessing/3.4/Content/HistogramContent.cs

Timestamp:

03/07/17 16:55:03 (7 years ago)

Author:

mkommend

Message:

#2709: Added grouping for multi scatter plot view.

File:

• branches/DataPreprocessing Enhancements/HeuristicLab.DataPreprocessing/3.4/Content/HistogramContent.cs (modified) (2 diffs)

branches/DataPreprocessing Enhancements/HeuristicLab.DataPreprocessing/3.4/Content/HistogramContent.cs

@@ -20 +20 @@
 #endregion
 
+using System;
 using System.Drawing;
+using System.Linq;
+using HeuristicLab.Analysis;
 using HeuristicLab.Common;
 using HeuristicLab.Core;
@@ -49 +52 @@
     }
 
+    public static DataTable CreateHistogram(IFilteredPreprocessingData preprocessingData, string variableName, string groupingVariableName, DataRowVisualProperties.DataRowHistogramAggregation aggregation) {
+      {
+        var dataTable = new DataTable();
+
+        if (string.IsNullOrEmpty(groupingVariableName)) {
+          var row = PreprocessingChartContent.CreateDataRow(preprocessingData, variableName, DataRowVisualProperties.DataRowChartType.Histogram);
+          dataTable.Rows.Add(row);
+          return dataTable;
+        }
+
+        dataTable.VisualProperties.Title = variableName;
+
+        int variableIndex = preprocessingData.GetColumnIndex(variableName);
+        var variableValues = preprocessingData.GetValues<double>(variableIndex);
+        int groupVariableIndex = preprocessingData.GetColumnIndex(groupingVariableName);
+        var groupingValues = Enumerable.Empty<string>();
+
+        if (preprocessingData.VariableHasType<double>(groupVariableIndex)) {
+          groupingValues = preprocessingData.GetValues<double>(groupVariableIndex).Select(x => x.ToString());
+        } else if (preprocessingData.VariableHasType<string>(groupVariableIndex)) {
+          groupingValues = preprocessingData.GetValues<string>(groupVariableIndex);
+        } else if (preprocessingData.VariableHasType<DateTime>(groupVariableIndex)) {
+          groupingValues = preprocessingData.GetValues<DateTime>(groupVariableIndex).Select(x => x.ToString());
+        }
+
+        var groups = groupingValues.Zip(variableValues, Tuple.Create).GroupBy(t => t.Item1, t => t.Item2);
+
+        foreach (var group in groups) {
+          var classRow = new DataRow();
+          classRow.Name = group.Key;
+          classRow.VisualProperties.ChartType = DataRowVisualProperties.DataRowChartType.Histogram;
+          classRow.VisualProperties.Aggregation = aggregation;
+          classRow.Values.AddRange(group);
+          dataTable.Rows.Add(classRow);
+        }
+        return dataTable;
+      }
+    }
+
 
   }