10個の色の異なる長方形をmatplotlibを用いて表示させたい。

Question

前提・実現したいこと

質問失礼致します。
今回実装したいことは、タイトルの通り、全て異なる色の１０個の長方形をmatplotlibで表示させたいと思っています。

発生している問題

下記のプログラムを実行したところグラフがひとつしか表示されない状態です。
どこにミスがあるのかご教授お願いします。

該当のソースコード

##プログラムの一部です。
##self.size = [[10, 5], [20, 10], [15, 40], [25, 25], [10, 10], [5, 20], [30, 5], [40, 20], [20, 15], [5, 5]]

import matplotlib
import matplotlib.pyplot as plt
import matplotlib.patches as patches

def mplib(self):
        
        fig = plt.figure()
        ax = fig.add_subplot(111)
        
        colors = ['blue','orange','green','red','purple','brown','pink','gray','olive','cyan']
        
        for x in range(len(self.size)):
            for y in range(1):
                    rec = patches.Rectangle(xy = self.xy[x], width = self.size[x][y], height = self.size[x][y], color = colors[x])
        ax.add_patch(rec)
        ax.set_xlim(0,150)
        ax.set_ylim(0,150)
        plt.show()

Answer 1

元のコードでは全て正方形になってしまいますので、そこも修正しました。

def mplib(self):
        fig = plt.figure()
        ax = fig.add_subplot(111)
        colors = ['blue','orange','green','red','purple','brown','pink','gray','olive','cyan']
        for x in range(len(self.size)):
            rec = patches.Rectangle(xy = self.xy[x], width = self.size[x][0], height = self.size[x][1], color = colors[x])
            ax.add_patch(rec)
        ax.set_xlim(0,150)
        ax.set_ylim(0,150)
        plt.show()

実行結果は以下です。

Comments

[iface]

プログラム修正していただきありがとうございます。
参考にさせていただきます。

Answer 2

動作確認はしていませんが、おそらく、ax.add_patch(rec) のインデントが不適切だからだと思います。正しくは以下の様になるのではないでしょうか。

        for x in range(len(self.size)):
            for y in range(1):
                    rec = patches.Rectangle(xy = self.xy[x], width = self.size[x][y], height = self.size[x][y], color = colors[x])
                    ax.add_patch(rec)

        ax.set_xlim(0,150)
        ax.set_ylim(0,150)
        plt.show()

Answer 3

サイズ、位置、色を自由に指定可能なように class を定義してみました。

p.py

import matplotlib
import matplotlib.pyplot as plt
import matplotlib.patches as patches

class Data:
  default_size_list = []
  default_pos_list = [0, 0]
  default_color_list = ['blue', 'orange', 'green', 'red', 'purple', 'brown', 'pink', 'gray', 'olive', 'cyan']

  def __init__(self, size_list = None, pos_list = None, color_list = None):
    self.size_list = [] if size_list is None else size_list
    self.pos_list = [] if pos_list is None else pos_list
    self.color_list = self.default_color_list if color_list is None else color_list

  def mplib(self):
    fig = plt.figure()
    ax = fig.add_subplot(111)
    pos_num = len(self.pos_list)
    color_num = len(self.color_list)

    for idx in range(len(self.size_list)):
      pos_idx = idx % pos_num
      color_idx = idx % color_num

      rec = patches.Rectangle(
        xy = self.pos_list[idx], 
        width = self.size_list[pos_idx][0], height = self.size_list[pos_idx][1],
        color = self.color_list[color_idx]
      )
      ax.add_patch(rec)

    ax.set_xlim(0,150)
    ax.set_ylim(0,150)
    plt.show()

# data = Data(None, None, None)

size_list = [[10, 5], [20, 10], [15, 40], [25, 25], [10, 10], [5, 20], [30, 5], [40, 20], [20, 15], [5, 5]] 
pos_list = [[10, 5], [20, 10], [15, 40], [25, 25], [10, 10], [5, 20], [30, 5], [40, 20], [20, 15], [5, 5]] 
data = Data(size_list, pos_list, None)

data.mplib()

実行例

Comments

[iface]

class定義をしたプログラムを作っていただきありがとうございます。
参考にさせていただきます。