Frame / column operations #27

HEL · 2026-07-03T10:29:14Z

HEL commented

2026-07-03 10:29:14 +00:00

This PR mainly adds method registries for frames, columns, and their group-by counterparts. It implements a number of methods and attributes, including arithmetic operations (+, -, *, /, etc.), comparisons (>, <, >=, <=, etc.) and aggregation methods (min, max, mean, median, etc.). The type checker will try and infer the resulting type, and fallback to a top type if it cannot properly express it. For example, ColumnType[int].mean() results in ColumnType[Any], whilst ColumnType[int].median() results in Column[int] because the median is necessarily one of the column's values.
Where possible, inference is delegated to the columns / inner types.

Element-wise operations also generate an assertion to ensure that the two operands have the same number of elements.

This PR mainly adds method registries for frames, columns, and their group-by counterparts. It implements a number of methods and attributes, including arithmetic operations (+, -, *, /, etc.), comparisons (>, <, >=, <=, etc.) and aggregation methods (min, max, mean, median, etc.). The type checker will try and infer the resulting type, and fallback to a top type if it cannot properly express it. For example, `ColumnType[int].mean()` results in `ColumnType[Any]`, whilst `ColumnType[int].median()` results in `Column[int]` because the median is necessarily one of the column's values. Where possible, inference is delegated to the columns / inner types. Element-wise operations also generate an assertion to ensure that the two operands have the same number of elements.

HEL self-assigned this 2026-07-03 10:29:14 +00:00

HEL added 35 commits 2026-07-03 10:29:14 +00:00

feat(gen): generate alias stubs a59a58d21a

feat(checker): add assertion collector ad2fabf471

fix(checker): pass AST expression to method registry f7a36f61b6

feat(gen): add Assertion class bd4d793ce0

feat(gen): add assertion collector to TypedAST 47b2dfdd73

feat(gen): materialize assertions from collector 8df01afd8c

feat(checker): add same length assertion on frames ff69b65171

safely adding two dataframes is only possible if the sizes are the same, or null values could be added dynamically to pad the shortest dataframe

feat(checker): statically evaluate casts to Any and None 955c2233ed

docs: mention eager evaluation in manual 5d20f8ec3e

feat(types): add FrameGroupBy type b5acae4078

feat(types): add ColumnGroupBy 3c92f0867d

refactor: restructure frame method registry in submodule 0d5840a4ce

refactor: make MethodRegistry generic on Call b48dfe5301

feat(checker): support unification of frames and columns 640f2d1771

feat(checker): add column method registry 2fce2f4bfc

feat(checker): add add/mean/groupby on columns 8e8ed62266

feat(checker): handle calls on group-bys b14f46d405

feat(checker): handle assignment to multiple columns 0288a05901

feat(checker): delegate element operation to columns 1bc4c704c3

delegate element-wise binary operation on frames to columns

feat(checker): delegate element operation to inner type be2fd4c837

delegate element-wise binary operation on columns to their inner types

feat(checker): add arithmetic binary ops on frames 74c07c9afb

feat(checker): add arithmetic binary ops on columns ece2e3a6a3

fix(checker): improve unknown method message 6e06f9078e

feat(checker): add comparison binary ops on frames d8c0b17512

tests: add dataframe operations test 819ab3c2bf

fix(checker): recursively check builtin subtypes 66b585c3d6

feat(checker): add comparison binary ops on columns 1f6189daa4

feat(checker): add statistical ops on frames 1c0c917873

feat(checker): add statistical ops on columns 0c70048b62

feat(checker): add aggregation ops on frame groupby a143972ef1

feat(tests): add colors and run all tests in base module 20173a0b07

feat(checker): add aggregation ops on column groupby 733c8736b8

fix(checker): delegate frame aggregate methods to columns 7a6e01cff8

feate(checker): add some frame/column attributes 4ea15519f3

feat(checker): add head/tail methods ec80b1e92e

HEL added 1 commit 2026-07-03 10:29:18 +00:00

fix(checker): correctly check length of frame/column 4d3e3f44a1

HEL merged commit c6cc38bfeb into main

2026-07-03 10:29:32 +00:00

HEL deleted branch feat/simple-frame-ops

2026-07-03 10:29:33 +00:00

HEL referenced this issue from a commit

2026-07-03 10:29:33 +00:00

Merge pull request 'Frame / column operations' (#27) from feat/simple-frame-ops into main

Sign in to join this conversation.

1 Participants

Notifications

Due Date

No due date set.

Dependencies

No dependencies set.

Reference: HEL/midas#27