Can't set strings as 'facet_col' in px.imshow()

Note that if we change facet_col_wrap to 1, the order of annotation should be different. See this post.