feat(bigframes): Add numpy ufunc support to col expressions by TrevorBergeron · Pull Request #16554 · googleapis/google-cloud-python
| pd_kwargs = { | |
| "sqrt": np.sqrt(pd.col("float64_col")), # type: ignore | |
| "add_const": np.add(pd.col("float64_col"), 2.4), # type: ignore | |
| "radd_const": np.add(2.4, pd.col("float64_col")), # type: ignore | |
| "add_cols": np.add(pd.col("float64_col"), pd.col("int64_col")), # type: ignore | |
| } | |
| pd_kwargs = dict(sorted({ | |
| "sqrt": np.sqrt(scalars_pandas_df["float64_col"]), | |
| "add_const": np.add(scalars_pandas_df["float64_col"], 2.4), | |
| "radd_const": np.add(2.4, scalars_pandas_df["float64_col"]), | |
| "add_cols": np.add(scalars_pandas_df["float64_col"], scalars_pandas_df["int64_col"]), | |
| }.items())) |