Matplotlib завершается с ошибкой ValueError: невозможно преобразовать число с плавающей запятой в целое число

Question

Я столкнулся со странной проблемой с библиотекой Seaborn. При создании барплота для данных в диапазоне от очень низких до очень высоких значений, например ::100100

       job      duration type
    0    1  83066.639344    A
    1    2    820.700000    B

не получается с:

ValueError: cannot convert float NaN to integer

Это похоже на ошибку в matplotlib и дубликат "pyplot.savefig завершается с ошибкой ValueError: невозможно преобразовать число с плавающей запятой в целое число" . Последнее еще не исправлено. Есть ли обходной путь для этого?

Вот минимальный рабочий пример для воспроизведения проблемы:

#!/usr/bin/env python3

import matplotlib.pyplot as plt
import pandas as pd
import seaborn as sns

d = {'job': [1, 2]),
     'duration': [83066.639344, 820.700000],
     'type': ['A', 'B']}

df = pd.DataFrame(d)

plot = sns.catplot(x="duration", y="job", data=df, hue='type',
                   color="b", kind="bar", height=3, aspect=4)

ax = plot.axes.flat[0]
for p in plt.gca().patches:
    ax.text(p.get_width(),
            p.get_y() + p.get_height() / 2,
            p.get_width())

plot.savefig("barplot.png")

Некоторые наблюдения:

1. Проблема не возникает, когда я не делаю различий между «типом» (без использования hue='type').

Вот полная трассировка стека:

posx and posy should be finite values
posx and posy should be finite values
/Users/dzieciou/virtualenvs/seaborn/lib/python3.7/site-packages/numpy/core/fromnumeric.py:83: RuntimeWarning: invalid value encountered in reduce
  return ufunc.reduce(obj, axis, dtype, out, **passkwargs)
posx and posy should be finite values
posx and posy should be finite values
posx and posy should be finite values
Traceback (most recent call last):
  File "/Users/dzieciou/projects/example/gocd/reproduce.py", line 31, in <module>
    plot.savefig("barplot.png")
  File "/Users/dzieciou/virtualenvs/seaborn/lib/python3.7/site-packages/seaborn/axisgrid.py", line 37, in savefig
    self.fig.savefig(*args, **kwargs)
  File "/Users/dzieciou/virtualenvs/seaborn/lib/python3.7/site-packages/matplotlib/figure.py", line 2094, in savefig
    self.canvas.print_figure(fname, **kwargs)
  File "/Users/dzieciou/virtualenvs/seaborn/lib/python3.7/site-packages/matplotlib/backend_bases.py", line 2075, in print_figure
    **kwargs)
  File "/Users/dzieciou/virtualenvs/seaborn/lib/python3.7/site-packages/matplotlib/backends/backend_agg.py", line 510, in print_png
    FigureCanvasAgg.draw(self)
  File "/Users/dzieciou/virtualenvs/seaborn/lib/python3.7/site-packages/matplotlib/backends/backend_agg.py", line 402, in draw
    self.figure.draw(self.renderer)
  File "/Users/dzieciou/virtualenvs/seaborn/lib/python3.7/site-packages/matplotlib/artist.py", line 50, in draw_wrapper
    return draw(artist, renderer, *args, **kwargs)
  File "/Users/dzieciou/virtualenvs/seaborn/lib/python3.7/site-packages/matplotlib/figure.py", line 1649, in draw
    renderer, self, artists, self.suppressComposite)
  File "/Users/dzieciou/virtualenvs/seaborn/lib/python3.7/site-packages/matplotlib/image.py", line 138, in _draw_list_compositing_images
    a.draw(renderer)
  File "/Users/dzieciou/virtualenvs/seaborn/lib/python3.7/site-packages/matplotlib/artist.py", line 50, in draw_wrapper
    return draw(artist, renderer, *args, **kwargs)
  File "/Users/dzieciou/virtualenvs/seaborn/lib/python3.7/site-packages/matplotlib/axes/_base.py", line 2610, in draw
    mimage._draw_list_compositing_images(renderer, self, artists)
  File "/Users/dzieciou/virtualenvs/seaborn/lib/python3.7/site-packages/matplotlib/image.py", line 138, in _draw_list_compositing_images
    a.draw(renderer)
  File "/Users/dzieciou/virtualenvs/seaborn/lib/python3.7/site-packages/matplotlib/artist.py", line 50, in draw_wrapper
    return draw(artist, renderer, *args, **kwargs)
  File "/Users/dzieciou/virtualenvs/seaborn/lib/python3.7/site-packages/matplotlib/axis.py", line 1185, in draw
    ticks_to_draw = self._update_ticks(renderer)
  File "/Users/dzieciou/virtualenvs/seaborn/lib/python3.7/site-packages/matplotlib/axis.py", line 1023, in _update_ticks
    tick_tups = list(self.iter_ticks())  # iter_ticks calls the locator
  File "/Users/dzieciou/virtualenvs/seaborn/lib/python3.7/site-packages/matplotlib/axis.py", line 967, in iter_ticks
    majorLocs = self.major.locator()
  File "/Users/dzieciou/virtualenvs/seaborn/lib/python3.7/site-packages/matplotlib/ticker.py", line 1985, in __call__
    return self.tick_values(vmin, vmax)
  File "/Users/dzieciou/virtualenvs/seaborn/lib/python3.7/site-packages/matplotlib/ticker.py", line 1993, in tick_values
    locs = self._raw_ticks(vmin, vmax)
  File "/Users/dzieciou/virtualenvs/seaborn/lib/python3.7/site-packages/matplotlib/ticker.py", line 1932, in _raw_ticks
    nbins = np.clip(self.axis.get_tick_space(),
  File "/Users/dzieciou/virtualenvs/seaborn/lib/python3.7/site-packages/matplotlib/axis.py", line 2543, in get_tick_space
    return int(np.floor(length / size))
ValueError: cannot convert float NaN to integer

Answer 1

Обратите внимание, что это на самом деле не ошибка, и она не связана со связанной ошибкой, которая действительно исправлена.

Можно утверждать, что при отображении текста с координатами nan должно быть лучшее сообщение об ошибке.

Прежде чем посмотреть на ошибку, кажется, у вас есть другая проблема в вашем коде, которая заключается в том, что вы устанавливаете координату x текста в ширину полосы. Они обычно не связаны, и вы могли бы вместо этого использовать p.get_x().

Теперь два варианта:

1. Не размещайте текст в недопустимых координатах.

    import numpy as np
    import matplotlib.pyplot as plt
    import pandas as pd
    import seaborn as sns

    d = {'job': list(range(1, 3)),
         'duration': [83066.639344, 820.700000],
         'type': ['A', 'B']}

    df = pd.DataFrame(d)

    plot = sns.catplot(x="duration", y="job", data=df, hue='type',
                       color="b", kind="bar", height=3, aspect=4)

    ax = plot.axes.flat[0]
    for p in plt.gca().patches:
        height = np.nan_to_num(p.get_height(), 0)
        ax.text(p.get_x(), p.get_y() + height/2., "My text")

    plot.savefig("barplot.png")
    plt.show()

2. Не используйте bbox_inches="tight".

Если вы хотите сохранить свой код таким, какой он есть, вы можете обойти это, не устанавливая параметры bbox_inches="tight" в savefig.

.

plot.savefig("barplot.png", bbox_inches=None)

Или используйте опцию savefig в matplotlib

plot.fig.savefig("barplot.png")