Update query_context_processor.py (#17657)
This commit is contained in:
parent
3350076eea
commit
418c0b4e48
|
|
@ -333,6 +333,10 @@ class QueryContextProcessor:
|
|||
def get_data(self, df: pd.DataFrame) -> Union[str, List[Dict[str, Any]]]:
|
||||
if self._query_context.result_format == ChartDataResultFormat.CSV:
|
||||
include_index = not isinstance(df.index, pd.RangeIndex)
|
||||
columns = list(df.columns)
|
||||
verbose_map = self._qc_datasource.data.get("verbose_map", {})
|
||||
if verbose_map:
|
||||
df.columns = [verbose_map.get(column, column) for column in columns]
|
||||
result = csv.df_to_escaped_csv(
|
||||
df, index=include_index, **config["CSV_EXPORT"]
|
||||
)
|
||||
|
|
|
|||
Loading…
Reference in New Issue