matplotlib использует цвет rgba для текста

Question

Я хотел бы добавить текст с цветом rgba, но matplotlib отклоняет его, это мой минимальный рабочий пример

import matplotlib.pyplot as plt
fig, ax = plt.subplots()
ax.text(0.5,0.5,'wololo', color=np.array([[1,1,1,1]]))
plt.show()

выдает ошибку

TypeError: unhashble type: 'numpy.narray'

Почему-то это работаетсо строковыми цветами (см. следующий пример), но мне бы хотелось, чтобы он работал с любым форматом RGBA

import matplotlib.pyplot as plt
fig, ax = plt.subplots()
ax.text(0.5,0.5,'wololo', color='r')
plt.show()

Как мне этого добиться?

Answer 1

Проблема в том, что указанный вами массив является 2D-массивом.Но цвета RGBA должны быть одномерным массивом или списком.Вы можете использовать

color=np.array([1,1,1,1])            # Use 1D array

или

color=np.array([[1,1,1,1]]).flat   # flatten 2D array