Элегантное изменение цвета рамки сюжета в matplotlib

Это своего рода последующий вопрос к сообщению this, в котором обсуждалась окраска осей, тиков и меток. Я надеюсь, что это нормально, чтобы открыть новый расширенный вопрос для этого.

Изменение цвета полного кадра (тиков и осей) вокруг двойного графика (через add_subplot) с помощью осей [ax1, ax2] приводит к большому количеству кода. Этот фрагмент изменяет цвет рамки верхнего участка:

ax1.spines['bottom'].set_color('green')
ax1.spines['top'].set_color('green')
ax1.spines['left'].set_color('green')
ax1.spines['right'].set_color('green')
for t in ax1.xaxis.get_ticklines(): t.set_color('green')
for t in ax1.yaxis.get_ticklines(): t.set_color('green')
for t in ax2.xaxis.get_ticklines(): t.set_color('green')
for t in ax2.yaxis.get_ticklines(): t.set_color('green')

Итак, для изменения цвета рамки двух графиков с двумя у-осями каждый мне понадобится 16 (!) строк кода... Вот как это выглядит:

enter image description here

Другие методы, которые я докопал до сих пор:

  • matplotlib.rc: обсуждался здесь; изменяет глобально, а не локально. Я хочу иметь некоторые другие сюжеты в разных цветах. Пожалуйста, не обсуждайте слишком много цветов на сюжетах...: -)

    matplotlib.rc('axes',edgecolor='green')

  • выкопайте шипы оси, затем измените ее: также обсуждается здесь; не очень элегантный, я думаю.

    for child in ax.get_children():
    if isinstance(child, matplotlib.spines.Spine):
        child.set_color('#dddddd')

Есть ли элегантный способ уплотнения вышеуказанного блока, что-то более "pythonic"?

Я использую python 2.6.5 с matplotlib 0.99.1.1 под ubuntu.

Ответы

Ответ 1

Рефакторинг вашего кода выше:

import matplotlib.pyplot as plt

for ax, color in zip([ax1, ax2, ax3, ax4], ['green', 'green', 'blue', 'blue']):
    plt.setp(ax.spines.values(), color=color)
    plt.setp([ax.get_xticklines(), ax.get_yticklines()], color=color)

Ответ 2

Предполагая, что вы используете достаточно обновленную версию matplotlib ( >= 1.0), возможно, попробуйте что-то вроде этого:

import matplotlib.pyplot as plt

# Make the plot...
fig, axes = plt.subplots(nrows=2)
axes[0].plot(range(10), 'r-')
axes[1].plot(range(10), 'bo-')

# Set the borders to a given color...
for ax in axes:
    ax.tick_params(color='green', labelcolor='green')
    for spine in ax.spines.values():
        spine.set_edgecolor('green')

plt.show()

enter image description here

Ответ 3

Может быть, немного грубо ответить на мой собственный вопрос, но я хотел бы поделиться тем, что мог найти до сих пор. Эта версия может окрасить два подзаголовка с осями [ax1, ax2] и [ax3, ax4] в двух разных цветах. Это намного короче, чем 16 строк, которые я изложил в моем вопросе выше. Он вдохновлен Джо Кингтоном на ответ здесь, и в doublex убивает цвет метки метки.

import matplotlib.pyplot as plt
import numpy as np

# Generate some data
num = 200
x = np.linspace(501, 1200, num)
yellow_data, green_data , blue_data= np.random.random((3,num))
green_data += np.linspace(0, 3, yellow_data.size)/2
blue_data += np.linspace(0, 3, yellow_data.size)/2

fig = plt.figure()
plt.subplot(211) # Upper Plot
ax1 = fig.add_subplot(211)
ax1.fill_between(x, 0, yellow_data, color='yellow')
ax2 = ax1.twinx()
ax2.plot(x, green_data, 'green')
plt.setp(plt.gca(), xticklabels=[])
plt.subplot(212) # Lower Plot
ax3 = fig.add_subplot(212)
ax3.fill_between(x, 0, yellow_data, color='yellow')
ax4 = ax3.twinx()
ax4.plot(x, blue_data, 'blue')

# Start coloring
for ax, color in zip([ax1, ax2, ax3, ax4], ['green', 'green', 'blue', 'blue']):
    for ticks in ax.xaxis.get_ticklines() + ax.yaxis.get_ticklines():
        ticks.set_color(color)
    for pos in ['top', 'bottom', 'right', 'left']:
        ax.spines[pos].set_edgecolor(color)
# End coloring

plt.show()

color_plots

Я отметил это как принятое, так как это самое компактное решение, которое я смог найти до сих пор. Тем не менее, я открыт для других, может быть, более элегантных способов его решения.