`zmax` argument not effective in `px.imshow` when `img` contains `np.nan`

Results for

import pandas as pd
import numpy as np
import plotly.express as px

df = pd.DataFrame({"foo": [1, 2, 3]})
px.imshow(df, zmax=2)

Results for

import pandas as pd
import numpy as np
import plotly.express as px

df = pd.DataFrame({"foo": [1, 2, 3, np.nan]})
px.imshow(df, zmax=2)

When df contains np.nan, zmax is not taking effect. Intesrestingly, if zmin is specified at the same time, it works.

Beginne damit, die beiden px.imshow-Beispiele aus dem Issue auszuführen, und vergleiche das Verhalten von zmax mit und ohne np.nan. Suche die Implementierung von px.imshow und die vorhandenen Tests und füge anschließend eine Abdeckung hinzu, die zeigt, dass zmax=2 für Daten mit np.nan weiterhin wirksam bleibt; das Issue ist abgeschlossen, wenn dieser Regressionstest erfolgreich ist.