Màu sắc một phần của văn bản trong matplotlib

Question

Có cách nào trong matplotlib xác định một phần màu của chuỗi không?

Ví dụ:

plt.ylabel("Today is cloudy.") 

Làm thế nào tôi có thể hiển thị "hôm nay" như đỏ, "là" như màu xanh lá cây và "mây". màu xanh?

Cảm ơn.

Answer 1

Tôi chỉ biết cách thực hiện điều này không tương tác và thậm chí chỉ với phần phụ trợ 'PS'.

Để làm điều này, tôi sẽ sử dụng Latex để định dạng văn bản. Sau đó, tôi sẽ bao gồm các gói 'màu', và thiết lập màu sắc của bạn như bạn muốn.

Dưới đây là một ví dụ để làm điều này:

import matplotlib 
matplotlib.use('ps') 
from matplotlib import rc 

rc('text',usetex=True) 
rc('text.latex', preamble='\usepackage{color}') 
import matplotlib.pyplot as plt 

plt.figure() 
plt.ylabel(r'\textcolor{red}{Today} '+ 
      r'\textcolor{green}{is} '+ 
      r'\textcolor{blue}{cloudy.}') 
plt.savefig('test.ps') 

Điều này dẫn đến (chuyển đổi từ ps để png sử dụng ImageMagick, vì vậy tôi có thể đăng nó ở đây):

Answer 2

đây là phiên bản tương tác (cùng một tôi được đưa lên the list)

import matplotlib.pyplot as plt 
from matplotlib import transforms 

def rainbow_text(x,y,ls,lc,**kw): 
    """ 
    Take a list of strings ``ls`` and colors ``lc`` and place them next to each 
    other, with text ls[i] being shown in color lc[i]. 

    This example shows how to do both vertical and horizontal text, and will 
    pass all keyword arguments to plt.text, so you can set the font size, 
    family, etc. 
    """ 
    t = plt.gca().transData 
    fig = plt.gcf() 
    plt.show() 

    #horizontal version 
    for s,c in zip(ls,lc): 
     text = plt.text(x,y," "+s+" ",color=c, transform=t, **kw) 
     text.draw(fig.canvas.get_renderer()) 
     ex = text.get_window_extent() 
     t = transforms.offset_copy(text._transform, x=ex.width, units='dots') 

    #vertical version 
    for s,c in zip(ls,lc): 
     text = plt.text(x,y," "+s+" ",color=c, transform=t, 
       rotation=90,va='bottom',ha='center',**kw) 
     text.draw(fig.canvas.get_renderer()) 
     ex = text.get_window_extent() 
     t = transforms.offset_copy(text._transform, y=ex.height, units='dots') 


plt.figure() 
rainbow_text(0.5,0.5,"all unicorns poop rainbows ! ! !".split(), 
     ['red', 'orange', 'brown', 'green', 'blue', 'purple', 'black'], 
     size=40) 

Answer 3

Mở rộng câu trả lời Yann của, màu LaTeX tại also works with PDF export:

import matplotlib 
from matplotlib.backends.backend_pgf import FigureCanvasPgf 
matplotlib.backend_bases.register_backend('pdf', FigureCanvasPgf) 

import matplotlib.pyplot as plt 

pgf_with_latex = { 
    "text.usetex": True,   # use LaTeX to write all text 
    "pgf.rcfonts": False,   # Ignore Matplotlibrc 
    "pgf.preamble": [ 
     r'\usepackage{color}'  # xcolor for colours 
    ] 
} 
matplotlib.rcParams.update(pgf_with_latex) 

plt.figure() 
plt.ylabel(r'\textcolor{red}{Today} '+ 
      r'\textcolor{green}{is} '+ 
      r'\textcolor{blue}{cloudy.}') 
plt.savefig("test.pdf") 

Lưu ý rằng kịch bản python này đôi khi không thành công với Undefined control sequence sai sót trong nỗ lực đầu tiên. Chạy nó một lần nữa là sau đó thành công.