chicken-ggplot/gg-data.scm at main · iraikov/chicken-ggplot · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
;;;; gg-data.scm
;;;; Grammar of Graphics - Data Utilities
;;;;
;;;; Utilities for working with columnar data

(module gg-data
  (;; Data frame operations
   data-column
   data-n-rows
   data-columns
   data-row
   data-rows

   ;; Data conversion
   columns->rows
   rows->columns

   ;; Collection utilities
   collection-unique)

  (import scheme
          (chicken base)
          (chicken format)
          srfi-1
          yasos-collections)


  ;;; ========================================================================
  ;;; Data Format
  ;;; ========================================================================
  ;;;
  ;;; Data is represented as an association list of columns:
  ;;;   '((column-name . (values...))
  ;;;     (column-name . (values...))
  ;;;     ...)
  ;;;
  ;;; Example:
  ;;;   '((x . (1 2 3 4))
  ;;;     (y . (10 20 30 40))
  ;;;     (group . ("A" "A" "B" "B")))

  ;;; ========================================================================
  ;;; Column Operations
  ;;; ========================================================================

  (define (data-column data column-name)
    "Extract a column by name"
    (let ((entry (assoc column-name data)))
      (if entry
          (cdr entry)
          (error (sprintf "Column not found: ~a" column-name)))))

  (define (data-columns data)
    "Get list of column names"
    (map car data))

  (define (data-n-rows data)
    "Get number of rows in data"
    (if (null? data)
        0
        (size (cdar data))))

  ;;; ========================================================================
  ;;; Row Operations
  ;;; ========================================================================

  (define (data-row data index)
    "Extract a single row as an association list"
    (map (lambda (col)
           (cons (car col)
                 (elt-ref (cdr col) index)))
         data))

  (define (data-rows data)
    "Convert columnar data to list of row association lists"
    (let ((n (data-n-rows data)))
      (map (lambda (i) (data-row data i))
           (iota n))))

  ;;; ========================================================================
  ;;; Data Conversion
  ;;; ========================================================================

  (define (columns->rows data)
    "Convert columnar format to row format"
    (data-rows data))

  (define (rows->columns rows)
    "Convert row format to columnar format"
    (if (null? rows)
        '()
        (let ((column-names (map car (car rows))))
          (map (lambda (col-name)
                 (cons col-name
                       (map (lambda (row)
                              (cdr (assoc col-name row)))
                            rows)))
               column-names))))

  ;;; ========================================================================
  ;;; Collection Utilities
  ;;; ========================================================================

  (define (collection-unique coll)
    "Return list of unique elements from a collection, preserving first-occurrence order."
    (reverse (reduce (lambda (v acc)
                       (if (member v acc) acc (cons v acc)))
                     '()
                     coll)))

) ; end module