Thay đổi màu sắc của các ô vĩ cầm của matplotlib

Question

Có cách nào để thay đổi màu của các ô violin trong matplotlib không?

Màu mặc định là màu "nâu" này, không quá tệ, nhưng tôi muốn tô màu, ví dụ, 3 loại violon đầu tiên khác nhau để làm nổi bật chúng. Tôi không tìm thấy bất kỳ thông số nào trong số documentation. Bất kỳ ý tưởng hoặc hacks để màu sắc violin khác nhau?

Answer 1

matplotlib.pyplot.violinplot() nói nó sẽ trả về:

Từ điển ánh xạ mỗi thành phần của violinplot vào một danh sách các trường hợp bộ sưu tập tương ứng tạo ra. Từ điển có các phím sau:

• bodies: Một danh sách các matplotlib.collections.PolyCollection trường hợp có chứa các khu vực điền của mỗi violin.
• [... số những người khác ...]

Phương pháp PolyCollections bao gồm:

• set_color(c) mà đặt cả facecolor và edgecolor,
• set_facecolor(c) và
• set_edgecolor(c) tất cả đều lấy "màu matplotlib arg hoặc chuỗi rgb một tuples"

Vì vậy, có vẻ như bạn có thể chỉ lặp qua danh sách cơ thể của kết quả và sửa đổi các facecolor của mỗi:

violin_parts = plt.violinplot(...) 

for pc in violin_parts['bodies']: 
    pc.set_facecolor('red') 
    pc.set_edgecolor('black') 

Nó là một chút lạ dù rằng bạn không thể thiết lập này khi tạo ra nó giống như các loại cốt truyện chung. Tôi đoán nó có thể là do các hoạt động tạo ra rất nhiều bit (kể trên PolyCollection cùng với 5 khác LineCollections), rằng các đối số bổ sung sẽ không rõ ràng.

Answer 2

import numpy as np 
import matplotlib.pyplot as plt 
%matplotlib inline 

rrred = '#ff2222' 
bluuu = '#2222ff' 
x = np.arange(2, 25) 
y = np.array([xi * np.random.uniform(0, 1, 10**3) for xi in x]).T 

# Create violin plot objects: 
fig, ax = plt.subplots(1, 1, figsize = (8,8)) 
violin_parts = ax.violinplot(y, x, widths = 0.9, showmeans = True, showextrema = True, showmedians = True) 

# Make all the violin statistics marks red: 
for partname in ('cbars','cmins','cmaxes','cmeans','cmedians'): 
    vp = violin_parts[partname] 
    vp.set_edgecolor(rrred) 
    vp.set_linewidth(1) 

# Make the violin body blue with a red border: 
for vp in violin_parts['bodies']: 
    vp.set_facecolor(bluuu) 
    vp.set_edgecolor(rrred) 
    vp.set_linewidth(1) 
    vp.set_alpha(0.5)