Matplotlib Текст не принимает значения массива

Question

Вот код, который я попробовал:

import matplotlib.pyplot as plt
import numpy as np
bb = [1,2,3,4,5,6,7,8,9,10]
cc = ["red","red","yellow","red","green","red","red","green","red","red"]
tt = ["\u2714","\u2714""\u2718","\u2714","\u2718","\u2714","\u2714","\u2718","\u2714","\u2714"]
x1=np.arange(10)
x2=np.arange(10)
fig = plt.figure()
fig.set_size_inches(50,70)
ax1 = fig.add_subplot(331)

ax1.bar(np.arange(len(bb)), bb, color=cc,width=0.6)
text_applied = ax1.text(x1,2,tt,color=cc)

plt.show()

Он давал бары без каких-либо проблем ранее.Но не работает с текстом.Я получаю следующую ошибку:

Traceback (most recent call last):
  File "C:\Python35\lib\tkinter\__init__.py", line 1549, in __call__
    return self.func(*args)
  File "C:\Python35\lib\tkinter\__init__.py", line 596, in callit
    func(*args)
  File "C:\Python35\lib\site-packages\matplotlib\backends\_backend_tk.py", line 310, in idle_draw
    self.draw()
  File "C:\Python35\lib\site-packages\matplotlib\backends\backend_tkagg.py", line 12, in draw
    super(FigureCanvasTkAgg, self).draw()
  File "C:\Python35\lib\site-packages\matplotlib\backends\backend_agg.py", line 433, in draw
    self.figure.draw(self.renderer)
  File "C:\Python35\lib\site-packages\matplotlib\artist.py", line 55, in draw_wrapper
    return draw(artist, renderer, *args, **kwargs)
  File "C:\Python35\lib\site-packages\matplotlib\figure.py", line 1475, in draw
    renderer, self, artists, self.suppressComposite)
  File "C:\Python35\lib\site-packages\matplotlib\image.py", line 141, in _draw_list_compositing_images
    a.draw(renderer)
  File "C:\Python35\lib\site-packages\matplotlib\artist.py", line 55, in draw_wrapper
    return draw(artist, renderer, *args, **kwargs)
  File "C:\Python35\lib\site-packages\matplotlib\axes\_base.py", line 2607, in draw
    mimage._draw_list_compositing_images(renderer, self, artists)
  File "C:\Python35\lib\site-packages\matplotlib\image.py", line 141, in _draw_list_compositing_images
    a.draw(renderer)
  File "C:\Python35\lib\site-packages\matplotlib\artist.py", line 55, in draw_wrapper
    return draw(artist, renderer, *args, **kwargs)
  File "C:\Python35\lib\site-packages\matplotlib\text.py", line 706, in draw
    bbox, info, descent = textobj._get_layout(renderer)
  File "C:\Python35\lib\site-packages\matplotlib\text.py", line 282, in _get_layout
    key = self.get_prop_tup(renderer=renderer)
  File "C:\Python35\lib\site-packages\matplotlib\text.py", line 862, in get_prop_tup
    x, y = self.get_unitless_position()
  File "C:\Python35\lib\site-packages\matplotlib\text.py", line 844, in get_unitless_position
    x = float(self.convert_xunits(self._x))
TypeError: only size-1 arrays can be converted to Python scalars

Пожалуйста, дайте мне знать, что я могу сделать, чтобы отобразить тексты в соответствии с цветом и указанным x values.

Answer 1

Я не совсем уверен, каков ваш ожидаемый результат, но если вы хотите, чтобы символы с разными цветами под каждой полосой вы могли установить их в виде галочек, как это:

import matplotlib.pyplot as plt
import numpy as np

bb = [1, 2, 3, 4, 5, 6, 7, 8, 9, 10]
cc = ["red", "red", "yellow", "red", "green", "red", "red", "green", "red", "red"]
tt = ["\u2714", "\u2714", "\u2718", "\u2714", "\u2718", "\u2714", "\u2714", "\u2718", "\u2714", "\u2714"]

fig = plt.figure()
ax1 = fig.add_subplot(111)
ax1.bar(range(len(bb)), bb, color=cc, width=0.6)
ax1.set_xticks(range(len(bb)))
ax1.set_xticklabels(tt)
for xtick, color in zip(ax1.get_xticklabels(), cc):
    xtick.set_color(color)
plt.show()

Вывод:

Кстати, обратите внимание, что в вашем коде tt пропущена запятая между вторым и третьим элементами.

Answer 2

Я не совсем понял вашу проблему.В документах plt.text() я ничего не могу найти о возможности ввода массивов строк в параметры:

Определение : текст (x,y, s, fontdict = Нет, withdash = False, ** kwargs)

Тип: Функция модуля matplotlib.pyplot

Добавить текст к осям.

Добавитьтекст s по осям в точке x, y в координатах данных.

Параметры

x, y:
скаляры Позиция для размещения текста.По умолчанию это в координатах данных.Систему координат можно изменить с помощью параметра преобразования.
s: str Текст.

Поэтому я бы порекомендовал просто

for x, t, c in zip(x1, tt, cc):
    ax1.text(x, 2, t, color=c)