polars-runtime-compat 1.34.0b3__cp39-abi3-macosx_11_0_arm64.whl → 1.34.0b5__cp39-abi3-macosx_11_0_arm64.whl

This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.

Potentially problematic release.


This version of polars-runtime-compat might be problematic. Click here for more details.

Files changed (204) hide show
  1. _polars_runtime_compat/_polars_runtime_compat.abi3.so +0 -0
  2. polars_runtime_compat-1.34.0b5.dist-info/METADATA +35 -0
  3. polars_runtime_compat-1.34.0b5.dist-info/RECORD +6 -0
  4. polars/__init__.py +0 -528
  5. polars/_cpu_check.py +0 -265
  6. polars/_dependencies.py +0 -355
  7. polars/_plr.py +0 -99
  8. polars/_plr.pyi +0 -2496
  9. polars/_reexport.py +0 -23
  10. polars/_typing.py +0 -478
  11. polars/_utils/__init__.py +0 -37
  12. polars/_utils/async_.py +0 -102
  13. polars/_utils/cache.py +0 -176
  14. polars/_utils/cloud.py +0 -40
  15. polars/_utils/constants.py +0 -29
  16. polars/_utils/construction/__init__.py +0 -46
  17. polars/_utils/construction/dataframe.py +0 -1397
  18. polars/_utils/construction/other.py +0 -72
  19. polars/_utils/construction/series.py +0 -560
  20. polars/_utils/construction/utils.py +0 -118
  21. polars/_utils/convert.py +0 -224
  22. polars/_utils/deprecation.py +0 -406
  23. polars/_utils/getitem.py +0 -457
  24. polars/_utils/logging.py +0 -11
  25. polars/_utils/nest_asyncio.py +0 -264
  26. polars/_utils/parquet.py +0 -15
  27. polars/_utils/parse/__init__.py +0 -12
  28. polars/_utils/parse/expr.py +0 -242
  29. polars/_utils/polars_version.py +0 -19
  30. polars/_utils/pycapsule.py +0 -53
  31. polars/_utils/scan.py +0 -27
  32. polars/_utils/serde.py +0 -63
  33. polars/_utils/slice.py +0 -215
  34. polars/_utils/udfs.py +0 -1251
  35. polars/_utils/unstable.py +0 -63
  36. polars/_utils/various.py +0 -782
  37. polars/_utils/wrap.py +0 -25
  38. polars/api.py +0 -370
  39. polars/catalog/__init__.py +0 -0
  40. polars/catalog/unity/__init__.py +0 -19
  41. polars/catalog/unity/client.py +0 -733
  42. polars/catalog/unity/models.py +0 -152
  43. polars/config.py +0 -1571
  44. polars/convert/__init__.py +0 -25
  45. polars/convert/general.py +0 -1046
  46. polars/convert/normalize.py +0 -261
  47. polars/dataframe/__init__.py +0 -5
  48. polars/dataframe/_html.py +0 -186
  49. polars/dataframe/frame.py +0 -12582
  50. polars/dataframe/group_by.py +0 -1067
  51. polars/dataframe/plotting.py +0 -257
  52. polars/datatype_expr/__init__.py +0 -5
  53. polars/datatype_expr/array.py +0 -56
  54. polars/datatype_expr/datatype_expr.py +0 -304
  55. polars/datatype_expr/list.py +0 -18
  56. polars/datatype_expr/struct.py +0 -69
  57. polars/datatypes/__init__.py +0 -122
  58. polars/datatypes/_parse.py +0 -195
  59. polars/datatypes/_utils.py +0 -48
  60. polars/datatypes/classes.py +0 -1213
  61. polars/datatypes/constants.py +0 -11
  62. polars/datatypes/constructor.py +0 -172
  63. polars/datatypes/convert.py +0 -366
  64. polars/datatypes/group.py +0 -130
  65. polars/exceptions.py +0 -230
  66. polars/expr/__init__.py +0 -7
  67. polars/expr/array.py +0 -964
  68. polars/expr/binary.py +0 -346
  69. polars/expr/categorical.py +0 -306
  70. polars/expr/datetime.py +0 -2620
  71. polars/expr/expr.py +0 -11272
  72. polars/expr/list.py +0 -1408
  73. polars/expr/meta.py +0 -444
  74. polars/expr/name.py +0 -321
  75. polars/expr/string.py +0 -3045
  76. polars/expr/struct.py +0 -357
  77. polars/expr/whenthen.py +0 -185
  78. polars/functions/__init__.py +0 -193
  79. polars/functions/aggregation/__init__.py +0 -33
  80. polars/functions/aggregation/horizontal.py +0 -298
  81. polars/functions/aggregation/vertical.py +0 -341
  82. polars/functions/as_datatype.py +0 -848
  83. polars/functions/business.py +0 -138
  84. polars/functions/col.py +0 -384
  85. polars/functions/datatype.py +0 -121
  86. polars/functions/eager.py +0 -524
  87. polars/functions/escape_regex.py +0 -29
  88. polars/functions/lazy.py +0 -2751
  89. polars/functions/len.py +0 -68
  90. polars/functions/lit.py +0 -210
  91. polars/functions/random.py +0 -22
  92. polars/functions/range/__init__.py +0 -19
  93. polars/functions/range/_utils.py +0 -15
  94. polars/functions/range/date_range.py +0 -303
  95. polars/functions/range/datetime_range.py +0 -370
  96. polars/functions/range/int_range.py +0 -348
  97. polars/functions/range/linear_space.py +0 -311
  98. polars/functions/range/time_range.py +0 -287
  99. polars/functions/repeat.py +0 -301
  100. polars/functions/whenthen.py +0 -353
  101. polars/interchange/__init__.py +0 -10
  102. polars/interchange/buffer.py +0 -77
  103. polars/interchange/column.py +0 -190
  104. polars/interchange/dataframe.py +0 -230
  105. polars/interchange/from_dataframe.py +0 -328
  106. polars/interchange/protocol.py +0 -303
  107. polars/interchange/utils.py +0 -170
  108. polars/io/__init__.py +0 -64
  109. polars/io/_utils.py +0 -317
  110. polars/io/avro.py +0 -49
  111. polars/io/clipboard.py +0 -36
  112. polars/io/cloud/__init__.py +0 -17
  113. polars/io/cloud/_utils.py +0 -80
  114. polars/io/cloud/credential_provider/__init__.py +0 -17
  115. polars/io/cloud/credential_provider/_builder.py +0 -520
  116. polars/io/cloud/credential_provider/_providers.py +0 -618
  117. polars/io/csv/__init__.py +0 -9
  118. polars/io/csv/_utils.py +0 -38
  119. polars/io/csv/batched_reader.py +0 -142
  120. polars/io/csv/functions.py +0 -1495
  121. polars/io/database/__init__.py +0 -6
  122. polars/io/database/_arrow_registry.py +0 -70
  123. polars/io/database/_cursor_proxies.py +0 -147
  124. polars/io/database/_executor.py +0 -578
  125. polars/io/database/_inference.py +0 -314
  126. polars/io/database/_utils.py +0 -144
  127. polars/io/database/functions.py +0 -516
  128. polars/io/delta.py +0 -499
  129. polars/io/iceberg/__init__.py +0 -3
  130. polars/io/iceberg/_utils.py +0 -697
  131. polars/io/iceberg/dataset.py +0 -556
  132. polars/io/iceberg/functions.py +0 -151
  133. polars/io/ipc/__init__.py +0 -8
  134. polars/io/ipc/functions.py +0 -514
  135. polars/io/json/__init__.py +0 -3
  136. polars/io/json/read.py +0 -101
  137. polars/io/ndjson.py +0 -332
  138. polars/io/parquet/__init__.py +0 -17
  139. polars/io/parquet/field_overwrites.py +0 -140
  140. polars/io/parquet/functions.py +0 -722
  141. polars/io/partition.py +0 -491
  142. polars/io/plugins.py +0 -187
  143. polars/io/pyarrow_dataset/__init__.py +0 -5
  144. polars/io/pyarrow_dataset/anonymous_scan.py +0 -109
  145. polars/io/pyarrow_dataset/functions.py +0 -79
  146. polars/io/scan_options/__init__.py +0 -5
  147. polars/io/scan_options/_options.py +0 -59
  148. polars/io/scan_options/cast_options.py +0 -126
  149. polars/io/spreadsheet/__init__.py +0 -6
  150. polars/io/spreadsheet/_utils.py +0 -52
  151. polars/io/spreadsheet/_write_utils.py +0 -647
  152. polars/io/spreadsheet/functions.py +0 -1323
  153. polars/lazyframe/__init__.py +0 -9
  154. polars/lazyframe/engine_config.py +0 -61
  155. polars/lazyframe/frame.py +0 -8564
  156. polars/lazyframe/group_by.py +0 -669
  157. polars/lazyframe/in_process.py +0 -42
  158. polars/lazyframe/opt_flags.py +0 -333
  159. polars/meta/__init__.py +0 -14
  160. polars/meta/build.py +0 -33
  161. polars/meta/index_type.py +0 -27
  162. polars/meta/thread_pool.py +0 -50
  163. polars/meta/versions.py +0 -120
  164. polars/ml/__init__.py +0 -0
  165. polars/ml/torch.py +0 -213
  166. polars/ml/utilities.py +0 -30
  167. polars/plugins.py +0 -155
  168. polars/py.typed +0 -0
  169. polars/pyproject.toml +0 -103
  170. polars/schema.py +0 -265
  171. polars/selectors.py +0 -3117
  172. polars/series/__init__.py +0 -5
  173. polars/series/array.py +0 -776
  174. polars/series/binary.py +0 -254
  175. polars/series/categorical.py +0 -246
  176. polars/series/datetime.py +0 -2275
  177. polars/series/list.py +0 -1087
  178. polars/series/plotting.py +0 -191
  179. polars/series/series.py +0 -9197
  180. polars/series/string.py +0 -2367
  181. polars/series/struct.py +0 -154
  182. polars/series/utils.py +0 -191
  183. polars/sql/__init__.py +0 -7
  184. polars/sql/context.py +0 -677
  185. polars/sql/functions.py +0 -139
  186. polars/string_cache.py +0 -185
  187. polars/testing/__init__.py +0 -13
  188. polars/testing/asserts/__init__.py +0 -9
  189. polars/testing/asserts/frame.py +0 -231
  190. polars/testing/asserts/series.py +0 -219
  191. polars/testing/asserts/utils.py +0 -12
  192. polars/testing/parametric/__init__.py +0 -33
  193. polars/testing/parametric/profiles.py +0 -107
  194. polars/testing/parametric/strategies/__init__.py +0 -22
  195. polars/testing/parametric/strategies/_utils.py +0 -14
  196. polars/testing/parametric/strategies/core.py +0 -615
  197. polars/testing/parametric/strategies/data.py +0 -452
  198. polars/testing/parametric/strategies/dtype.py +0 -436
  199. polars/testing/parametric/strategies/legacy.py +0 -169
  200. polars/type_aliases.py +0 -24
  201. polars_runtime_compat-1.34.0b3.dist-info/METADATA +0 -190
  202. polars_runtime_compat-1.34.0b3.dist-info/RECORD +0 -203
  203. {polars_runtime_compat-1.34.0b3.dist-info → polars_runtime_compat-1.34.0b5.dist-info}/WHEEL +0 -0
  204. {polars_runtime_compat-1.34.0b3.dist-info → polars_runtime_compat-1.34.0b5.dist-info}/licenses/LICENSE +0 -0
polars/_plr.pyi DELETED
@@ -1,2496 +0,0 @@
1
- from collections.abc import Callable, Sequence
2
- from typing import Any, Literal, overload
3
-
4
- from numpy.typing import NDArray
5
- from typing_extensions import TypeAlias
6
-
7
- # This file mirrors all the definitions made in the polars-python Rust API.
8
-
9
- __version__: str
10
- __build__: Any
11
- _ir_nodes: Any
12
- _allocator: Any
13
- _debug: bool
14
- RUNTIME_REPR: str
15
-
16
- CompatLevel: TypeAlias = int | bool
17
- BufferInfo: TypeAlias = tuple[int, int, int]
18
- UnicodeForm: TypeAlias = Literal["NFC", "NFKC", "NFD", "NFKD"]
19
- KeyValueMetadata: TypeAlias = Sequence[tuple[str, str]] | Any
20
- TimeZone: TypeAlias = str | None
21
- UpcastOrForbid: TypeAlias = Literal["upcast", "forbid"]
22
- ExtraColumnsPolicy: TypeAlias = Literal["ignore", "raise"]
23
- MissingColumnsPolicy: TypeAlias = Literal["insert", "raise"]
24
- MissingColumnsPolicyOrExpr: TypeAlias = Literal["insert", "raise"] | Any
25
- ColumnMapping: TypeAlias = Any
26
- DeletionFilesList: TypeAlias = Any
27
- DefaultFieldValues: TypeAlias = Any
28
- Path: TypeAlias = str | Any
29
- Schema: TypeAlias = Any
30
- NullValues: TypeAlias = Any
31
- DataType: TypeAlias = Any
32
- SyncOnCloseType: TypeAlias = Literal["none", "data", "all"]
33
- SinkOptions: TypeAlias = dict[str, Any]
34
- SinkTarget: TypeAlias = str | Any | PyPartitioning
35
- AsofStrategy: TypeAlias = Literal["backward", "forward", "nearest"]
36
- InterpolationMethod: TypeAlias = Literal["linear", "nearest"]
37
- AvroCompression: TypeAlias = Literal["uncompressed", "snappy", "deflate"]
38
- CategoricalOrdering: TypeAlias = Literal["physical", "lexical"]
39
- StartBy: TypeAlias = Literal[
40
- "window",
41
- "datapoint",
42
- "monday",
43
- "tuesday",
44
- "wednesday",
45
- "thursday",
46
- "friday",
47
- "saturday",
48
- "sunday",
49
- ]
50
- ClosedWindow: TypeAlias = Literal["left", "right", "both", "none"]
51
- RoundMode: TypeAlias = Literal["half_to_even", "half_away_from_zero"]
52
- CsvEncoding: TypeAlias = Literal["utf8", "utf8-lossy"]
53
- IpcCompression: TypeAlias = Literal["uncompressed", "lz4", "zstd"]
54
- JoinType: TypeAlias = Literal["inner", "left", "right", "full", "semi", "anti", "cross"]
55
- Label: TypeAlias = Literal["left", "right", "datapoint"]
56
- ListToStructWidthStrategy: TypeAlias = Literal["first_non_null", "max_width"]
57
- NonExistent: TypeAlias = Literal["null", "raise"]
58
- NullBehavior: TypeAlias = Literal["drop", "ignore"]
59
- NullStrategy: TypeAlias = Literal["ignore", "propagate"]
60
- ParallelStrategy: TypeAlias = Literal[
61
- "auto", "columns", "row_groups", "prefiltered", "none"
62
- ]
63
- IndexOrder: TypeAlias = Literal["fortran", "c"]
64
- QuantileMethod: TypeAlias = Literal[
65
- "lower", "higher", "nearest", "linear", "midpoint", "equiprobable"
66
- ]
67
- RankMethod: TypeAlias = Literal["min", "max", "average", "dense", "ordinal", "random"]
68
- Roll: TypeAlias = Literal["raise", "forward", "backward"]
69
- TimeUnit: TypeAlias = Literal["ns", "us", "ms"]
70
- UniqueKeepStrategy: TypeAlias = Literal["first", "last", "any", "none"]
71
- SearchSortedSide: TypeAlias = Literal["any", "left", "right"]
72
- ClosedInterval: TypeAlias = Literal["both", "left", "right", "none"]
73
- WindowMapping: TypeAlias = Literal["group_to_rows", "join", "explode"]
74
- JoinValidation: TypeAlias = Literal["m:m", "m:1", "1:m", "1:1"]
75
- MaintainOrderJoin: TypeAlias = Literal[
76
- "none", "left", "right", "left_right", "right_left"
77
- ]
78
- QuoteStyle: TypeAlias = Literal["always", "necessary", "non_numeric", "never"]
79
- SetOperation: TypeAlias = Literal[
80
- "union", "difference", "intersection", "symmetric_difference"
81
- ]
82
- FloatFmt: TypeAlias = Literal["full", "mixed"]
83
- NDArray1D: TypeAlias = NDArray[Any]
84
- ParquetFieldOverwrites: TypeAlias = Any
85
- StatisticsOptions: TypeAlias = Any
86
- EngineType: TypeAlias = Literal["auto", "in-memory", "streaming", "gpu"]
87
- PyScanOptions: TypeAlias = Any
88
-
89
- # exceptions
90
- class PolarsError(Exception): ...
91
- class ColumnNotFoundError(PolarsError): ...
92
- class ComputeError(PolarsError): ...
93
- class DuplicateError(PolarsError): ...
94
- class InvalidOperationError(PolarsError): ...
95
- class NoDataError(PolarsError): ...
96
- class OutOfBoundsError(PolarsError): ...
97
- class SQLInterfaceError(PolarsError): ...
98
- class SQLSyntaxError(PolarsError): ...
99
- class SchemaError(PolarsError): ...
100
- class SchemaFieldNotFoundError(PolarsError): ...
101
- class ShapeError(PolarsError): ...
102
- class StringCacheMismatchError(PolarsError): ...
103
- class StructFieldNotFoundError(PolarsError): ...
104
- class PolarsWarning(Warning): ...
105
- class PerformanceWarning(PolarsWarning): ...
106
- class CategoricalRemappingWarning(PerformanceWarning): ...
107
- class MapWithoutReturnDtypeWarning(PolarsWarning): ...
108
- class PanicException(PolarsError): ...
109
-
110
- class PySeries:
111
- # map
112
- def map_elements(
113
- self, function: Any, return_dtype: Any | None, skip_nulls: bool
114
- ) -> PySeries: ...
115
-
116
- # general
117
- def struct_unnest(self) -> PyDataFrame: ...
118
- def struct_fields(self) -> list[str]: ...
119
- def is_sorted_ascending_flag(self) -> bool: ...
120
- def is_sorted_descending_flag(self) -> bool: ...
121
- def can_fast_explode_flag(self) -> bool: ...
122
- def cat_uses_lexical_ordering(self) -> bool: ...
123
- def cat_is_local(self) -> bool: ...
124
- def cat_to_local(self) -> PySeries: ...
125
- def estimated_size(self) -> int: ...
126
- def get_object(self, index: int) -> Any: ...
127
- def reshape(self, dims: Sequence[int]) -> PySeries: ...
128
- def get_fmt(self, index: int, str_len_limit: int) -> str: ...
129
- def rechunk(self, in_place: bool) -> PySeries | None: ...
130
- def get_index(self, index: int) -> Any: ...
131
- def get_index_signed(self, index: int) -> Any: ...
132
- def bitand(self, other: PySeries) -> PySeries: ...
133
- def bitor(self, other: PySeries) -> PySeries: ...
134
- def bitxor(self, other: PySeries) -> PySeries: ...
135
- def chunk_lengths(self) -> list[int]: ...
136
- def name(self) -> str: ...
137
- def rename(self, name: str) -> None: ...
138
- def dtype(self) -> Any: ...
139
- def set_sorted_flag(self, descending: bool) -> PySeries: ...
140
- def n_chunks(self) -> int: ...
141
- def append(self, other: PySeries) -> None: ...
142
- def extend(self, other: PySeries) -> None: ...
143
- def new_from_index(self, index: int, length: int) -> PySeries: ...
144
- def filter(self, filter: PySeries) -> PySeries: ...
145
- def sort(
146
- self, descending: bool, nulls_last: bool, multithreaded: bool
147
- ) -> PySeries: ...
148
- def gather_with_series(self, indices: PySeries) -> PySeries: ...
149
- def null_count(self) -> int: ...
150
- def has_nulls(self) -> bool: ...
151
- def equals(
152
- self, other: PySeries, check_dtypes: bool, check_names: bool, null_equal: bool
153
- ) -> bool: ...
154
- def as_str(self) -> str: ...
155
- def len(self) -> int: ...
156
- def as_single_ptr(self) -> int: ...
157
- def clone(self) -> PySeries: ...
158
- def zip_with(self, mask: PySeries, other: PySeries) -> PySeries: ...
159
- def to_dummies(
160
- self, separator: str | None, drop_first: bool, drop_nulls: bool
161
- ) -> PyDataFrame: ...
162
- def get_list(self, index: int) -> PySeries | None: ...
163
- def n_unique(self) -> int: ...
164
- def floor(self) -> PySeries: ...
165
- def shrink_to_fit(self) -> None: ...
166
- def dot(self, other: PySeries) -> Any: ...
167
- def __getstate__(self) -> bytes: ...
168
- def __setstate__(self, state: bytes) -> None: ...
169
- def skew(self, bias: bool) -> float | None: ...
170
- def kurtosis(self, fisher: bool, bias: bool) -> float | None: ...
171
- def cast(self, dtype: Any, strict: bool, wrap_numerical: bool) -> PySeries: ...
172
- def get_chunks(self) -> list[Any]: ...
173
- def is_sorted(self, descending: bool, nulls_last: bool) -> bool: ...
174
- def clear(self) -> PySeries: ...
175
- def head(self, n: int) -> PySeries: ...
176
- def tail(self, n: int) -> PySeries: ...
177
- def value_counts(
178
- self, sort: bool, parallel: bool, name: str, normalize: bool
179
- ) -> PyDataFrame: ...
180
- def slice(self, offset: int, length: int | None) -> PySeries: ...
181
- def not_(self) -> PySeries: ...
182
- def shrink_dtype(self) -> PySeries: ...
183
- def str_to_datetime_infer(
184
- self,
185
- time_unit: TimeUnit | None,
186
- strict: bool,
187
- exact: bool,
188
- ambiguous: PySeries,
189
- ) -> PySeries: ...
190
- def str_to_decimal_infer(self, inference_length: int) -> PySeries: ...
191
- def list_to_struct(
192
- self, width_strat: ListToStructWidthStrategy, name_gen: Any | None
193
- ) -> PySeries: ...
194
- def str_json_decode(self, infer_schema_length: int | None) -> PySeries: ...
195
-
196
- # aggregations
197
- def any(self, ignore_nulls: bool) -> bool | None: ...
198
- def all(self, ignore_nulls: bool) -> bool | None: ...
199
- def arg_max(self) -> int | None: ...
200
- def arg_min(self) -> int | None: ...
201
- def min(self) -> Any: ...
202
- def max(self) -> Any: ...
203
- def mean(self) -> Any: ...
204
- def median(self) -> Any: ...
205
- def product(self) -> Any: ...
206
- def quantile(self, quantile: float, interpolation: QuantileMethod) -> Any: ...
207
- def std(self, ddof: int) -> Any: ...
208
- def var(self, ddof: int) -> Any: ...
209
- def sum(self) -> Any: ...
210
- def first(self) -> Any: ...
211
- def last(self) -> Any: ...
212
- def approx_n_unique(self) -> int: ...
213
- def bitwise_and(self) -> Any: ...
214
- def bitwise_or(self) -> Any: ...
215
- def bitwise_xor(self) -> Any: ...
216
-
217
- # arithmetic
218
- # Operations with another PySeries
219
- def add(self, other: PySeries) -> PySeries: ...
220
- def sub(self, other: PySeries) -> PySeries: ...
221
- def mul(self, other: PySeries) -> PySeries: ...
222
- def div(self, other: PySeries) -> PySeries: ...
223
- def rem(self, other: PySeries) -> PySeries: ...
224
-
225
- # Operations with integer/float/datetime/duration scalars
226
- def add_u8(self, other: int) -> PySeries: ...
227
- def add_u16(self, other: int) -> PySeries: ...
228
- def add_u32(self, other: int) -> PySeries: ...
229
- def add_u64(self, other: int) -> PySeries: ...
230
- def add_i8(self, other: int) -> PySeries: ...
231
- def add_i16(self, other: int) -> PySeries: ...
232
- def add_i32(self, other: int) -> PySeries: ...
233
- def add_i64(self, other: int) -> PySeries: ...
234
- def add_datetime(self, other: int) -> PySeries: ...
235
- def add_duration(self, other: int) -> PySeries: ...
236
- def add_f32(self, other: float) -> PySeries: ...
237
- def add_f64(self, other: float) -> PySeries: ...
238
- def sub_u8(self, other: int) -> PySeries: ...
239
- def sub_u16(self, other: int) -> PySeries: ...
240
- def sub_u32(self, other: int) -> PySeries: ...
241
- def sub_u64(self, other: int) -> PySeries: ...
242
- def sub_i8(self, other: int) -> PySeries: ...
243
- def sub_i16(self, other: int) -> PySeries: ...
244
- def sub_i32(self, other: int) -> PySeries: ...
245
- def sub_i64(self, other: int) -> PySeries: ...
246
- def sub_datetime(self, other: int) -> PySeries: ...
247
- def sub_duration(self, other: int) -> PySeries: ...
248
- def sub_f32(self, other: float) -> PySeries: ...
249
- def sub_f64(self, other: float) -> PySeries: ...
250
- def div_u8(self, other: int) -> PySeries: ...
251
- def div_u16(self, other: int) -> PySeries: ...
252
- def div_u32(self, other: int) -> PySeries: ...
253
- def div_u64(self, other: int) -> PySeries: ...
254
- def div_i8(self, other: int) -> PySeries: ...
255
- def div_i16(self, other: int) -> PySeries: ...
256
- def div_i32(self, other: int) -> PySeries: ...
257
- def div_i64(self, other: int) -> PySeries: ...
258
- def div_f32(self, other: float) -> PySeries: ...
259
- def div_f64(self, other: float) -> PySeries: ...
260
- def mul_u8(self, other: int) -> PySeries: ...
261
- def mul_u16(self, other: int) -> PySeries: ...
262
- def mul_u32(self, other: int) -> PySeries: ...
263
- def mul_u64(self, other: int) -> PySeries: ...
264
- def mul_i8(self, other: int) -> PySeries: ...
265
- def mul_i16(self, other: int) -> PySeries: ...
266
- def mul_i32(self, other: int) -> PySeries: ...
267
- def mul_i64(self, other: int) -> PySeries: ...
268
- def mul_f32(self, other: float) -> PySeries: ...
269
- def mul_f64(self, other: float) -> PySeries: ...
270
- def rem_u8(self, other: int) -> PySeries: ...
271
- def rem_u16(self, other: int) -> PySeries: ...
272
- def rem_u32(self, other: int) -> PySeries: ...
273
- def rem_u64(self, other: int) -> PySeries: ...
274
- def rem_i8(self, other: int) -> PySeries: ...
275
- def rem_i16(self, other: int) -> PySeries: ...
276
- def rem_i32(self, other: int) -> PySeries: ...
277
- def rem_i64(self, other: int) -> PySeries: ...
278
- def rem_f32(self, other: float) -> PySeries: ...
279
- def rem_f64(self, other: float) -> PySeries: ...
280
-
281
- # Reverse operations (rhs)
282
- def add_u8_rhs(self, other: int) -> PySeries: ...
283
- def add_u16_rhs(self, other: int) -> PySeries: ...
284
- def add_u32_rhs(self, other: int) -> PySeries: ...
285
- def add_u64_rhs(self, other: int) -> PySeries: ...
286
- def add_i8_rhs(self, other: int) -> PySeries: ...
287
- def add_i16_rhs(self, other: int) -> PySeries: ...
288
- def add_i32_rhs(self, other: int) -> PySeries: ...
289
- def add_i64_rhs(self, other: int) -> PySeries: ...
290
- def add_f32_rhs(self, other: float) -> PySeries: ...
291
- def add_f64_rhs(self, other: float) -> PySeries: ...
292
- def sub_u8_rhs(self, other: int) -> PySeries: ...
293
- def sub_u16_rhs(self, other: int) -> PySeries: ...
294
- def sub_u32_rhs(self, other: int) -> PySeries: ...
295
- def sub_u64_rhs(self, other: int) -> PySeries: ...
296
- def sub_i8_rhs(self, other: int) -> PySeries: ...
297
- def sub_i16_rhs(self, other: int) -> PySeries: ...
298
- def sub_i32_rhs(self, other: int) -> PySeries: ...
299
- def sub_i64_rhs(self, other: int) -> PySeries: ...
300
- def sub_f32_rhs(self, other: float) -> PySeries: ...
301
- def sub_f64_rhs(self, other: float) -> PySeries: ...
302
- def div_u8_rhs(self, other: int) -> PySeries: ...
303
- def div_u16_rhs(self, other: int) -> PySeries: ...
304
- def div_u32_rhs(self, other: int) -> PySeries: ...
305
- def div_u64_rhs(self, other: int) -> PySeries: ...
306
- def div_i8_rhs(self, other: int) -> PySeries: ...
307
- def div_i16_rhs(self, other: int) -> PySeries: ...
308
- def div_i32_rhs(self, other: int) -> PySeries: ...
309
- def div_i64_rhs(self, other: int) -> PySeries: ...
310
- def div_f32_rhs(self, other: float) -> PySeries: ...
311
- def div_f64_rhs(self, other: float) -> PySeries: ...
312
- def mul_u8_rhs(self, other: int) -> PySeries: ...
313
- def mul_u16_rhs(self, other: int) -> PySeries: ...
314
- def mul_u32_rhs(self, other: int) -> PySeries: ...
315
- def mul_u64_rhs(self, other: int) -> PySeries: ...
316
- def mul_i8_rhs(self, other: int) -> PySeries: ...
317
- def mul_i16_rhs(self, other: int) -> PySeries: ...
318
- def mul_i32_rhs(self, other: int) -> PySeries: ...
319
- def mul_i64_rhs(self, other: int) -> PySeries: ...
320
- def mul_f32_rhs(self, other: float) -> PySeries: ...
321
- def mul_f64_rhs(self, other: float) -> PySeries: ...
322
- def rem_u8_rhs(self, other: int) -> PySeries: ...
323
- def rem_u16_rhs(self, other: int) -> PySeries: ...
324
- def rem_u32_rhs(self, other: int) -> PySeries: ...
325
- def rem_u64_rhs(self, other: int) -> PySeries: ...
326
- def rem_i8_rhs(self, other: int) -> PySeries: ...
327
- def rem_i16_rhs(self, other: int) -> PySeries: ...
328
- def rem_i32_rhs(self, other: int) -> PySeries: ...
329
- def rem_i64_rhs(self, other: int) -> PySeries: ...
330
- def rem_f32_rhs(self, other: float) -> PySeries: ...
331
- def rem_f64_rhs(self, other: float) -> PySeries: ...
332
-
333
- # buffers
334
- @staticmethod
335
- def _from_buffers(
336
- dtype: Any,
337
- data: Sequence[PySeries],
338
- validity: PySeries | None,
339
- ) -> PySeries: ...
340
- @staticmethod
341
- def _from_buffer(
342
- dtype: DataType,
343
- buffer_info: BufferInfo,
344
- owner: Any,
345
- ) -> PySeries: ...
346
- def _get_buffer_info(self) -> BufferInfo: ...
347
- def _get_buffers(self) -> tuple[PySeries, PySeries | None, PySeries | None]: ...
348
-
349
- # c_interface
350
- @staticmethod
351
- def _import_arrow_from_c(
352
- name: str, chunks: Sequence[tuple[int, int]]
353
- ) -> PySeries: ...
354
- def _export_arrow_to_c(self, out_ptr: int, out_schema_ptr: int) -> None: ...
355
-
356
- # comparison
357
- # Comparison with another PySeries
358
- def eq(self, rhs: PySeries) -> PySeries: ...
359
- def neq(self, rhs: PySeries) -> PySeries: ...
360
- def gt(self, rhs: PySeries) -> PySeries: ...
361
- def gt_eq(self, rhs: PySeries) -> PySeries: ...
362
- def lt(self, rhs: PySeries) -> PySeries: ...
363
- def lt_eq(self, rhs: PySeries) -> PySeries: ...
364
-
365
- # Comparison with scalar values
366
- def eq_u8(self, rhs: int) -> PySeries: ...
367
- def eq_u16(self, rhs: int) -> PySeries: ...
368
- def eq_u32(self, rhs: int) -> PySeries: ...
369
- def eq_u64(self, rhs: int) -> PySeries: ...
370
- def eq_i8(self, rhs: int) -> PySeries: ...
371
- def eq_i16(self, rhs: int) -> PySeries: ...
372
- def eq_i32(self, rhs: int) -> PySeries: ...
373
- def eq_i64(self, rhs: int) -> PySeries: ...
374
- def eq_i128(self, rhs: int) -> PySeries: ...
375
- def eq_f32(self, rhs: float) -> PySeries: ...
376
- def eq_f64(self, rhs: float) -> PySeries: ...
377
- def eq_str(self, rhs: str) -> PySeries: ...
378
- def eq_decimal(self, rhs: Any) -> PySeries: ...
379
- def neq_u8(self, rhs: int) -> PySeries: ...
380
- def neq_u16(self, rhs: int) -> PySeries: ...
381
- def neq_u32(self, rhs: int) -> PySeries: ...
382
- def neq_u64(self, rhs: int) -> PySeries: ...
383
- def neq_i8(self, rhs: int) -> PySeries: ...
384
- def neq_i16(self, rhs: int) -> PySeries: ...
385
- def neq_i32(self, rhs: int) -> PySeries: ...
386
- def neq_i64(self, rhs: int) -> PySeries: ...
387
- def neq_i128(self, rhs: int) -> PySeries: ...
388
- def neq_f32(self, rhs: float) -> PySeries: ...
389
- def neq_f64(self, rhs: float) -> PySeries: ...
390
- def neq_str(self, rhs: str) -> PySeries: ...
391
- def neq_decimal(self, rhs: Any) -> PySeries: ...
392
- def gt_u8(self, rhs: int) -> PySeries: ...
393
- def gt_u16(self, rhs: int) -> PySeries: ...
394
- def gt_u32(self, rhs: int) -> PySeries: ...
395
- def gt_u64(self, rhs: int) -> PySeries: ...
396
- def gt_i8(self, rhs: int) -> PySeries: ...
397
- def gt_i16(self, rhs: int) -> PySeries: ...
398
- def gt_i32(self, rhs: int) -> PySeries: ...
399
- def gt_i64(self, rhs: int) -> PySeries: ...
400
- def gt_i128(self, rhs: int) -> PySeries: ...
401
- def gt_f32(self, rhs: float) -> PySeries: ...
402
- def gt_f64(self, rhs: float) -> PySeries: ...
403
- def gt_str(self, rhs: str) -> PySeries: ...
404
- def gt_decimal(self, rhs: Any) -> PySeries: ...
405
- def gt_eq_u8(self, rhs: int) -> PySeries: ...
406
- def gt_eq_u16(self, rhs: int) -> PySeries: ...
407
- def gt_eq_u32(self, rhs: int) -> PySeries: ...
408
- def gt_eq_u64(self, rhs: int) -> PySeries: ...
409
- def gt_eq_i8(self, rhs: int) -> PySeries: ...
410
- def gt_eq_i16(self, rhs: int) -> PySeries: ...
411
- def gt_eq_i32(self, rhs: int) -> PySeries: ...
412
- def gt_eq_i64(self, rhs: int) -> PySeries: ...
413
- def gt_eq_i128(self, rhs: int) -> PySeries: ...
414
- def gt_eq_f32(self, rhs: float) -> PySeries: ...
415
- def gt_eq_f64(self, rhs: float) -> PySeries: ...
416
- def gt_eq_str(self, rhs: str) -> PySeries: ...
417
- def gt_eq_decimal(self, rhs: Any) -> PySeries: ...
418
- def lt_u8(self, rhs: int) -> PySeries: ...
419
- def lt_u16(self, rhs: int) -> PySeries: ...
420
- def lt_u32(self, rhs: int) -> PySeries: ...
421
- def lt_u64(self, rhs: int) -> PySeries: ...
422
- def lt_i8(self, rhs: int) -> PySeries: ...
423
- def lt_i16(self, rhs: int) -> PySeries: ...
424
- def lt_i32(self, rhs: int) -> PySeries: ...
425
- def lt_i64(self, rhs: int) -> PySeries: ...
426
- def lt_i128(self, rhs: int) -> PySeries: ...
427
- def lt_f32(self, rhs: float) -> PySeries: ...
428
- def lt_f64(self, rhs: float) -> PySeries: ...
429
- def lt_str(self, rhs: str) -> PySeries: ...
430
- def lt_decimal(self, rhs: Any) -> PySeries: ...
431
- def lt_eq_u8(self, rhs: int) -> PySeries: ...
432
- def lt_eq_u16(self, rhs: int) -> PySeries: ...
433
- def lt_eq_u32(self, rhs: int) -> PySeries: ...
434
- def lt_eq_u64(self, rhs: int) -> PySeries: ...
435
- def lt_eq_i8(self, rhs: int) -> PySeries: ...
436
- def lt_eq_i16(self, rhs: int) -> PySeries: ...
437
- def lt_eq_i32(self, rhs: int) -> PySeries: ...
438
- def lt_eq_i64(self, rhs: int) -> PySeries: ...
439
- def lt_eq_i128(self, rhs: int) -> PySeries: ...
440
- def lt_eq_f32(self, rhs: float) -> PySeries: ...
441
- def lt_eq_f64(self, rhs: float) -> PySeries: ...
442
- def lt_eq_str(self, rhs: str) -> PySeries: ...
443
- def lt_eq_decimal(self, rhs: Any) -> PySeries: ...
444
-
445
- # construction
446
- @staticmethod
447
- def new_i8(name: str, array: NDArray1D, _strict: bool) -> PySeries: ...
448
- @staticmethod
449
- def new_i16(name: str, array: NDArray1D, _strict: bool) -> PySeries: ...
450
- @staticmethod
451
- def new_i32(name: str, array: NDArray1D, _strict: bool) -> PySeries: ...
452
- @staticmethod
453
- def new_i64(name: str, array: NDArray1D, _strict: bool) -> PySeries: ...
454
- @staticmethod
455
- def new_u8(name: str, array: NDArray1D, _strict: bool) -> PySeries: ...
456
- @staticmethod
457
- def new_u16(name: str, array: NDArray1D, _strict: bool) -> PySeries: ...
458
- @staticmethod
459
- def new_u32(name: str, array: NDArray1D, _strict: bool) -> PySeries: ...
460
- @staticmethod
461
- def new_u64(name: str, array: NDArray1D, _strict: bool) -> PySeries: ...
462
- @staticmethod
463
- def new_bool(name: str, array: NDArray1D, _strict: bool) -> PySeries: ...
464
- @staticmethod
465
- def new_f32(name: str, array: NDArray1D, nan_is_null: bool) -> PySeries: ...
466
- @staticmethod
467
- def new_f64(name: str, array: NDArray1D, nan_is_null: bool) -> PySeries: ...
468
- @staticmethod
469
- def new_opt_bool(name: str, values: Any, _strict: bool) -> PySeries: ...
470
- @staticmethod
471
- def new_opt_u8(name: str, obj: Any, strict: bool) -> PySeries: ...
472
- @staticmethod
473
- def new_opt_u16(name: str, obj: Any, strict: bool) -> PySeries: ...
474
- @staticmethod
475
- def new_opt_u32(name: str, obj: Any, strict: bool) -> PySeries: ...
476
- @staticmethod
477
- def new_opt_u64(name: str, obj: Any, strict: bool) -> PySeries: ...
478
- @staticmethod
479
- def new_opt_u128(name: str, obj: Any, strict: bool) -> PySeries: ...
480
- @staticmethod
481
- def new_opt_i8(name: str, obj: Any, strict: bool) -> PySeries: ...
482
- @staticmethod
483
- def new_opt_i16(name: str, obj: Any, strict: bool) -> PySeries: ...
484
- @staticmethod
485
- def new_opt_i32(name: str, obj: Any, strict: bool) -> PySeries: ...
486
- @staticmethod
487
- def new_opt_i64(name: str, obj: Any, strict: bool) -> PySeries: ...
488
- @staticmethod
489
- def new_opt_i128(name: str, obj: Any, strict: bool) -> PySeries: ...
490
- @staticmethod
491
- def new_opt_f32(name: str, obj: Any, strict: bool) -> PySeries: ...
492
- @staticmethod
493
- def new_opt_f64(name: str, obj: Any, strict: bool) -> PySeries: ...
494
- @staticmethod
495
- def new_from_any_values(name: str, values: Any, strict: bool) -> PySeries: ...
496
- @staticmethod
497
- def new_from_any_values_and_dtype(
498
- name: str, values: Any, dtype: DataType, strict: bool
499
- ) -> PySeries: ...
500
- @staticmethod
501
- def new_str(name: str, values: Any, _strict: bool) -> PySeries: ...
502
- @staticmethod
503
- def new_binary(name: str, values: Any, _strict: bool) -> PySeries: ...
504
- @staticmethod
505
- def new_decimal(name: str, values: Any, strict: bool) -> PySeries: ...
506
- @staticmethod
507
- def new_series_list(
508
- name: str, values: Sequence[PySeries | None], _strict: bool
509
- ) -> PySeries: ...
510
- @staticmethod
511
- def new_array(
512
- name: str, values: Any, strict: bool, dtype: DataType
513
- ) -> PySeries: ...
514
- @staticmethod
515
- def new_object(name: str, values: Sequence[Any], _strict: bool) -> PySeries: ...
516
- @staticmethod
517
- def new_null(name: str, values: Any, _strict: bool) -> PySeries: ...
518
- @staticmethod
519
- def from_arrow(name: str, array: Any) -> PySeries: ...
520
-
521
- # export
522
- def to_list(self) -> list[Any]: ...
523
- def to_arrow(self, compat_level: Any) -> Any: ...
524
- def __arrow_c_stream__(self, requested_schema: Any | None) -> Any: ...
525
- def _export(self, location: int) -> None: ...
526
-
527
- # import
528
- @classmethod
529
- def from_arrow_c_array(cls, ob: Any) -> PySeries: ...
530
- @classmethod
531
- def from_arrow_c_stream(cls, ob: Any) -> PySeries: ...
532
- @classmethod
533
- def _import(cls, location: int) -> PySeries: ...
534
-
535
- # numpy ufunc
536
- def apply_ufunc_f32(self, lambda_func: Any, allocate_out: bool) -> PySeries: ...
537
- def apply_ufunc_f64(self, lambda_func: Any, allocate_out: bool) -> PySeries: ...
538
- def apply_ufunc_u8(self, lambda_func: Any, allocate_out: bool) -> PySeries: ...
539
- def apply_ufunc_u16(self, lambda_func: Any, allocate_out: bool) -> PySeries: ...
540
- def apply_ufunc_u32(self, lambda_func: Any, allocate_out: bool) -> PySeries: ...
541
- def apply_ufunc_u64(self, lambda_func: Any, allocate_out: bool) -> PySeries: ...
542
- def apply_ufunc_i8(self, lambda_func: Any, allocate_out: bool) -> PySeries: ...
543
- def apply_ufunc_i16(self, lambda_func: Any, allocate_out: bool) -> PySeries: ...
544
- def apply_ufunc_i32(self, lambda_func: Any, allocate_out: bool) -> PySeries: ...
545
- def apply_ufunc_i64(self, lambda_func: Any, allocate_out: bool) -> PySeries: ...
546
-
547
- # scatter
548
- def scatter(self, idx: PySeries, values: PySeries) -> None: ...
549
-
550
- # interop
551
- def to_numpy(self, writable: bool, allow_copy: bool) -> Any: ...
552
- def to_numpy_view(self) -> Any | None: ...
553
- @staticmethod
554
- def _import_decimal_from_iceberg_binary_repr(
555
- *, bytes_list: Sequence[bytes | None], precision: int, scale: int
556
- ) -> PySeries: ...
557
-
558
- class PyDataFrame:
559
- # general
560
- @overload
561
- def __init__(self, columns: Sequence[PySeries]) -> None: ...
562
- @overload
563
- def __init__(self, data: Any, columns: Any, orient: Any) -> None: ...
564
- @overload
565
- def __init__(self, schema: dict[str, Any]) -> None: ...
566
- def estimated_size(self) -> int: ...
567
- def dtype_strings(self) -> list[str]: ...
568
- def add(self, s: PySeries) -> PyDataFrame: ...
569
- def sub(self, s: PySeries) -> PyDataFrame: ...
570
- def mul(self, s: PySeries) -> PyDataFrame: ...
571
- def div(self, s: PySeries) -> PyDataFrame: ...
572
- def rem(self, s: PySeries) -> PyDataFrame: ...
573
- def add_df(self, s: PyDataFrame) -> PyDataFrame: ...
574
- def sub_df(self, s: PyDataFrame) -> PyDataFrame: ...
575
- def mul_df(self, s: PyDataFrame) -> PyDataFrame: ...
576
- def div_df(self, s: PyDataFrame) -> PyDataFrame: ...
577
- def rem_df(self, s: PyDataFrame) -> PyDataFrame: ...
578
- def sample_n(
579
- self, n: PySeries, with_replacement: bool, shuffle: bool, seed: int | None
580
- ) -> PyDataFrame: ...
581
- def sample_frac(
582
- self,
583
- frac: PySeries,
584
- with_replacement: bool,
585
- shuffle: bool,
586
- seed: int | None,
587
- ) -> PyDataFrame: ...
588
- def rechunk(self) -> PyDataFrame: ...
589
- def as_str(self) -> str: ...
590
- def get_columns(self) -> list[PySeries]: ...
591
- def columns(self) -> list[str]: ...
592
- def set_column_names(self, names: Sequence[str]) -> None: ...
593
- def dtypes(self) -> list[Any]: ...
594
- def n_chunks(self) -> int: ...
595
- def shape(self) -> tuple[int, int]: ...
596
- def height(self) -> int: ...
597
- def width(self) -> int: ...
598
- def is_empty(self) -> bool: ...
599
- def hstack(self, columns: Sequence[PySeries]) -> PyDataFrame: ...
600
- def hstack_mut(self, columns: Sequence[PySeries]) -> None: ...
601
- def vstack(self, other: PyDataFrame) -> PyDataFrame: ...
602
- def vstack_mut(self, other: PyDataFrame) -> None: ...
603
- def extend(self, other: PyDataFrame) -> None: ...
604
- def drop_in_place(self, name: str) -> PySeries: ...
605
- def to_series(self, index: int) -> PySeries: ...
606
- def get_column_index(self, name: str) -> int: ...
607
- def get_column(self, name: str) -> PySeries: ...
608
- def select(self, columns: Sequence[str]) -> PyDataFrame: ...
609
- def gather(self, indices: Sequence[int]) -> PyDataFrame: ...
610
- def gather_with_series(self, indices: PySeries) -> PyDataFrame: ...
611
- def replace(self, column: str, new_col: PySeries) -> None: ...
612
- def replace_column(self, index: int, new_column: PySeries) -> None: ...
613
- def insert_column(self, index: int, column: PySeries) -> None: ...
614
- def slice(self, offset: int, length: int | None) -> PyDataFrame: ...
615
- def head(self, n: int) -> PyDataFrame: ...
616
- def tail(self, n: int) -> PyDataFrame: ...
617
- def is_unique(self) -> PySeries: ...
618
- def is_duplicated(self) -> PySeries: ...
619
- def equals(self, other: PyDataFrame, null_equal: bool) -> bool: ...
620
- def with_row_index(self, name: str, offset: int | None) -> PyDataFrame: ...
621
- def _to_metadata(self) -> PyDataFrame: ...
622
- def group_by_map_groups(
623
- self, by: Sequence[str], lambda_func: Any, maintain_order: bool
624
- ) -> PyDataFrame: ...
625
- def clone(self) -> PyDataFrame: ...
626
- def unpivot(
627
- self,
628
- on: Sequence[str],
629
- index: Sequence[str],
630
- value_name: str | None,
631
- variable_name: str | None,
632
- ) -> PyDataFrame: ...
633
- def pivot_expr(
634
- self,
635
- on: Sequence[str],
636
- index: Sequence[str] | None,
637
- values: Sequence[str] | None,
638
- maintain_order: bool,
639
- sort_columns: bool,
640
- aggregate_expr: Any | None,
641
- separator: str | None,
642
- ) -> PyDataFrame: ...
643
- def partition_by(
644
- self, by: Sequence[str], maintain_order: bool, include_key: bool
645
- ) -> list[PyDataFrame]: ...
646
- def lazy(self) -> PyLazyFrame: ...
647
- def to_dummies(
648
- self,
649
- columns: Sequence[str] | None,
650
- separator: str | None,
651
- drop_first: bool,
652
- drop_nulls: bool,
653
- ) -> PyDataFrame: ...
654
- def null_count(self) -> PyDataFrame: ...
655
- def map_rows(
656
- self,
657
- lambda_func: Any,
658
- output_type: Any | None,
659
- inference_size: int,
660
- ) -> tuple[Any, bool]: ...
661
- def shrink_to_fit(self) -> None: ...
662
- def hash_rows(self, k0: int, k1: int, k2: int, k3: int) -> PySeries: ...
663
- def transpose(
664
- self, keep_names_as: str | None, column_names: None | str | Sequence[str]
665
- ) -> PyDataFrame: ...
666
- def upsample(
667
- self,
668
- by: Sequence[str],
669
- index_column: str,
670
- every: str,
671
- stable: bool,
672
- ) -> PyDataFrame: ...
673
- def to_struct(self, name: str, invalid_indices: Sequence[int]) -> PySeries: ...
674
- def clear(self) -> PyDataFrame: ...
675
- def _export_columns(self, location: int) -> None: ...
676
- @classmethod
677
- def _import_columns(cls, location: int, width: int) -> PyDataFrame: ...
678
- def _row_encode(self, opts: Sequence[tuple[bool, bool, bool]]) -> PySeries: ...
679
-
680
- # construction
681
- @staticmethod
682
- def from_rows(
683
- data: Sequence[PySeries],
684
- schema: Any | None,
685
- infer_schema_length: int | None,
686
- ) -> PyDataFrame: ...
687
- @staticmethod
688
- def from_dicts(
689
- data: Any,
690
- schema: Any | None,
691
- schema_overrides: Any | None,
692
- strict: bool,
693
- infer_schema_length: int | None,
694
- ) -> PyDataFrame: ...
695
- @staticmethod
696
- def from_arrow_record_batches(
697
- rb: Sequence[Any],
698
- schema: Any,
699
- ) -> PyDataFrame: ...
700
-
701
- # export
702
- def row_tuple(self, idx: int) -> tuple[Any, ...]: ...
703
- def row_tuples(self) -> list[tuple[Any, ...]]: ...
704
- def to_arrow(self, compat_level: Any) -> list[Any]: ...
705
- def to_pandas(self) -> list[Any]: ...
706
- def __arrow_c_stream__(self, requested_schema: Any | None) -> Any: ...
707
-
708
- # io
709
- @staticmethod
710
- def read_csv(
711
- py_f: Any,
712
- infer_schema_length: int | None,
713
- chunk_size: int,
714
- has_header: bool,
715
- ignore_errors: bool,
716
- n_rows: int | None,
717
- skip_rows: int,
718
- skip_lines: int,
719
- projection: Sequence[int] | None,
720
- separator: str,
721
- rechunk: bool,
722
- columns: Sequence[str] | None,
723
- encoding: Any,
724
- n_threads: int | None,
725
- path: str | None,
726
- overwrite_dtype: Sequence[tuple[str, DataType]] | None,
727
- overwrite_dtype_slice: Sequence[DataType] | None,
728
- low_memory: bool,
729
- comment_prefix: str | None,
730
- quote_char: str | None,
731
- null_values: Any | None,
732
- missing_utf8_is_empty_string: bool,
733
- try_parse_dates: bool,
734
- skip_rows_after_header: int,
735
- row_index: tuple[str, int] | None,
736
- eol_char: str,
737
- raise_if_empty: bool,
738
- truncate_ragged_lines: bool,
739
- decimal_comma: bool,
740
- schema: Any | None,
741
- ) -> PyDataFrame: ...
742
- @staticmethod
743
- def read_json(
744
- py_f: Any,
745
- infer_schema_length: int | None,
746
- schema: Any | None,
747
- schema_overrides: Any | None,
748
- ) -> PyDataFrame: ...
749
- @staticmethod
750
- def read_ipc(
751
- py_f: Any,
752
- columns: Sequence[str] | None,
753
- projection: Sequence[int] | None,
754
- n_rows: int | None,
755
- row_index: tuple[str, int] | None,
756
- memory_map: bool,
757
- ) -> PyDataFrame: ...
758
- @staticmethod
759
- def read_ipc_stream(
760
- py_f: Any,
761
- columns: Sequence[str] | None,
762
- projection: Sequence[int] | None,
763
- n_rows: int | None,
764
- row_index: tuple[str, int] | None,
765
- rechunk: bool,
766
- ) -> PyDataFrame: ...
767
- @staticmethod
768
- def read_avro(
769
- py_f: Any,
770
- columns: Sequence[str] | None,
771
- projection: Sequence[int] | None,
772
- n_rows: int | None,
773
- ) -> PyDataFrame: ...
774
- def write_json(self, py_f: Any) -> None: ...
775
- def write_ipc_stream(
776
- self, py_f: Any, compression: Any, compat_level: Any
777
- ) -> None: ...
778
- def write_avro(self, py_f: Any, compression: Any, name: str) -> None: ...
779
-
780
- # serde
781
- def serialize_binary(self, py_f: Any) -> None: ...
782
- @staticmethod
783
- def deserialize_binary(py_f: Any) -> PyDataFrame: ...
784
- def serialize_json(self, py_f: Any) -> None: ...
785
- @staticmethod
786
- def deserialize_json(py_f: Any) -> PyDataFrame: ...
787
-
788
- # interop
789
- def to_numpy(
790
- self,
791
- order: IndexOrder,
792
- writable: bool,
793
- allow_copy: bool,
794
- ) -> Any: ...
795
-
796
- class PyLazyFrame:
797
- @staticmethod
798
- def new_from_ndjson(
799
- source: Any | None,
800
- sources: Any,
801
- infer_schema_length: int | None,
802
- schema: Any | None,
803
- schema_overrides: Any | None,
804
- batch_size: int | None,
805
- n_rows: int | None,
806
- low_memory: bool,
807
- rechunk: bool,
808
- row_index: tuple[str, int] | None,
809
- ignore_errors: bool,
810
- include_file_paths: str | None,
811
- cloud_options: dict[str, Any] | None,
812
- credential_provider: Any | None,
813
- retries: int,
814
- file_cache_ttl: int | None,
815
- ) -> PyLazyFrame: ...
816
- @staticmethod
817
- def new_from_csv(
818
- source: Any | None,
819
- sources: Any,
820
- separator: str,
821
- has_header: bool,
822
- ignore_errors: bool,
823
- skip_rows: int,
824
- skip_lines: int,
825
- n_rows: int | None,
826
- cache: bool,
827
- overwrite_dtype: Sequence[tuple[str, Any]] | None,
828
- low_memory: bool,
829
- comment_prefix: str | None,
830
- quote_char: str | None,
831
- null_values: Any | None,
832
- missing_utf8_is_empty_string: bool,
833
- infer_schema_length: int | None,
834
- with_schema_modify: Any | None,
835
- rechunk: bool,
836
- skip_rows_after_header: int,
837
- encoding: Any,
838
- row_index: tuple[str, int] | None,
839
- try_parse_dates: bool,
840
- eol_char: str,
841
- raise_if_empty: bool,
842
- truncate_ragged_lines: bool,
843
- decimal_comma: bool,
844
- glob: bool,
845
- schema: Any | None,
846
- cloud_options: dict[str, Any] | None,
847
- credential_provider: Any | None,
848
- retries: int,
849
- file_cache_ttl: int | None,
850
- include_file_paths: str | None,
851
- ) -> PyLazyFrame: ...
852
- @staticmethod
853
- def new_from_parquet(
854
- sources: Any,
855
- schema: Any | None,
856
- scan_options: PyScanOptions,
857
- parallel: Any,
858
- low_memory: bool,
859
- use_statistics: bool,
860
- ) -> PyLazyFrame: ...
861
- @staticmethod
862
- def new_from_ipc(
863
- source: Any | None,
864
- sources: Any,
865
- n_rows: int | None,
866
- cache: bool,
867
- rechunk: bool,
868
- row_index: tuple[str, int] | None,
869
- cloud_options: dict[str, Any] | None,
870
- credential_provider: Any | None,
871
- hive_partitioning: bool | None,
872
- hive_schema: Any | None,
873
- try_parse_hive_dates: bool,
874
- retries: int,
875
- file_cache_ttl: int | None,
876
- include_file_paths: str | None,
877
- ) -> PyLazyFrame: ...
878
- @staticmethod
879
- def new_from_dataset_object(dataset_object: Any) -> PyLazyFrame: ...
880
- @staticmethod
881
- def scan_from_python_function_arrow_schema(
882
- schema: Any, scan_fn: Any, pyarrow: bool, validate_schema: bool, is_pure: bool
883
- ) -> PyLazyFrame: ...
884
- @staticmethod
885
- def scan_from_python_function_pl_schema(
886
- schema: Sequence[tuple[str, Any]],
887
- scan_fn: Any,
888
- pyarrow: bool,
889
- validate_schema: bool,
890
- is_pure: bool,
891
- ) -> PyLazyFrame: ...
892
- @staticmethod
893
- def scan_from_python_function_schema_function(
894
- schema_fn: Any, scan_fn: Any, validate_schema: bool, is_pure: bool
895
- ) -> PyLazyFrame: ...
896
- def pipe_with_schema(
897
- self, callback: Callable[[tuple[PyLazyFrame, Schema]], PyLazyFrame]
898
- ) -> PyLazyFrame: ...
899
- def describe_plan(self) -> str: ...
900
- def describe_optimized_plan(self) -> str: ...
901
- def describe_plan_tree(self) -> str: ...
902
- def describe_optimized_plan_tree(self) -> str: ...
903
- def to_dot(self, optimized: bool) -> str: ...
904
- def to_dot_streaming_phys(self, optimized: bool) -> str: ...
905
- def optimization_toggle(
906
- self,
907
- type_coercion: bool,
908
- type_check: bool,
909
- predicate_pushdown: bool,
910
- projection_pushdown: bool,
911
- simplify_expression: bool,
912
- slice_pushdown: bool,
913
- comm_subplan_elim: bool,
914
- comm_subexpr_elim: bool,
915
- cluster_with_columns: bool,
916
- _eager: bool,
917
- _check_order: bool,
918
- new_streaming: bool,
919
- ) -> PyLazyFrame: ...
920
- def sort(
921
- self,
922
- by_column: str,
923
- descending: bool,
924
- nulls_last: bool,
925
- maintain_order: bool,
926
- multithreaded: bool,
927
- ) -> PyLazyFrame: ...
928
- def sort_by_exprs(
929
- self,
930
- by: Sequence[PyExpr],
931
- descending: Sequence[bool],
932
- nulls_last: Sequence[bool],
933
- maintain_order: bool,
934
- multithreaded: bool,
935
- ) -> PyLazyFrame: ...
936
- def top_k(
937
- self, k: int, by: Sequence[PyExpr], reverse: Sequence[bool]
938
- ) -> PyLazyFrame: ...
939
- def bottom_k(
940
- self, k: int, by: Sequence[PyExpr], reverse: Sequence[bool]
941
- ) -> PyLazyFrame: ...
942
- def cache(self) -> PyLazyFrame: ...
943
- def with_optimizations(self, optflags: PyOptFlags) -> PyLazyFrame: ...
944
- def profile(
945
- self, lambda_post_opt: Any | None
946
- ) -> tuple[PyDataFrame, PyDataFrame]: ...
947
- def collect(self, engine: Any, lambda_post_opt: Any | None) -> PyDataFrame: ...
948
- def collect_with_callback(self, engine: Any, lambda_func: Any) -> None: ...
949
- def sink_parquet(
950
- self,
951
- target: SinkTarget,
952
- compression: str,
953
- compression_level: int | None,
954
- statistics: StatisticsOptions,
955
- row_group_size: int | None,
956
- data_page_size: int | None,
957
- cloud_options: dict[str, Any] | None,
958
- credential_provider: Any | None,
959
- retries: int,
960
- sink_options: Any,
961
- metadata: KeyValueMetadata | None,
962
- field_overwrites: Sequence[ParquetFieldOverwrites],
963
- ) -> PyLazyFrame: ...
964
- def sink_ipc(
965
- self,
966
- target: SinkTarget,
967
- compression: IpcCompression | None,
968
- compat_level: CompatLevel,
969
- cloud_options: dict[str, Any] | None,
970
- credential_provider: Any | None,
971
- retries: int,
972
- sink_options: Any,
973
- ) -> PyLazyFrame: ...
974
- def sink_csv(
975
- self,
976
- target: SinkTarget,
977
- include_bom: bool,
978
- include_header: bool,
979
- separator: int,
980
- line_terminator: str,
981
- quote_char: int,
982
- batch_size: int,
983
- datetime_format: str | None,
984
- date_format: str | None,
985
- time_format: str | None,
986
- float_scientific: bool | None,
987
- float_precision: int | None,
988
- decimal_comma: bool,
989
- null_value: str | None,
990
- quote_style: QuoteStyle | None,
991
- cloud_options: dict[str, Any] | None,
992
- credential_provider: Any | None,
993
- retries: int,
994
- sink_options: Any,
995
- ) -> PyLazyFrame: ...
996
- def sink_json(
997
- self,
998
- target: SinkTarget,
999
- cloud_options: dict[str, Any] | None,
1000
- credential_provider: Any | None,
1001
- retries: int,
1002
- sink_options: Any,
1003
- ) -> PyLazyFrame: ...
1004
- def sink_batches(
1005
- self,
1006
- function: Callable[[PyDataFrame], bool],
1007
- maintain_order: bool,
1008
- chunk_size: int | None,
1009
- ) -> PyLazyFrame: ...
1010
- def filter(self, predicate: PyExpr) -> PyLazyFrame: ...
1011
- def remove(self, predicate: PyExpr) -> PyLazyFrame: ...
1012
- def select(self, exprs: Sequence[PyExpr]) -> PyLazyFrame: ...
1013
- def select_seq(self, exprs: Sequence[PyExpr]) -> PyLazyFrame: ...
1014
- def group_by(self, by: Sequence[PyExpr], maintain_order: bool) -> PyLazyGroupBy: ...
1015
- def rolling(
1016
- self,
1017
- index_column: PyExpr,
1018
- period: str,
1019
- offset: str,
1020
- closed: ClosedWindow,
1021
- by: Sequence[PyExpr],
1022
- ) -> PyLazyGroupBy: ...
1023
- def group_by_dynamic(
1024
- self,
1025
- index_column: PyExpr,
1026
- every: str,
1027
- period: str,
1028
- offset: str,
1029
- label: Label,
1030
- include_boundaries: bool,
1031
- closed: ClosedWindow,
1032
- group_by: Sequence[PyExpr],
1033
- start_by: StartBy,
1034
- ) -> PyLazyGroupBy: ...
1035
- def with_context(self, contexts: Sequence[PyLazyFrame]) -> PyLazyFrame: ...
1036
- def join_asof(
1037
- self,
1038
- other: PyLazyFrame,
1039
- left_on: PyExpr,
1040
- right_on: PyExpr,
1041
- left_by: Sequence[str] | None,
1042
- right_by: Sequence[str] | None,
1043
- allow_parallel: bool,
1044
- force_parallel: bool,
1045
- suffix: str,
1046
- strategy: AsofStrategy,
1047
- tolerance: Any | None,
1048
- tolerance_str: str | None,
1049
- coalesce: bool,
1050
- allow_eq: bool,
1051
- check_sortedness: bool,
1052
- ) -> PyLazyFrame: ...
1053
- def join(
1054
- self,
1055
- other: PyLazyFrame,
1056
- left_on: Sequence[PyExpr],
1057
- right_on: Sequence[PyExpr],
1058
- allow_parallel: bool,
1059
- force_parallel: bool,
1060
- nulls_equal: bool,
1061
- how: JoinType,
1062
- suffix: str,
1063
- validate: JoinValidation,
1064
- maintain_order: MaintainOrderJoin,
1065
- coalesce: bool | None,
1066
- ) -> PyLazyFrame: ...
1067
- def join_where(
1068
- self, other: PyLazyFrame, predicates: Sequence[PyExpr], suffix: str
1069
- ) -> PyLazyFrame: ...
1070
- def with_columns(self, exprs: Sequence[PyExpr]) -> PyLazyFrame: ...
1071
- def with_columns_seq(self, exprs: Sequence[PyExpr]) -> PyLazyFrame: ...
1072
- def match_to_schema(
1073
- self,
1074
- schema: Schema,
1075
- missing_columns: Any,
1076
- missing_struct_fields: Any,
1077
- extra_columns: ExtraColumnsPolicy,
1078
- extra_struct_fields: Any,
1079
- integer_cast: Any,
1080
- float_cast: Any,
1081
- ) -> PyLazyFrame: ...
1082
- def rename(
1083
- self, existing: Sequence[str], new: Sequence[str], strict: bool
1084
- ) -> PyLazyFrame: ...
1085
- def reverse(self) -> PyLazyFrame: ...
1086
- def shift(self, n: PyExpr, fill_value: PyExpr | None) -> PyLazyFrame: ...
1087
- def fill_nan(self, fill_value: PyExpr) -> PyLazyFrame: ...
1088
- def min(self) -> PyLazyFrame: ...
1089
- def max(self) -> PyLazyFrame: ...
1090
- def sum(self) -> PyLazyFrame: ...
1091
- def mean(self) -> PyLazyFrame: ...
1092
- def std(self, ddof: int) -> PyLazyFrame: ...
1093
- def var(self, ddof: int) -> PyLazyFrame: ...
1094
- def median(self) -> PyLazyFrame: ...
1095
- def quantile(
1096
- self, quantile: PyExpr, interpolation: QuantileMethod
1097
- ) -> PyLazyFrame: ...
1098
- def explode(self, subset: PySelector) -> PyLazyFrame: ...
1099
- def null_count(self) -> PyLazyFrame: ...
1100
- def unique(
1101
- self,
1102
- maintain_order: bool,
1103
- subset: PySelector | None,
1104
- keep: UniqueKeepStrategy,
1105
- ) -> PyLazyFrame: ...
1106
- def drop_nans(self, subset: PySelector | None) -> PyLazyFrame: ...
1107
- def drop_nulls(self, subset: PySelector | None) -> PyLazyFrame: ...
1108
- def slice(self, offset: int, len: int | None) -> PyLazyFrame: ...
1109
- def tail(self, n: int) -> PyLazyFrame: ...
1110
- def unpivot(
1111
- self,
1112
- on: PySelector,
1113
- index: PySelector,
1114
- value_name: str | None,
1115
- variable_name: str | None,
1116
- ) -> PyLazyFrame: ...
1117
- def with_row_index(self, name: str, offset: int | None = None) -> PyLazyFrame: ...
1118
- def map_batches(
1119
- self,
1120
- function: Any,
1121
- predicate_pushdown: bool,
1122
- projection_pushdown: bool,
1123
- slice_pushdown: bool,
1124
- streamable: bool,
1125
- schema: Schema | None,
1126
- validate_output: bool,
1127
- ) -> PyLazyFrame: ...
1128
- def drop(self, columns: PySelector) -> PyLazyFrame: ...
1129
- def cast(self, dtypes: dict[str, DataType], strict: bool) -> PyLazyFrame: ...
1130
- def cast_all(self, dtype: PyDataTypeExpr, strict: bool) -> PyLazyFrame: ...
1131
- def clone(self) -> PyLazyFrame: ...
1132
- def collect_schema(self) -> dict[str, Any]: ...
1133
- def unnest(self, columns: PySelector) -> PyLazyFrame: ...
1134
- def count(self) -> PyLazyFrame: ...
1135
- def merge_sorted(self, other: PyLazyFrame, key: str) -> PyLazyFrame: ...
1136
-
1137
- # exitable
1138
- def collect_concurrently(self) -> PyInProcessQuery: ...
1139
-
1140
- # serde
1141
- def serialize_binary(self, py_f: Any) -> None: ...
1142
- def serialize_json(self, py_f: Any) -> None: ...
1143
- @staticmethod
1144
- def deserialize_binary(py_f: Any) -> PyLazyFrame: ...
1145
- @staticmethod
1146
- def deserialize_json(py_f: Any) -> PyLazyFrame: ...
1147
-
1148
- # visit
1149
- def visit(self) -> NodeTraverser: ...
1150
-
1151
- class PyInProcessQuery:
1152
- def cancel(self) -> None: ...
1153
- def fetch(self) -> PyDataFrame | None: ...
1154
- def fetch_blocking(self) -> PyDataFrame: ...
1155
-
1156
- class PyExpr:
1157
- def __init__(self, inner: Any) -> None: ...
1158
- def __richcmp__(self, other: PyExpr, op: Any) -> PyExpr: ...
1159
- def __add__(self, rhs: PyExpr) -> PyExpr: ...
1160
- def __sub__(self, rhs: PyExpr) -> PyExpr: ...
1161
- def __mul__(self, rhs: PyExpr) -> PyExpr: ...
1162
- def __truediv__(self, rhs: PyExpr) -> PyExpr: ...
1163
- def __mod__(self, rhs: PyExpr) -> PyExpr: ...
1164
- def __floordiv__(self, rhs: PyExpr) -> PyExpr: ...
1165
- def __neg__(self) -> PyExpr: ...
1166
- def to_str(self) -> str: ...
1167
- def eq(self, other: PyExpr) -> PyExpr: ...
1168
- def eq_missing(self, other: PyExpr) -> PyExpr: ...
1169
- def neq(self, other: PyExpr) -> PyExpr: ...
1170
- def neq_missing(self, other: PyExpr) -> PyExpr: ...
1171
- def gt(self, other: PyExpr) -> PyExpr: ...
1172
- def gt_eq(self, other: PyExpr) -> PyExpr: ...
1173
- def lt_eq(self, other: PyExpr) -> PyExpr: ...
1174
- def lt(self, other: PyExpr) -> PyExpr: ...
1175
- def alias(self, name: str) -> PyExpr: ...
1176
- def not_(self) -> PyExpr: ...
1177
- def is_null(self) -> PyExpr: ...
1178
- def is_not_null(self) -> PyExpr: ...
1179
- def is_infinite(self) -> PyExpr: ...
1180
- def is_finite(self) -> PyExpr: ...
1181
- def is_nan(self) -> PyExpr: ...
1182
- def is_not_nan(self) -> PyExpr: ...
1183
- def min(self) -> PyExpr: ...
1184
- def max(self) -> PyExpr: ...
1185
- def nan_max(self) -> PyExpr: ...
1186
- def nan_min(self) -> PyExpr: ...
1187
- def mean(self) -> PyExpr: ...
1188
- def median(self) -> PyExpr: ...
1189
- def sum(self) -> PyExpr: ...
1190
- def n_unique(self) -> PyExpr: ...
1191
- def arg_unique(self) -> PyExpr: ...
1192
- def unique(self) -> PyExpr: ...
1193
- def unique_stable(self) -> PyExpr: ...
1194
- def first(self) -> PyExpr: ...
1195
- def last(self) -> PyExpr: ...
1196
- def implode(self) -> PyExpr: ...
1197
- def quantile(self, quantile: PyExpr, interpolation: Any) -> PyExpr: ...
1198
- def cut(
1199
- self,
1200
- breaks: Sequence[float],
1201
- labels: Sequence[str] | None,
1202
- left_closed: bool,
1203
- include_breaks: bool,
1204
- ) -> PyExpr: ...
1205
- def qcut(
1206
- self,
1207
- probs: Sequence[float],
1208
- labels: Sequence[str] | None,
1209
- left_closed: bool,
1210
- allow_duplicates: bool,
1211
- include_breaks: bool,
1212
- ) -> PyExpr: ...
1213
- def qcut_uniform(
1214
- self,
1215
- n_bins: int,
1216
- labels: Sequence[str] | None,
1217
- left_closed: bool,
1218
- allow_duplicates: bool,
1219
- include_breaks: bool,
1220
- ) -> PyExpr: ...
1221
- def rle(self) -> PyExpr: ...
1222
- def rle_id(self) -> PyExpr: ...
1223
- def agg_groups(self) -> PyExpr: ...
1224
- def count(self) -> PyExpr: ...
1225
- def len(self) -> PyExpr: ...
1226
- def value_counts(
1227
- self, sort: bool, parallel: bool, name: str, normalize: bool
1228
- ) -> PyExpr: ...
1229
- def unique_counts(self) -> PyExpr: ...
1230
- def null_count(self) -> PyExpr: ...
1231
- def cast(
1232
- self, dtype: PyDataTypeExpr, strict: bool, wrap_numerical: bool
1233
- ) -> PyExpr: ...
1234
- def sort_with(self, descending: bool, nulls_last: bool) -> PyExpr: ...
1235
- def arg_sort(self, descending: bool, nulls_last: bool) -> PyExpr: ...
1236
- def top_k(self, k: PyExpr) -> PyExpr: ...
1237
- def top_k_by(
1238
- self, by: Sequence[PyExpr], k: PyExpr, reverse: Sequence[bool]
1239
- ) -> PyExpr: ...
1240
- def bottom_k(self, k: PyExpr) -> PyExpr: ...
1241
- def bottom_k_by(
1242
- self, by: Sequence[PyExpr], k: PyExpr, reverse: Sequence[bool]
1243
- ) -> PyExpr: ...
1244
- def peak_min(self) -> PyExpr: ...
1245
- def peak_max(self) -> PyExpr: ...
1246
- def arg_max(self) -> PyExpr: ...
1247
- def arg_min(self) -> PyExpr: ...
1248
- def index_of(self, element: PyExpr) -> PyExpr: ...
1249
- def search_sorted(self, element: PyExpr, side: Any, descending: bool) -> PyExpr: ...
1250
- def gather(self, idx: PyExpr) -> PyExpr: ...
1251
- def get(self, idx: PyExpr) -> PyExpr: ...
1252
- def sort_by(
1253
- self,
1254
- by: Sequence[PyExpr],
1255
- descending: Sequence[bool],
1256
- nulls_last: Sequence[bool],
1257
- multithreaded: bool,
1258
- maintain_order: bool,
1259
- ) -> PyExpr: ...
1260
- def shift(self, n: PyExpr, fill_value: PyExpr | None) -> PyExpr: ...
1261
- def fill_null(self, expr: PyExpr) -> PyExpr: ...
1262
- def fill_null_with_strategy(self, strategy: str, limit: Any) -> PyExpr: ...
1263
- def fill_nan(self, expr: PyExpr) -> PyExpr: ...
1264
- def drop_nulls(self) -> PyExpr: ...
1265
- def drop_nans(self) -> PyExpr: ...
1266
- def filter(self, predicate: PyExpr) -> PyExpr: ...
1267
- def reverse(self) -> PyExpr: ...
1268
- def std(self, ddof: int) -> PyExpr: ...
1269
- def var(self, ddof: int) -> PyExpr: ...
1270
- def is_unique(self) -> PyExpr: ...
1271
- def is_between(self, lower: PyExpr, upper: PyExpr, closed: Any) -> PyExpr: ...
1272
- def is_close(
1273
- self, other: PyExpr, abs_tol: float, rel_tol: float, nans_equal: bool
1274
- ) -> PyExpr: ...
1275
- def approx_n_unique(self) -> PyExpr: ...
1276
- def is_first_distinct(self) -> PyExpr: ...
1277
- def is_last_distinct(self) -> PyExpr: ...
1278
- def explode(self) -> PyExpr: ...
1279
- def gather_every(self, n: int, offset: int) -> PyExpr: ...
1280
- def slice(self, offset: PyExpr, length: PyExpr) -> PyExpr: ...
1281
- def append(self, other: PyExpr, upcast: bool) -> PyExpr: ...
1282
- def rechunk(self) -> PyExpr: ...
1283
- def round(self, decimals: int, mode: Any) -> PyExpr: ...
1284
- def round_sig_figs(self, digits: int) -> PyExpr: ...
1285
- def floor(self) -> PyExpr: ...
1286
- def ceil(self) -> PyExpr: ...
1287
- def clip(self, min: PyExpr | None, max: PyExpr | None) -> PyExpr: ...
1288
- def abs(self) -> PyExpr: ...
1289
- def sin(self) -> PyExpr: ...
1290
- def cos(self) -> PyExpr: ...
1291
- def tan(self) -> PyExpr: ...
1292
- def cot(self) -> PyExpr: ...
1293
- def arcsin(self) -> PyExpr: ...
1294
- def arccos(self) -> PyExpr: ...
1295
- def arctan(self) -> PyExpr: ...
1296
- def arctan2(self, y: PyExpr) -> PyExpr: ...
1297
- def sinh(self) -> PyExpr: ...
1298
- def cosh(self) -> PyExpr: ...
1299
- def tanh(self) -> PyExpr: ...
1300
- def arcsinh(self) -> PyExpr: ...
1301
- def arccosh(self) -> PyExpr: ...
1302
- def arctanh(self) -> PyExpr: ...
1303
- def degrees(self) -> PyExpr: ...
1304
- def radians(self) -> PyExpr: ...
1305
- def sign(self) -> PyExpr: ...
1306
- def is_duplicated(self) -> PyExpr: ...
1307
- def over(
1308
- self,
1309
- partition_by: Sequence[PyExpr] | None,
1310
- order_by: Sequence[PyExpr] | None,
1311
- order_by_descending: bool,
1312
- order_by_nulls_last: bool,
1313
- mapping_strategy: Any,
1314
- ) -> PyExpr: ...
1315
- def rolling(
1316
- self, index_column: str, period: str, offset: str, closed: Any
1317
- ) -> PyExpr: ...
1318
- def and_(self, expr: PyExpr) -> PyExpr: ...
1319
- def or_(self, expr: PyExpr) -> PyExpr: ...
1320
- def xor_(self, expr: PyExpr) -> PyExpr: ...
1321
- def is_in(self, expr: PyExpr, nulls_equal: bool) -> PyExpr: ...
1322
- def repeat_by(self, by: PyExpr) -> PyExpr: ...
1323
- def pow(self, exponent: PyExpr) -> PyExpr: ...
1324
- def sqrt(self) -> PyExpr: ...
1325
- def cbrt(self) -> PyExpr: ...
1326
- def cum_sum(self, reverse: bool) -> PyExpr: ...
1327
- def cum_max(self, reverse: bool) -> PyExpr: ...
1328
- def cum_min(self, reverse: bool) -> PyExpr: ...
1329
- def cum_prod(self, reverse: bool) -> PyExpr: ...
1330
- def cum_count(self, reverse: bool) -> PyExpr: ...
1331
- def cumulative_eval(self, expr: PyExpr, min_samples: int) -> PyExpr: ...
1332
- def product(self) -> PyExpr: ...
1333
- def shrink_dtype(self) -> PyExpr: ...
1334
- def dot(self, other: PyExpr) -> PyExpr: ...
1335
- def reinterpret(self, signed: bool) -> PyExpr: ...
1336
- def mode(self) -> PyExpr: ...
1337
- def interpolate(self, method: Any) -> PyExpr: ...
1338
- def interpolate_by(self, by: PyExpr) -> PyExpr: ...
1339
- def lower_bound(self) -> PyExpr: ...
1340
- def upper_bound(self) -> PyExpr: ...
1341
- def rank(self, method: Any, descending: bool, seed: int | None) -> PyExpr: ...
1342
- def diff(self, n: PyExpr, null_behavior: Any) -> PyExpr: ...
1343
- def pct_change(self, n: PyExpr) -> PyExpr: ...
1344
- def skew(self, bias: bool) -> PyExpr: ...
1345
- def kurtosis(self, fisher: bool, bias: bool) -> PyExpr: ...
1346
- def reshape(self, dims: Sequence[int]) -> PyExpr: ...
1347
- def to_physical(self) -> PyExpr: ...
1348
- def shuffle(self, seed: int | None) -> PyExpr: ...
1349
- def sample_n(
1350
- self, n: PyExpr, with_replacement: bool, shuffle: bool, seed: int | None
1351
- ) -> PyExpr: ...
1352
- def sample_frac(
1353
- self, frac: PyExpr, with_replacement: bool, shuffle: bool, seed: int | None
1354
- ) -> PyExpr: ...
1355
- def ewm_mean(
1356
- self, alpha: float, adjust: bool, min_periods: int, ignore_nulls: bool
1357
- ) -> PyExpr: ...
1358
- def ewm_mean_by(self, times: PyExpr, half_life: str) -> PyExpr: ...
1359
- def ewm_std(
1360
- self,
1361
- alpha: float,
1362
- adjust: bool,
1363
- bias: bool,
1364
- min_periods: int,
1365
- ignore_nulls: bool,
1366
- ) -> PyExpr: ...
1367
- def ewm_var(
1368
- self,
1369
- alpha: float,
1370
- adjust: bool,
1371
- bias: bool,
1372
- min_periods: int,
1373
- ignore_nulls: bool,
1374
- ) -> PyExpr: ...
1375
- def extend_constant(self, value: PyExpr, n: PyExpr) -> PyExpr: ...
1376
- def any(self, ignore_nulls: bool) -> PyExpr: ...
1377
- def all(self, ignore_nulls: bool) -> PyExpr: ...
1378
- def log(self, base: PyExpr) -> PyExpr: ...
1379
- def log1p(self) -> PyExpr: ...
1380
- def exp(self) -> PyExpr: ...
1381
- def entropy(self, base: float, normalize: bool) -> PyExpr: ...
1382
- def hash(self, seed: int, seed_1: int, seed_2: int, seed_3: int) -> PyExpr: ...
1383
- def set_sorted_flag(self, descending: bool) -> PyExpr: ...
1384
- def replace(self, old: PyExpr, new: PyExpr) -> PyExpr: ...
1385
- def replace_strict(
1386
- self,
1387
- old: PyExpr,
1388
- new: PyExpr,
1389
- default: PyExpr | None,
1390
- return_dtype: PyDataTypeExpr | None,
1391
- ) -> PyExpr: ...
1392
- def hist(
1393
- self,
1394
- bins: PyExpr | None,
1395
- bin_count: int | None,
1396
- include_category: bool,
1397
- include_breakpoint: bool,
1398
- ) -> PyExpr: ...
1399
- def skip_batch_predicate(self, schema: Any) -> PyExpr | None: ...
1400
- @staticmethod
1401
- def row_encode_unordered(exprs: Sequence[PyExpr]) -> PyExpr: ...
1402
- @staticmethod
1403
- def row_encode_ordered(
1404
- exprs: Sequence[PyExpr],
1405
- descending: Sequence[bool] | None,
1406
- nulls_last: Sequence[bool] | None,
1407
- ) -> PyExpr: ...
1408
- def row_decode_unordered(
1409
- self, names: Sequence[str], datatypes: Sequence[PyDataTypeExpr]
1410
- ) -> PyExpr: ...
1411
- def row_decode_ordered(
1412
- self,
1413
- names: Sequence[str],
1414
- datatypes: Sequence[PyDataTypeExpr],
1415
- descending: Sequence[bool] | None,
1416
- nulls_last: Sequence[bool] | None,
1417
- ) -> PyExpr: ...
1418
- def into_selector(self) -> Any: ...
1419
- @staticmethod
1420
- def new_selector(selector: Any) -> PyExpr: ...
1421
-
1422
- # array
1423
- def arr_len(self) -> PyExpr: ...
1424
- def arr_max(self) -> PyExpr: ...
1425
- def arr_min(self) -> PyExpr: ...
1426
- def arr_sum(self) -> PyExpr: ...
1427
- def arr_std(self, ddof: int) -> PyExpr: ...
1428
- def arr_var(self, ddof: int) -> PyExpr: ...
1429
- def arr_mean(self) -> PyExpr: ...
1430
- def arr_median(self) -> PyExpr: ...
1431
- def arr_unique(self, maintain_order: bool) -> PyExpr: ...
1432
- def arr_n_unique(self) -> PyExpr: ...
1433
- def arr_to_list(self) -> PyExpr: ...
1434
- def arr_all(self) -> PyExpr: ...
1435
- def arr_any(self) -> PyExpr: ...
1436
- def arr_sort(self, descending: bool, nulls_last: bool) -> PyExpr: ...
1437
- def arr_reverse(self) -> PyExpr: ...
1438
- def arr_arg_min(self) -> PyExpr: ...
1439
- def arr_arg_max(self) -> PyExpr: ...
1440
- def arr_get(self, index: PyExpr, null_on_oob: bool) -> PyExpr: ...
1441
- def arr_join(self, separator: PyExpr, ignore_nulls: bool) -> PyExpr: ...
1442
- def arr_contains(self, other: PyExpr, nulls_equal: bool) -> PyExpr: ...
1443
- def arr_count_matches(self, expr: PyExpr) -> PyExpr: ...
1444
- def arr_to_struct(self, name_gen: Any | None = None) -> PyExpr: ...
1445
- def arr_slice(
1446
- self, offset: PyExpr, length: PyExpr | None = None, as_array: bool = False
1447
- ) -> PyExpr: ...
1448
- def arr_tail(self, n: PyExpr, as_array: bool = False) -> PyExpr: ...
1449
- def arr_shift(self, n: PyExpr) -> PyExpr: ...
1450
- def arr_explode(self) -> PyExpr: ...
1451
-
1452
- # binary
1453
- def bin_contains(self, lit: PyExpr) -> PyExpr: ...
1454
- def bin_ends_with(self, sub: PyExpr) -> PyExpr: ...
1455
- def bin_starts_with(self, sub: PyExpr) -> PyExpr: ...
1456
- def bin_hex_decode(self, strict: bool) -> PyExpr: ...
1457
- def bin_base64_decode(self, strict: bool) -> PyExpr: ...
1458
- def bin_hex_encode(self) -> PyExpr: ...
1459
- def bin_base64_encode(self) -> PyExpr: ...
1460
- def bin_reinterpret(self, dtype: PyDataTypeExpr, kind: str) -> PyExpr: ...
1461
- def bin_size_bytes(self) -> PyExpr: ...
1462
-
1463
- # bitwise
1464
- def bitwise_count_ones(self) -> PyExpr: ...
1465
- def bitwise_count_zeros(self) -> PyExpr: ...
1466
- def bitwise_leading_ones(self) -> PyExpr: ...
1467
- def bitwise_leading_zeros(self) -> PyExpr: ...
1468
- def bitwise_trailing_ones(self) -> PyExpr: ...
1469
- def bitwise_trailing_zeros(self) -> PyExpr: ...
1470
- def bitwise_and(self) -> PyExpr: ...
1471
- def bitwise_or(self) -> PyExpr: ...
1472
- def bitwise_xor(self) -> PyExpr: ...
1473
-
1474
- # categorical
1475
- def cat_get_categories(self) -> PyExpr: ...
1476
- def cat_len_bytes(self) -> PyExpr: ...
1477
- def cat_len_chars(self) -> PyExpr: ...
1478
- def cat_starts_with(self, prefix: str) -> PyExpr: ...
1479
- def cat_ends_with(self, suffix: str) -> PyExpr: ...
1480
- def cat_slice(self, offset: int, length: int | None = None) -> PyExpr: ...
1481
-
1482
- # datetime
1483
- def dt_add_business_days(
1484
- self, n: PyExpr, week_mask: Sequence[bool], holidays: Sequence[int], roll: Roll
1485
- ) -> PyExpr: ...
1486
- def dt_to_string(self, format: str) -> PyExpr: ...
1487
- def dt_offset_by(self, by: PyExpr) -> PyExpr: ...
1488
- def dt_with_time_unit(self, time_unit: TimeUnit) -> PyExpr: ...
1489
- def dt_convert_time_zone(self, time_zone: str) -> PyExpr: ...
1490
- def dt_cast_time_unit(self, time_unit: TimeUnit) -> PyExpr: ...
1491
- def dt_replace_time_zone(
1492
- self,
1493
- time_zone: str | None,
1494
- ambiguous: PyExpr,
1495
- non_existent: NonExistent,
1496
- ) -> PyExpr: ...
1497
- def dt_truncate(self, every: PyExpr) -> PyExpr: ...
1498
- def dt_month_start(self) -> PyExpr: ...
1499
- def dt_month_end(self) -> PyExpr: ...
1500
- def dt_base_utc_offset(self) -> PyExpr: ...
1501
- def dt_dst_offset(self) -> PyExpr: ...
1502
- def dt_round(self, every: PyExpr) -> PyExpr: ...
1503
- def dt_replace(
1504
- self,
1505
- year: PyExpr,
1506
- month: PyExpr,
1507
- day: PyExpr,
1508
- hour: PyExpr,
1509
- minute: PyExpr,
1510
- second: PyExpr,
1511
- microsecond: PyExpr,
1512
- ambiguous: PyExpr,
1513
- ) -> PyExpr: ...
1514
- def dt_combine(self, time: PyExpr, time_unit: TimeUnit) -> PyExpr: ...
1515
- def dt_millennium(self) -> PyExpr: ...
1516
- def dt_century(self) -> PyExpr: ...
1517
- def dt_year(self) -> PyExpr: ...
1518
- def dt_is_business_day(
1519
- self, week_mask: Sequence[bool], holidays: Sequence[int]
1520
- ) -> PyExpr: ...
1521
- def dt_is_leap_year(self) -> PyExpr: ...
1522
- def dt_iso_year(self) -> PyExpr: ...
1523
- def dt_quarter(self) -> PyExpr: ...
1524
- def dt_month(self) -> PyExpr: ...
1525
- def dt_days_in_month(self) -> PyExpr: ...
1526
- def dt_week(self) -> PyExpr: ...
1527
- def dt_weekday(self) -> PyExpr: ...
1528
- def dt_day(self) -> PyExpr: ...
1529
- def dt_ordinal_day(self) -> PyExpr: ...
1530
- def dt_time(self) -> PyExpr: ...
1531
- def dt_date(self) -> PyExpr: ...
1532
- def dt_datetime(self) -> PyExpr: ...
1533
- def dt_hour(self) -> PyExpr: ...
1534
- def dt_minute(self) -> PyExpr: ...
1535
- def dt_second(self) -> PyExpr: ...
1536
- def dt_millisecond(self) -> PyExpr: ...
1537
- def dt_microsecond(self) -> PyExpr: ...
1538
- def dt_nanosecond(self) -> PyExpr: ...
1539
- def dt_timestamp(self, time_unit: TimeUnit) -> PyExpr: ...
1540
- def dt_total_days(self) -> PyExpr: ...
1541
- def dt_total_hours(self) -> PyExpr: ...
1542
- def dt_total_minutes(self) -> PyExpr: ...
1543
- def dt_total_seconds(self) -> PyExpr: ...
1544
- def dt_total_milliseconds(self) -> PyExpr: ...
1545
- def dt_total_microseconds(self) -> PyExpr: ...
1546
- def dt_total_nanoseconds(self) -> PyExpr: ...
1547
-
1548
- # list
1549
- def list_all(self) -> PyExpr: ...
1550
- def list_any(self) -> PyExpr: ...
1551
- def list_arg_max(self) -> PyExpr: ...
1552
- def list_arg_min(self) -> PyExpr: ...
1553
- def list_contains(self, other: PyExpr, nulls_equal: bool) -> PyExpr: ...
1554
- def list_count_matches(self, expr: PyExpr) -> PyExpr: ...
1555
- def list_diff(self, n: int, null_behavior: NullBehavior) -> PyExpr: ...
1556
- def list_eval(self, expr: PyExpr, _parallel: bool) -> PyExpr: ...
1557
- def list_filter(self, predicate: PyExpr) -> PyExpr: ...
1558
- def list_get(self, index: PyExpr, null_on_oob: bool) -> PyExpr: ...
1559
- def list_join(self, separator: PyExpr, ignore_nulls: bool) -> PyExpr: ...
1560
- def list_len(self) -> PyExpr: ...
1561
- def list_max(self) -> PyExpr: ...
1562
- def list_mean(self) -> PyExpr: ...
1563
- def list_median(self) -> PyExpr: ...
1564
- def list_std(self, ddof: int) -> PyExpr: ...
1565
- def list_var(self, ddof: int) -> PyExpr: ...
1566
- def list_min(self) -> PyExpr: ...
1567
- def list_reverse(self) -> PyExpr: ...
1568
- def list_shift(self, periods: PyExpr) -> PyExpr: ...
1569
- def list_slice(self, offset: PyExpr, length: PyExpr | None = None) -> PyExpr: ...
1570
- def list_tail(self, n: PyExpr) -> PyExpr: ...
1571
- def list_sort(self, descending: bool, nulls_last: bool) -> PyExpr: ...
1572
- def list_sum(self) -> PyExpr: ...
1573
- def list_drop_nulls(self) -> PyExpr: ...
1574
- def list_sample_n(
1575
- self, n: PyExpr, with_replacement: bool, shuffle: bool, seed: int | None = None
1576
- ) -> PyExpr: ...
1577
- def list_sample_fraction(
1578
- self,
1579
- fraction: PyExpr,
1580
- with_replacement: bool,
1581
- shuffle: bool,
1582
- seed: int | None = None,
1583
- ) -> PyExpr: ...
1584
- def list_gather(self, index: PyExpr, null_on_oob: bool) -> PyExpr: ...
1585
- def list_gather_every(self, n: PyExpr, offset: PyExpr) -> PyExpr: ...
1586
- def list_to_array(self, width: int) -> PyExpr: ...
1587
- def list_to_struct(self, names: Sequence[str]) -> PyExpr: ...
1588
- def list_to_struct_fixed_width(self, names: Sequence[str]) -> PyExpr: ...
1589
- def list_n_unique(self) -> PyExpr: ...
1590
- def list_unique(self, maintain_order: bool) -> PyExpr: ...
1591
- def list_set_operation(self, other: PyExpr, operation: SetOperation) -> PyExpr: ...
1592
-
1593
- # meta
1594
- def meta_eq(self, other: PyExpr) -> bool: ...
1595
- def meta_pop(self, schema: Schema | None = None) -> list[PyExpr]: ...
1596
- def meta_root_names(self) -> list[str]: ...
1597
- def meta_output_name(self) -> str: ...
1598
- def meta_undo_aliases(self) -> PyExpr: ...
1599
- def meta_has_multiple_outputs(self) -> bool: ...
1600
- def meta_is_column(self) -> bool: ...
1601
- def meta_is_regex_projection(self) -> bool: ...
1602
- def meta_is_column_selection(self, allow_aliasing: bool) -> bool: ...
1603
- def meta_is_literal(self, allow_aliasing: bool) -> bool: ...
1604
- def compute_tree_format(
1605
- self, display_as_dot: bool, schema: Schema | None
1606
- ) -> str: ...
1607
- def meta_tree_format(self, schema: Schema | None = None) -> str: ...
1608
- def meta_show_graph(self, schema: Schema | None = None) -> str: ...
1609
-
1610
- # name
1611
- def name_keep(self) -> PyExpr: ...
1612
- def name_map(self, lambda_function: Any) -> PyExpr: ...
1613
- def name_prefix(self, prefix: str) -> PyExpr: ...
1614
- def name_suffix(self, suffix: str) -> PyExpr: ...
1615
- def name_to_lowercase(self) -> PyExpr: ...
1616
- def name_to_uppercase(self) -> PyExpr: ...
1617
- def name_map_fields(self, name_mapper: Any) -> PyExpr: ...
1618
- def name_prefix_fields(self, prefix: str) -> PyExpr: ...
1619
- def name_suffix_fields(self, suffix: str) -> PyExpr: ...
1620
-
1621
- # rolling
1622
- def rolling_sum(
1623
- self,
1624
- window_size: int,
1625
- weights: Sequence[float] | None = None,
1626
- min_periods: int | None = None,
1627
- center: bool = False,
1628
- ) -> PyExpr: ...
1629
- def rolling_sum_by(
1630
- self,
1631
- by: PyExpr,
1632
- window_size: str,
1633
- min_periods: int,
1634
- closed: ClosedWindow,
1635
- ) -> PyExpr: ...
1636
- def rolling_min(
1637
- self,
1638
- window_size: int,
1639
- weights: Sequence[float] | None = None,
1640
- min_periods: int | None = None,
1641
- center: bool = False,
1642
- ) -> PyExpr: ...
1643
- def rolling_min_by(
1644
- self,
1645
- by: PyExpr,
1646
- window_size: str,
1647
- min_periods: int,
1648
- closed: ClosedWindow,
1649
- ) -> PyExpr: ...
1650
- def rolling_max(
1651
- self,
1652
- window_size: int,
1653
- weights: Sequence[float] | None = None,
1654
- min_periods: int | None = None,
1655
- center: bool = False,
1656
- ) -> PyExpr: ...
1657
- def rolling_max_by(
1658
- self,
1659
- by: PyExpr,
1660
- window_size: str,
1661
- min_periods: int,
1662
- closed: ClosedWindow,
1663
- ) -> PyExpr: ...
1664
- def rolling_mean(
1665
- self,
1666
- window_size: int,
1667
- weights: Sequence[float] | None = None,
1668
- min_periods: int | None = None,
1669
- center: bool = False,
1670
- ) -> PyExpr: ...
1671
- def rolling_mean_by(
1672
- self,
1673
- by: PyExpr,
1674
- window_size: str,
1675
- min_periods: int,
1676
- closed: ClosedWindow,
1677
- ) -> PyExpr: ...
1678
- def rolling_std(
1679
- self,
1680
- window_size: int,
1681
- weights: Sequence[float] | None = None,
1682
- min_periods: int | None = None,
1683
- center: bool = False,
1684
- ddof: int = 1,
1685
- ) -> PyExpr: ...
1686
- def rolling_std_by(
1687
- self,
1688
- by: PyExpr,
1689
- window_size: str,
1690
- min_periods: int,
1691
- closed: ClosedWindow,
1692
- ddof: int = 1,
1693
- ) -> PyExpr: ...
1694
- def rolling_var(
1695
- self,
1696
- window_size: int,
1697
- weights: Sequence[float] | None = None,
1698
- min_periods: int | None = None,
1699
- center: bool = False,
1700
- ddof: int = 1,
1701
- ) -> PyExpr: ...
1702
- def rolling_var_by(
1703
- self,
1704
- by: PyExpr,
1705
- window_size: str,
1706
- min_periods: int,
1707
- closed: ClosedWindow,
1708
- ddof: int = 1,
1709
- ) -> PyExpr: ...
1710
- def rolling_median(
1711
- self,
1712
- window_size: int,
1713
- weights: Sequence[float] | None = None,
1714
- min_periods: int | None = None,
1715
- center: bool = False,
1716
- ) -> PyExpr: ...
1717
- def rolling_median_by(
1718
- self,
1719
- by: PyExpr,
1720
- window_size: str,
1721
- min_periods: int,
1722
- closed: ClosedWindow,
1723
- ) -> PyExpr: ...
1724
- def rolling_quantile(
1725
- self,
1726
- quantile: float,
1727
- interpolation: QuantileMethod,
1728
- window_size: int,
1729
- weights: Sequence[float] | None = None,
1730
- min_periods: int | None = None,
1731
- center: bool = False,
1732
- ) -> PyExpr: ...
1733
- def rolling_quantile_by(
1734
- self,
1735
- by: PyExpr,
1736
- quantile: float,
1737
- interpolation: QuantileMethod,
1738
- window_size: str,
1739
- min_periods: int,
1740
- closed: ClosedWindow,
1741
- ) -> PyExpr: ...
1742
- def rolling_skew(
1743
- self,
1744
- window_size: int,
1745
- bias: bool,
1746
- min_periods: int | None = None,
1747
- center: bool = False,
1748
- ) -> PyExpr: ...
1749
- def rolling_kurtosis(
1750
- self,
1751
- window_size: int,
1752
- fisher: bool,
1753
- bias: bool,
1754
- min_periods: int | None = None,
1755
- center: bool = False,
1756
- ) -> PyExpr: ...
1757
- def rolling_map(
1758
- self,
1759
- lambda_function: Any,
1760
- window_size: int,
1761
- weights: Sequence[float] | None = None,
1762
- min_periods: int | None = None,
1763
- center: bool = False,
1764
- ) -> PyExpr: ...
1765
-
1766
- # serde
1767
- def __getstate__(self) -> bytes: ...
1768
- def __setstate__(self, state: Any) -> None: ...
1769
- def serialize_binary(self, py_f: Any) -> None: ...
1770
- def serialize_json(self, py_f: Any) -> None: ...
1771
- @staticmethod
1772
- def deserialize_binary(py_f: Any) -> PyExpr: ...
1773
- @staticmethod
1774
- def deserialize_json(py_f: Any) -> PyExpr: ...
1775
-
1776
- # string
1777
- def str_join(self, delimiter: str, ignore_nulls: bool) -> PyExpr: ...
1778
- def str_to_date(
1779
- self,
1780
- format: str | None = None,
1781
- strict: bool = True,
1782
- exact: bool = True,
1783
- cache: bool = True,
1784
- ) -> PyExpr: ...
1785
- def str_to_datetime(
1786
- self,
1787
- format: str | None,
1788
- time_unit: TimeUnit | None,
1789
- time_zone: TimeZone | None,
1790
- strict: bool,
1791
- exact: bool,
1792
- cache: bool,
1793
- ambiguous: PyExpr,
1794
- ) -> PyExpr: ...
1795
- def str_to_time(
1796
- self,
1797
- format: str | None = None,
1798
- strict: bool = True,
1799
- cache: bool = True,
1800
- ) -> PyExpr: ...
1801
- def str_strip_chars(self, matches: PyExpr) -> PyExpr: ...
1802
- def str_strip_chars_start(self, matches: PyExpr) -> PyExpr: ...
1803
- def str_strip_chars_end(self, matches: PyExpr) -> PyExpr: ...
1804
- def str_strip_prefix(self, prefix: PyExpr) -> PyExpr: ...
1805
- def str_strip_suffix(self, suffix: PyExpr) -> PyExpr: ...
1806
- def str_slice(self, offset: PyExpr, length: PyExpr) -> PyExpr: ...
1807
- def str_head(self, n: PyExpr) -> PyExpr: ...
1808
- def str_tail(self, n: PyExpr) -> PyExpr: ...
1809
- def str_to_uppercase(self) -> PyExpr: ...
1810
- def str_to_lowercase(self) -> PyExpr: ...
1811
- def str_to_titlecase(self) -> PyExpr: ...
1812
- def str_len_bytes(self) -> PyExpr: ...
1813
- def str_len_chars(self) -> PyExpr: ...
1814
- def str_replace_n(
1815
- self, pat: PyExpr, val: PyExpr, literal: bool, n: int
1816
- ) -> PyExpr: ...
1817
- def str_replace_all(self, pat: PyExpr, val: PyExpr, literal: bool) -> PyExpr: ...
1818
- def str_normalize(self, form: UnicodeForm) -> PyExpr: ...
1819
- def str_reverse(self) -> PyExpr: ...
1820
- def str_pad_start(self, length: PyExpr, fill_char: str) -> PyExpr: ...
1821
- def str_pad_end(self, length: PyExpr, fill_char: str) -> PyExpr: ...
1822
- def str_zfill(self, length: PyExpr) -> PyExpr: ...
1823
- def str_contains(
1824
- self, pat: PyExpr, literal: bool | None = None, strict: bool = True
1825
- ) -> PyExpr: ...
1826
- def str_find(
1827
- self, pat: PyExpr, literal: bool | None = None, strict: bool = True
1828
- ) -> PyExpr: ...
1829
- def str_ends_with(self, sub: PyExpr) -> PyExpr: ...
1830
- def str_starts_with(self, sub: PyExpr) -> PyExpr: ...
1831
- def str_hex_encode(self) -> PyExpr: ...
1832
- def str_hex_decode(self, strict: bool) -> PyExpr: ...
1833
- def str_base64_encode(self) -> PyExpr: ...
1834
- def str_base64_decode(self, strict: bool) -> PyExpr: ...
1835
- def str_to_integer(
1836
- self, base: PyExpr, dtype: Any | None = None, strict: bool = True
1837
- ) -> PyExpr: ...
1838
- def str_json_decode(
1839
- self, dtype: PyDataTypeExpr | None = None, infer_schema_len: int | None = None
1840
- ) -> PyExpr: ...
1841
- def str_json_path_match(self, pat: PyExpr) -> PyExpr: ...
1842
- def str_extract(self, pat: PyExpr, group_index: int) -> PyExpr: ...
1843
- def str_extract_all(self, pat: PyExpr) -> PyExpr: ...
1844
- def str_extract_groups(self, pat: str) -> PyExpr: ...
1845
- def str_count_matches(self, pat: PyExpr, literal: bool) -> PyExpr: ...
1846
- def str_split(self, by: PyExpr) -> PyExpr: ...
1847
- def str_split_inclusive(self, by: PyExpr) -> PyExpr: ...
1848
- def str_split_exact(self, by: PyExpr, n: int) -> PyExpr: ...
1849
- def str_split_exact_inclusive(self, by: PyExpr, n: int) -> PyExpr: ...
1850
- def str_splitn(self, by: PyExpr, n: int) -> PyExpr: ...
1851
- def str_to_decimal(self, scale: int) -> PyExpr: ...
1852
- def str_contains_any(
1853
- self, patterns: PyExpr, ascii_case_insensitive: bool
1854
- ) -> PyExpr: ...
1855
- def str_replace_many(
1856
- self, patterns: PyExpr, replace_with: PyExpr, ascii_case_insensitive: bool
1857
- ) -> PyExpr: ...
1858
- def str_extract_many(
1859
- self, patterns: PyExpr, ascii_case_insensitive: bool, overlapping: bool
1860
- ) -> PyExpr: ...
1861
- def str_find_many(
1862
- self, patterns: PyExpr, ascii_case_insensitive: bool, overlapping: bool
1863
- ) -> PyExpr: ...
1864
- def str_escape_regex(self) -> PyExpr: ...
1865
-
1866
- # struct
1867
- def struct_field_by_index(self, index: int) -> PyExpr: ...
1868
- def struct_field_by_name(self, name: str) -> PyExpr: ...
1869
- def struct_multiple_fields(self, names: Sequence[str]) -> PyExpr: ...
1870
- def struct_rename_fields(self, names: Sequence[str]) -> PyExpr: ...
1871
- def struct_json_encode(self) -> PyExpr: ...
1872
- def struct_with_fields(self, fields: Sequence[PyExpr]) -> PyExpr: ...
1873
-
1874
- class PyDataTypeExpr:
1875
- def __init__(self, inner: Any) -> None: ...
1876
- @staticmethod
1877
- def from_dtype(datatype: Any) -> PyDataTypeExpr: ...
1878
- @staticmethod
1879
- def of_expr(expr: PyExpr) -> PyDataTypeExpr: ...
1880
- @staticmethod
1881
- def self_dtype() -> PyDataTypeExpr: ...
1882
- def collect_dtype(self, schema: Any) -> Any: ...
1883
- def inner_dtype(self) -> PyDataTypeExpr: ...
1884
- def equals(self, other: PyDataTypeExpr) -> PyExpr: ...
1885
- def display(self) -> PyExpr: ...
1886
- def matches(self, selector: Any) -> PyExpr: ...
1887
- @staticmethod
1888
- def struct_with_fields(
1889
- fields: Sequence[tuple[str, PyDataTypeExpr]],
1890
- ) -> PyDataTypeExpr: ...
1891
- def wrap_in_list(self) -> PyDataTypeExpr: ...
1892
- def wrap_in_array(self, width: int) -> PyDataTypeExpr: ...
1893
- def to_unsigned_integer(self) -> PyDataTypeExpr: ...
1894
- def to_signed_integer(self) -> PyDataTypeExpr: ...
1895
- def default_value(
1896
- self, n: int, numeric_to_one: bool, num_list_values: int
1897
- ) -> PyExpr: ...
1898
-
1899
- # list
1900
- def list_inner_dtype(self) -> PyDataTypeExpr: ...
1901
-
1902
- # array
1903
- def arr_inner_dtype(self) -> PyDataTypeExpr: ...
1904
- def arr_width(self) -> PyExpr: ...
1905
- def arr_shape(self) -> PyExpr: ...
1906
-
1907
- # struct
1908
- def struct_field_dtype_by_index(self, index: int) -> PyDataTypeExpr: ...
1909
- def struct_field_dtype_by_name(self, name: str) -> PyDataTypeExpr: ...
1910
- def struct_field_names(self) -> PyExpr: ...
1911
-
1912
- class PySelector:
1913
- def __init__(self, inner: Any) -> None: ...
1914
- def union(self, other: PySelector) -> PySelector: ...
1915
- def difference(self, other: PySelector) -> PySelector: ...
1916
- def exclusive_or(self, other: PySelector) -> PySelector: ...
1917
- def intersect(self, other: PySelector) -> PySelector: ...
1918
- @staticmethod
1919
- def by_dtype(dtypes: Sequence[Any]) -> PySelector: ...
1920
- @staticmethod
1921
- def by_name(names: Sequence[str], strict: bool) -> PySelector: ...
1922
- @staticmethod
1923
- def by_index(indices: Sequence[int], strict: bool) -> PySelector: ...
1924
- @staticmethod
1925
- def first(strict: bool) -> PySelector: ...
1926
- @staticmethod
1927
- def last(strict: bool) -> PySelector: ...
1928
- @staticmethod
1929
- def matches(pattern: str) -> PySelector: ...
1930
- @staticmethod
1931
- def enum_() -> PySelector: ...
1932
- @staticmethod
1933
- def categorical() -> PySelector: ...
1934
- @staticmethod
1935
- def nested() -> PySelector: ...
1936
- @staticmethod
1937
- def list(inner_dst: PySelector | None) -> PySelector: ...
1938
- @staticmethod
1939
- def array(inner_dst: PySelector | None, width: int | None) -> PySelector: ...
1940
- @staticmethod
1941
- def struct_() -> PySelector: ...
1942
- @staticmethod
1943
- def integer() -> PySelector: ...
1944
- @staticmethod
1945
- def signed_integer() -> PySelector: ...
1946
- @staticmethod
1947
- def unsigned_integer() -> PySelector: ...
1948
- @staticmethod
1949
- def float() -> PySelector: ...
1950
- @staticmethod
1951
- def decimal() -> PySelector: ...
1952
- @staticmethod
1953
- def numeric() -> PySelector: ...
1954
- @staticmethod
1955
- def temporal() -> PySelector: ...
1956
- @staticmethod
1957
- def datetime(tu: Sequence[Any], tz: Sequence[Any]) -> PySelector: ...
1958
- @staticmethod
1959
- def duration(tu: Sequence[Any]) -> PySelector: ...
1960
- @staticmethod
1961
- def object() -> PySelector: ...
1962
- @staticmethod
1963
- def empty() -> PySelector: ...
1964
- @staticmethod
1965
- def all() -> PySelector: ...
1966
- def hash(self) -> int: ...
1967
-
1968
- class PyOptFlags:
1969
- def __init__(self) -> None: ...
1970
- @staticmethod
1971
- def empty() -> PyOptFlags: ...
1972
- @staticmethod
1973
- def default() -> PyOptFlags: ...
1974
- def no_optimizations(self) -> None: ...
1975
- def copy(self) -> PyOptFlags: ...
1976
- @property
1977
- def type_coercion(self) -> bool: ...
1978
- @type_coercion.setter
1979
- def type_coercion(self, value: bool) -> None: ...
1980
- @property
1981
- def type_check(self) -> bool: ...
1982
- @type_check.setter
1983
- def type_check(self, value: bool) -> None: ...
1984
- @property
1985
- def projection_pushdown(self) -> bool: ...
1986
- @projection_pushdown.setter
1987
- def projection_pushdown(self, value: bool) -> None: ...
1988
- @property
1989
- def predicate_pushdown(self) -> bool: ...
1990
- @predicate_pushdown.setter
1991
- def predicate_pushdown(self, value: bool) -> None: ...
1992
- @property
1993
- def cluster_with_columns(self) -> bool: ...
1994
- @cluster_with_columns.setter
1995
- def cluster_with_columns(self, value: bool) -> None: ...
1996
- @property
1997
- def simplify_expression(self) -> bool: ...
1998
- @simplify_expression.setter
1999
- def simplify_expression(self, value: bool) -> None: ...
2000
- @property
2001
- def slice_pushdown(self) -> bool: ...
2002
- @slice_pushdown.setter
2003
- def slice_pushdown(self, value: bool) -> None: ...
2004
- @property
2005
- def comm_subplan_elim(self) -> bool: ...
2006
- @comm_subplan_elim.setter
2007
- def comm_subplan_elim(self, value: bool) -> None: ...
2008
- @property
2009
- def comm_subexpr_elim(self) -> bool: ...
2010
- @comm_subexpr_elim.setter
2011
- def comm_subexpr_elim(self, value: bool) -> None: ...
2012
- @property
2013
- def check_order_observe(self) -> bool: ...
2014
- @check_order_observe.setter
2015
- def check_order_observe(self, value: bool) -> None: ...
2016
- @property
2017
- def fast_projection(self) -> bool: ...
2018
- @fast_projection.setter
2019
- def fast_projection(self, value: bool) -> None: ...
2020
- @property
2021
- def eager(self) -> bool: ...
2022
- @eager.setter
2023
- def eager(self, value: bool) -> None: ...
2024
- @property
2025
- def streaming(self) -> bool: ...
2026
- @streaming.setter
2027
- def streaming(self, value: bool) -> None: ...
2028
-
2029
- class PyPartitioning:
2030
- def __init__(self) -> None: ...
2031
- base_path: Any
2032
-
2033
- @staticmethod
2034
- def new_max_size(
2035
- base_path: Any,
2036
- file_path_cb: Any | None,
2037
- max_size: int,
2038
- per_partition_sort_by: Sequence[PyExpr] | None,
2039
- finish_callback: Any | None,
2040
- ) -> PyPartitioning: ...
2041
- @staticmethod
2042
- def new_by_key(
2043
- base_path: Any,
2044
- file_path_cb: Any | None,
2045
- by: Sequence[PyExpr],
2046
- include_key: bool,
2047
- per_partition_sort_by: Sequence[PyExpr] | None,
2048
- finish_callback: Any | None,
2049
- ) -> PyPartitioning: ...
2050
- @staticmethod
2051
- def new_parted(
2052
- base_path: Any,
2053
- file_path_cb: Any | None,
2054
- by: Sequence[PyExpr],
2055
- include_key: bool,
2056
- per_partition_sort_by: Sequence[PyExpr] | None,
2057
- finish_callback: Any | None,
2058
- ) -> PyPartitioning: ...
2059
-
2060
- # functions.lazy
2061
- def rolling_corr(
2062
- x: PyExpr, y: PyExpr, window_size: int, min_periods: int, ddof: int
2063
- ) -> PyExpr: ...
2064
- def rolling_cov(
2065
- x: PyExpr, y: PyExpr, window_size: int, min_periods: int, ddof: int
2066
- ) -> PyExpr: ...
2067
- def arg_sort_by(
2068
- by: Sequence[PyExpr],
2069
- descending: Sequence[bool],
2070
- nulls_last: Sequence[bool],
2071
- multithreaded: bool,
2072
- maintain_order: bool,
2073
- ) -> PyExpr: ...
2074
- def arg_where(condition: PyExpr) -> PyExpr: ...
2075
- def as_struct(exprs: Sequence[PyExpr]) -> PyExpr: ...
2076
- def field(names: Sequence[str]) -> PyExpr: ...
2077
- def coalesce(exprs: Sequence[PyExpr]) -> PyExpr: ...
2078
- def col(name: str) -> PyExpr: ...
2079
- def collect_all(
2080
- lfs: Sequence[PyLazyFrame], engine: Any, optflags: PyOptFlags
2081
- ) -> list[PyDataFrame]: ...
2082
- def explain_all(lfs: Sequence[PyLazyFrame], optflags: PyOptFlags) -> str: ...
2083
- def collect_all_with_callback(
2084
- lfs: Sequence[PyLazyFrame], engine: Any, optflags: PyOptFlags, lambda_func: Any
2085
- ) -> None: ...
2086
- def concat_lf(
2087
- seq: Any, rechunk: bool, parallel: bool, to_supertypes: bool
2088
- ) -> PyLazyFrame: ...
2089
- def concat_list(s: Sequence[PyExpr]) -> PyExpr: ...
2090
- def concat_arr(s: Sequence[PyExpr]) -> PyExpr: ...
2091
- def concat_str(s: Sequence[PyExpr], separator: str, ignore_nulls: bool) -> PyExpr: ...
2092
- def len() -> PyExpr: ...
2093
- def cov(a: PyExpr, b: PyExpr, ddof: int) -> PyExpr: ...
2094
- def arctan2(y: PyExpr, x: PyExpr) -> PyExpr: ...
2095
- def cum_fold(
2096
- acc: PyExpr,
2097
- lambda_func: Any,
2098
- exprs: Sequence[PyExpr],
2099
- returns_scalar: bool,
2100
- return_dtype: PyDataTypeExpr | None,
2101
- include_init: bool,
2102
- ) -> PyExpr: ...
2103
- def cum_reduce(
2104
- lambda_func: Any,
2105
- exprs: Sequence[PyExpr],
2106
- returns_scalar: bool,
2107
- return_dtype: PyDataTypeExpr | None,
2108
- ) -> PyExpr: ...
2109
- def datetime(
2110
- year: PyExpr,
2111
- month: PyExpr,
2112
- day: PyExpr,
2113
- hour: PyExpr | None,
2114
- minute: PyExpr | None,
2115
- second: PyExpr | None,
2116
- microsecond: PyExpr | None,
2117
- time_unit: TimeUnit, # Default set by Rust code
2118
- time_zone: TimeZone | None, # Default set by Rust code
2119
- ambiguous: PyExpr, # Default set by Rust code
2120
- ) -> PyExpr: ...
2121
- def concat_lf_diagonal(
2122
- lfs: Any, rechunk: bool, parallel: bool, to_supertypes: bool
2123
- ) -> PyLazyFrame: ...
2124
- def concat_lf_horizontal(lfs: Any, parallel: bool) -> PyLazyFrame: ...
2125
- def concat_expr(e: Sequence[PyExpr], rechunk: bool) -> PyExpr: ...
2126
- def duration(
2127
- weeks: PyExpr | None,
2128
- days: PyExpr | None,
2129
- hours: PyExpr | None,
2130
- minutes: PyExpr | None,
2131
- seconds: PyExpr | None,
2132
- milliseconds: PyExpr | None,
2133
- microseconds: PyExpr | None,
2134
- nanoseconds: PyExpr | None,
2135
- time_unit: TimeUnit, # Default set by Rust code
2136
- ) -> PyExpr: ...
2137
- def fold(
2138
- acc: PyExpr,
2139
- lambda_func: Any,
2140
- exprs: Sequence[PyExpr],
2141
- returns_scalar: bool,
2142
- return_dtype: PyDataTypeExpr | None,
2143
- ) -> PyExpr: ...
2144
- def lit(value: Any, allow_object: bool, is_scalar: bool) -> PyExpr: ...
2145
- def map_expr(
2146
- pyexpr: Sequence[PyExpr],
2147
- lambda_func: Any,
2148
- output_type: PyDataTypeExpr | None,
2149
- is_elementwise: bool,
2150
- returns_scalar: bool,
2151
- ) -> PyExpr: ...
2152
- def pearson_corr(a: PyExpr, b: PyExpr) -> PyExpr: ...
2153
- def reduce(
2154
- lambda_func: Any,
2155
- exprs: Sequence[PyExpr],
2156
- returns_scalar: bool,
2157
- return_dtype: PyDataTypeExpr | None,
2158
- ) -> PyExpr: ...
2159
- def repeat(value: PyExpr, n: PyExpr, dtype: Any | None = None) -> PyExpr: ...
2160
- def spearman_rank_corr(a: PyExpr, b: PyExpr, propagate_nans: bool) -> PyExpr: ...
2161
- def sql_expr(sql: str) -> PyExpr: ...
2162
-
2163
- # functions.aggregations
2164
- def all_horizontal(exprs: Sequence[PyExpr]) -> PyExpr: ...
2165
- def any_horizontal(exprs: Sequence[PyExpr]) -> PyExpr: ...
2166
- def max_horizontal(exprs: Sequence[PyExpr]) -> PyExpr: ...
2167
- def min_horizontal(exprs: Sequence[PyExpr]) -> PyExpr: ...
2168
- def sum_horizontal(exprs: Sequence[PyExpr], ignore_nulls: bool) -> PyExpr: ...
2169
- def mean_horizontal(exprs: Sequence[PyExpr], ignore_nulls: bool) -> PyExpr: ...
2170
-
2171
- # functions.business
2172
- def business_day_count(
2173
- start: PyExpr,
2174
- end: PyExpr,
2175
- week_mask: Sequence[bool],
2176
- holidays: Sequence[int],
2177
- ) -> PyExpr: ...
2178
-
2179
- # functions.eager
2180
- def concat_df(dfs: Any) -> PyDataFrame: ...
2181
- def concat_series(series: Any) -> PySeries: ...
2182
- def concat_df_diagonal(dfs: Any) -> PyDataFrame: ...
2183
- def concat_df_horizontal(dfs: Any) -> PyDataFrame: ...
2184
-
2185
- # functions.io
2186
- def read_ipc_schema(py_f: Any) -> dict[str, Any]: ...
2187
- def read_parquet_metadata(
2188
- py_f: Any, storage_options: Any, credential_provider: Any, retries: int
2189
- ) -> dict[str, str]: ...
2190
- def read_clipboard_string() -> str: ...
2191
- def write_clipboard_string(s: str) -> None: ...
2192
-
2193
- # functions.meta
2194
- def get_index_type() -> Any: ...
2195
- def thread_pool_size() -> int: ...
2196
- def set_float_fmt(fmt: FloatFmt) -> None: ...
2197
- def get_float_fmt() -> str: ...
2198
- def set_float_precision(precision: int | None) -> None: ...
2199
- def get_float_precision() -> int | None: ...
2200
- def set_thousands_separator(sep: str | None) -> None: ...
2201
- def get_thousands_separator() -> str | None: ...
2202
- def set_decimal_separator(sep: str | None) -> None: ...
2203
- def get_decimal_separator() -> str | None: ...
2204
- def set_trim_decimal_zeros(trim: bool | None) -> None: ...
2205
- def get_trim_decimal_zeros() -> bool | None: ...
2206
-
2207
- # functions.misc
2208
- def dtype_str_repr(dtype: Any) -> str: ...
2209
- def register_plugin_function(
2210
- plugin_path: str,
2211
- function_name: str,
2212
- args: Sequence[PyExpr],
2213
- kwargs: Sequence[int],
2214
- is_elementwise: bool,
2215
- input_wildcard_expansion: bool,
2216
- returns_scalar: bool,
2217
- cast_to_supertype: bool,
2218
- pass_name_to_apply: bool,
2219
- changes_length: bool,
2220
- ) -> PyExpr: ...
2221
- def __register_startup_deps() -> None: ...
2222
-
2223
- # functions.random
2224
- def set_random_seed(seed: int) -> None: ...
2225
-
2226
- # functions.range
2227
- def int_range(
2228
- start: PyExpr, end: PyExpr, step: int, dtype: PyDataTypeExpr
2229
- ) -> PyExpr: ...
2230
- def eager_int_range(
2231
- lower: Any, upper: Any, step: Any, dtype: PyDataTypeExpr
2232
- ) -> PySeries: ...
2233
- def int_ranges(
2234
- start: PyExpr, end: PyExpr, step: PyExpr, dtype: PyDataTypeExpr
2235
- ) -> PyExpr: ...
2236
- def date_range(
2237
- start: PyExpr, end: PyExpr, interval: str, closed: ClosedWindow
2238
- ) -> PyExpr: ...
2239
- def date_ranges(
2240
- start: PyExpr, end: PyExpr, interval: str, closed: ClosedWindow
2241
- ) -> PyExpr: ...
2242
- def datetime_range(
2243
- start: PyExpr,
2244
- end: PyExpr,
2245
- every: str,
2246
- closed: ClosedWindow,
2247
- time_unit: TimeUnit | None,
2248
- time_zone: TimeZone | None,
2249
- ) -> PyExpr: ...
2250
- def datetime_ranges(
2251
- start: PyExpr,
2252
- end: PyExpr,
2253
- every: str,
2254
- closed: ClosedWindow,
2255
- time_unit: TimeUnit | None,
2256
- time_zone: TimeZone | None,
2257
- ) -> PyExpr: ...
2258
- def time_range(
2259
- start: PyExpr, end: PyExpr, every: str, closed: ClosedWindow
2260
- ) -> PyExpr: ...
2261
- def time_ranges(
2262
- start: PyExpr, end: PyExpr, every: str, closed: ClosedWindow
2263
- ) -> PyExpr: ...
2264
- def linear_space(
2265
- start: PyExpr, end: PyExpr, num_samples: PyExpr, closed: ClosedInterval
2266
- ) -> PyExpr: ...
2267
- def linear_spaces(
2268
- start: PyExpr,
2269
- end: PyExpr,
2270
- num_samples: PyExpr,
2271
- closed: ClosedInterval,
2272
- as_array: bool,
2273
- ) -> PyExpr: ...
2274
-
2275
- # functions.string_cache
2276
- class PyStringCacheHolder: ...
2277
-
2278
- def enable_string_cache() -> None: ...
2279
- def disable_string_cache() -> None: ...
2280
- def using_string_cache() -> bool: ...
2281
-
2282
- # functions.strings
2283
- def escape_regex(s: str) -> str: ...
2284
-
2285
- # functions.strings
2286
- def check_length(check: bool) -> None: ...
2287
- def get_engine_affinity() -> EngineType: ...
2288
-
2289
- # functions.when
2290
- class PyWhen:
2291
- def then(self, statement: PyExpr) -> PyThen: ...
2292
-
2293
- class PyThen:
2294
- def when(self, condition: PyExpr) -> PyChainedWhen: ...
2295
- def otherwise(self, statement: PyExpr) -> PyExpr: ...
2296
-
2297
- class PyChainedWhen:
2298
- def then(self, statement: PyExpr) -> PyChainedThen: ...
2299
-
2300
- class PyChainedThen:
2301
- def when(self, condition: PyExpr) -> PyChainedWhen: ...
2302
- def otherwise(self, statement: PyExpr) -> PyExpr: ...
2303
-
2304
- def when(condition: PyExpr) -> PyWhen: ...
2305
-
2306
- # functions: schema
2307
- def init_polars_schema_from_arrow_c_schema(
2308
- polars_schema: Any, schema_object: Any
2309
- ) -> None: ...
2310
- def polars_schema_field_from_arrow_c_schema(schema_object: Any) -> tuple[Any, Any]: ...
2311
- def polars_schema_to_pycapsule(schema: Schema, compat_level: CompatLevel) -> Any: ...
2312
-
2313
- class PyLazyGroupBy:
2314
- def agg(self, aggs: list[PyExpr]) -> PyLazyFrame: ...
2315
- def head(self, n: int) -> PyLazyFrame: ...
2316
- def tail(self, n: int) -> PyLazyFrame: ...
2317
- def map_groups(
2318
- self, lambda_function: Any, schema: Schema | None
2319
- ) -> PyLazyFrame: ...
2320
-
2321
- # categorical
2322
- class PyCategories:
2323
- def __init__(self, name: str, namespace: str, physical: str) -> None: ...
2324
- @staticmethod
2325
- def global_categories() -> PyCategories: ...
2326
- @staticmethod
2327
- def random(namespace: str, physical: str) -> PyCategories: ...
2328
- def __eq__(self, other: PyCategories) -> bool: ... # type: ignore[override]
2329
- def __hash__(self) -> int: ...
2330
- def name(self) -> str: ...
2331
- def namespace(self) -> str: ...
2332
- def physical(self) -> str: ...
2333
- def get_cat(self, s: str) -> int | None: ...
2334
- def cat_to_str(self, cat: int) -> str | None: ...
2335
- def is_global(self) -> bool: ...
2336
-
2337
- class PyBatchedCsv:
2338
- @staticmethod
2339
- def new(
2340
- infer_schema_length: int | None,
2341
- chunk_size: int,
2342
- has_header: bool,
2343
- ignore_errors: bool,
2344
- n_rows: int | None,
2345
- skip_rows: int,
2346
- skip_lines: int,
2347
- projection: Sequence[int] | None,
2348
- separator: str,
2349
- rechunk: bool,
2350
- columns: Sequence[str] | None,
2351
- encoding: CsvEncoding,
2352
- n_threads: int | None,
2353
- path: Any,
2354
- schema_overrides: Sequence[tuple[str, DataType]] | None,
2355
- overwrite_dtype_slice: Sequence[DataType] | None,
2356
- low_memory: bool,
2357
- comment_prefix: str | None,
2358
- quote_char: str | None,
2359
- null_values: NullValues | None,
2360
- missing_utf8_is_empty_string: bool,
2361
- try_parse_dates: bool,
2362
- skip_rows_after_header: int,
2363
- row_index: tuple[str, int] | None,
2364
- eol_char: str,
2365
- raise_if_empty: bool,
2366
- truncate_ragged_lines: bool,
2367
- decimal_comma: bool,
2368
- ) -> PyBatchedCsv: ...
2369
- def next_batches(self, n: int) -> list[PyDataFrame] | None: ...
2370
-
2371
- # catalog
2372
- class PyCatalogClient:
2373
- @staticmethod
2374
- def new(workspace_url: str, bearer_token: str | None) -> PyCatalogClient: ...
2375
- def list_catalogs(self) -> list[Any]: ...
2376
- def list_namespaces(self, catalog_name: str) -> list[Any]: ...
2377
- def list_tables(self, catalog_name: str, namespace: str) -> list[Any]: ...
2378
- def get_table_info(
2379
- self, table_name: str, catalog_name: str, namespace: str
2380
- ) -> Any: ...
2381
- def get_table_credentials(
2382
- self, table_id: str, write: bool
2383
- ) -> tuple[Any, Any, Any]: ...
2384
- def scan_table(
2385
- self,
2386
- catalog_name: str,
2387
- namespace: str,
2388
- table_name: str,
2389
- cloud_options: dict[str, str] | None,
2390
- credential_provider: Any | None,
2391
- retries: int,
2392
- ) -> PyLazyFrame: ...
2393
- def create_catalog(
2394
- self, catalog_name: str, comment: str | None, storage_root: str | None
2395
- ) -> Any: ...
2396
- def delete_catalog(self, catalog_name: str, force: bool) -> None: ...
2397
- def create_namespace(
2398
- self,
2399
- catalog_name: str,
2400
- namespace: str,
2401
- comment: str | None,
2402
- storage_root: str | None,
2403
- ) -> Any: ...
2404
- def delete_namespace(
2405
- self, catalog_name: str, namespace: str, force: bool
2406
- ) -> None: ...
2407
- def create_table(
2408
- self,
2409
- catalog_name: str,
2410
- namespace: str,
2411
- table_name: str,
2412
- schema: Any | None,
2413
- table_type: str,
2414
- data_source_format: str | None,
2415
- comment: str | None,
2416
- storage_root: str | None,
2417
- properties: Sequence[tuple[str, str]],
2418
- ) -> Any: ...
2419
- def delete_table(
2420
- self, catalog_name: str, namespace: str, table_name: str
2421
- ) -> None: ...
2422
- @staticmethod
2423
- def type_json_to_polars_type(type_json: str) -> Any: ...
2424
- @staticmethod
2425
- def init_classes(
2426
- catalog_info_cls: Any,
2427
- namespace_info_cls: Any,
2428
- table_info_cls: Any,
2429
- column_info_cls: Any,
2430
- ) -> None: ...
2431
-
2432
- # sql
2433
- class PySQLContext:
2434
- @staticmethod
2435
- def new() -> PySQLContext: ...
2436
- def execute(self, query: str) -> PyLazyFrame: ...
2437
- def get_tables(self) -> list[str]: ...
2438
- def register(self, name: str, lf: PyLazyFrame) -> None: ...
2439
- def unregister(self, name: str) -> None: ...
2440
-
2441
- # testing
2442
- def assert_series_equal_py(
2443
- left: PySeries,
2444
- right: PySeries,
2445
- *,
2446
- check_dtypes: bool,
2447
- check_names: bool,
2448
- check_order: bool,
2449
- check_exact: bool,
2450
- rel_tol: float,
2451
- abs_tol: float,
2452
- categorical_as_str: bool,
2453
- ) -> None: ...
2454
- def assert_dataframe_equal_py(
2455
- left: PyDataFrame,
2456
- right: PyDataFrame,
2457
- *,
2458
- check_row_order: bool,
2459
- check_column_order: bool,
2460
- check_dtypes: bool,
2461
- check_exact: bool,
2462
- rel_tol: float,
2463
- abs_tol: float,
2464
- categorical_as_str: bool,
2465
- ) -> None: ...
2466
-
2467
- # datatypes
2468
- def _get_dtype_max(dt: DataType) -> PyExpr: ...
2469
- def _get_dtype_min(dt: DataType) -> PyExpr: ...
2470
- def _known_timezones() -> list[str]: ...
2471
-
2472
- # cloud_client
2473
- def prepare_cloud_plan(lf: PyLazyFrame) -> bytes: ...
2474
-
2475
- # cloud_server
2476
- def _execute_ir_plan_with_gpu(ir_plan_ser: Sequence[int]) -> PyDataFrame: ...
2477
-
2478
- # visit
2479
- class PyExprIR:
2480
- node: int
2481
- output_name: str
2482
-
2483
- class NodeTraverser:
2484
- def get_exprs(self) -> list[PyExprIR]: ...
2485
- def get_inputs(self) -> list[int]: ...
2486
- def version(self) -> tuple[int, int]: ...
2487
- def get_schema(self) -> dict[str, DataType]: ...
2488
- def get_dtype(self, expr_node: int) -> DataType: ...
2489
- def set_node(self, node: int) -> None: ...
2490
- def get_node(self) -> int: ...
2491
- def set_udf(self, function: Any, is_pure: bool = False) -> None: ...
2492
- def view_current_node(self) -> Any: ...
2493
- def view_expression(self, node: int) -> Any: ...
2494
- def add_expressions(self, expressions: list[PyExpr]) -> tuple[list[int], int]: ...
2495
- def set_expr_mapping(self, mapping: list[int]) -> None: ...
2496
- def unset_expr_mapping(self) -> None: ...