QT/superset2
1
0
Fork 0
mirror of https://github.com/apache/superset.git synced 2026-05-11 02:45:46 +00:00
Code Issues Packages Projects Releases Wiki Activity

feat: add renameOperator (#19776)

Browse Source 
(cherry picked from commit 3c28cd4625)
This commit is contained in:
Yongjie Zhao
2022-04-20 19:48:12 +08:00
committed by Ville Brofeldt
parent 2b6e35e039
commit 8c8bbfb89f
12 changed files with 512 additions and 71 deletions
Download Patch File Download Diff File Expand all files Collapse all files

18
superset/utils/pandas_postprocessing/flatten.py
View File

@@ -81,14 +81,16 @@ def flatten(
"""
if _is_multi_index_on_columns(df):
df.columns = df.columns.droplevel(drop_levels)
# every cell should be converted to string
df.columns = [
FLAT_COLUMN_SEPARATOR.join(
# pylint: disable=superfluous-parens
[str(cell) for cell in (series if is_sequence(series) else [series])]
)
for series in df.columns.to_flat_index()
]
_columns = []
for series in df.columns.to_flat_index():
_cells = []
for cell in series if is_sequence(series) else [series]:
if pd.notnull(cell):
# every cell should be converted to string
_cells.append(str(cell))
_columns.append(FLAT_COLUMN_SEPARATOR.join(_cells))
df.columns = _columns
if reset_index and not isinstance(df.index, pd.RangeIndex):
df = df.reset_index(level=0)