mirror of https://github.com/apache/superset.git
56 lines
1.7 KiB
Python
56 lines
1.7 KiB
Python
# Licensed to the Apache Software Foundation (ASF) under one
|
|
# or more contributor license agreements. See the NOTICE file
|
|
# distributed with this work for additional information
|
|
# regarding copyright ownership. The ASF licenses this file
|
|
# to you under the Apache License, Version 2.0 (the
|
|
# "License"); you may not use this file except in compliance
|
|
# with the License. You may obtain a copy of the License at
|
|
#
|
|
# http://www.apache.org/licenses/LICENSE-2.0
|
|
#
|
|
# Unless required by applicable law or agreed to in writing,
|
|
# software distributed under the License is distributed on an
|
|
# "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY
|
|
# KIND, either express or implied. See the License for the
|
|
# specific language governing permissions and limitations
|
|
# under the License.
|
|
import math
|
|
from typing import Any, Optional
|
|
|
|
from pandas import Series
|
|
|
|
AGGREGATES_SINGLE = {"idx_nulls": {"operator": "sum"}}
|
|
AGGREGATES_MULTIPLE = {
|
|
"idx_nulls": {"operator": "sum"},
|
|
"asc_idx": {"operator": "mean"},
|
|
}
|
|
|
|
|
|
def series_to_list(series: Series) -> list[Any]:
|
|
"""
|
|
Converts a `Series` to a regular list, and replaces non-numeric values to
|
|
Nones.
|
|
|
|
:param series: Series to convert
|
|
:return: list without nan or inf
|
|
"""
|
|
return [
|
|
None
|
|
if not isinstance(val, str) and (math.isnan(val) or math.isinf(val))
|
|
else val
|
|
for val in series.tolist()
|
|
]
|
|
|
|
|
|
def round_floats(
|
|
floats: list[Optional[float]], precision: int
|
|
) -> list[Optional[float]]:
|
|
"""
|
|
Round list of floats to certain precision
|
|
|
|
:param floats: floats to round
|
|
:param precision: intended decimal precision
|
|
:return: rounded floats
|
|
"""
|
|
return [round(val, precision) if val else None for val in floats]
|