Как поменять ось заркаски. В heatmap plotly

Есть heatmap plotly, построенный с помощью примерно вот такого кода:

trace.append(
    go.Heatmap(
    z = pivot_df.values,
    y = pivot_df.index,
    x = pivot_df.columns
))

Который выводит в итоге вот такой heatmap

Таким образом получается что закраска идет по оси абсциссы, то есть какое самое больше число в ряду, то и закрашиваем как самое "горячее".

Вопрос: есть ли возможность сделать так, что бы закраска была по ординате, то есть самое большое число внутри колонки, закрашивается как самое "горячее"