Как заполнить полигон пользовательским штриховкой в ​​matplotlib?

Я использую python и matplotlib для создания нескольких замкнутых полигонов. Затем мне нужно заполнить их штриховкой, что можно сделать через set_hatch.

http://matplotlib.org/api/artist_api.html

http://matplotlib.org/examples/pylab_examples/hatch_demo.html

К сожалению, я работаю с полутоновыми изображениями, и мне нужно больше штриховок, чем предусмотрено по умолчанию - я бы предпочел предоставить растровое изображение (или похожее изображение), которое можно было бы выложить на листы вместо использования этих штриховок с различной плотностью.

Я открыт для других библиотек Python (pyglet, pygame, PIL и т. Д.), Но я бы предпочел, чтобы решение было в Python.

1 ответ

Решение

Вы можете подкласс matplotlib.hatch.Shapes и определить пользовательский люк на основе какой-либо опорной траектории, проведенной внутри единичного квадрата [[-0,5, 0,5] х [-0,5, 0,5]].

Ориентировочное:

import numpy as np
import matplotlib.hatch
import matplotlib.pyplot as plt
from matplotlib.patches import Ellipse, Polygon


house_path = Polygon(
    [[-0.3, -0.4], [0.3, -0.4], [0.3, 0.1], [0., 0.4], [-0.3, 0.1]],
    closed=True, fill=False).get_path()

class CustomHatch(matplotlib.hatch.Shapes):
    """
    Custom hatches defined by a path drawn inside [-0.5, 0.5] square.
    Identifier 'c'.
    """
    filled = True
    size = 1.0
    path = house_path

    def __init__(self, hatch, density):
        self.num_rows = (hatch.count('c')) * density
        self.shape_vertices = self.path.vertices
        self.shape_codes = self.path.codes
        matplotlib.hatch.Shapes.__init__(self, hatch, density)

matplotlib.hatch._hatch_types.append(CustomHatch)

fig = plt.figure()
ax = fig.add_subplot(111)

ellipse = ax.add_patch(Ellipse((0.5, 0.5), 0.3, 0.5, fill=False))
ellipse.set_hatch('c')
ellipse.set_color('red')
plt.show()

Предоставление:

введите описание изображения здесь

Другие вопросы по тегам