4
5

Delete article

Deleted articles cannot be recovered.

Draft of this article would be also deleted.

Are you sure you want to delete this article?

More than 5 years have passed since last update.

axes.revelで2次元配列を1次元配列に変換し、効率よくプロットする。

Last updated at Posted at 2018-03-09

毎日、pythonと格闘しているumesakuです。「pythonで始める機械学習」のP.144に謎の一文があったのでちょっと掘り下げてみました。

###画像を愚直にプロットする

qiita.rb
from sklearn.datasets import fetch_lfw_people
import matplotlib.pyplot as plt
%matplotlib inline
#jupyter notebookで動かす時はこの呪文が必要になります。

people = fetch_lfw_people(min_faces_per_person=20, resize=0.7)

# 先に名前のリストをとっておく
names = []
for t in people.target:
    names.append(people.target_names[t])

# 名前を短くしたいので写真は別の変数名にする
images = people.images

# テキストでは2行5列。ちょっと楽をして2行3列に写真を並べた。
fix, axis = plt.subplots(2, 3, figsize=(15, 8), subplot_kw={'xticks':[], 'yticks':[]}) 

# サブプロットの画像表示メソッドがimshow
# サブプロットのタイトルに名前表示。setを忘れずに
axis[0, 0].imshow(images[0]) 
axis[0, 0].set_title(names[0]) 
axis[0, 1].imshow(images[1])
axis[0, 1].set_title(names[1])
axis[0, 2].imshow(images[2])
axis[0, 2].set_title(names[2])
axis[1, 0].imshow(images[3])
axis[1, 0].set_title(names[3])
axis[1, 1].imshow(images[4])
axis[1, 1].set_title(names[4])
axis[1, 2].imshow(images[5])
axis[1, 2].set_title(names[5])
plt.show()

上記のコードでは一つずつに対して写真と名前をセットしているが、サブプロットの数が増えてくると、どんどんどんどん大変。100枚とか、もー無理(汗)。

ループで効率よく処理したいんだけど、axis が 2 次元配列で、names/images が 1 次元配列だから単純なループにしたらエラーになっちゃう。私では到底処理できない(いやいや簡単にできるよという方、ぜひ教えてください!)

だから、axisをaxis.revel()で一次元配列にする。
そうすれば、names/images/axis が一つのループで書けるようになる。

###テキスト通り書いてみた
P.144です。ちょっとaxis1 = axis.ravel()でアレンジしています。

qiita.rb
fix, axis = plt.subplots(2, 3, figsize=(15, 8), subplot_kw={'xticks':[], 'yticks':[]}) 

#axisを1 次元配列にする
axis1 = axis.ravel() 

# これで一つのループで書ける
for name, image, ax in zip(names, images, axis1): 
    ax.imshow(image,cmap='Greys_r')
    ax.set_title(name)

plt.show()

###アウトプット

どちらもアウトプットは同じです。なぜか黄緑色になっちゃいますけど(涙
axes.revelは覚えておこう!

※以前axesとaxisは違うとご指摘頂きました。調べても良く理解できませんでした。なので、今回両方試行したのですがどちらも同じ結果になりました。また、見て頂けてご指摘頂けたら嬉しいです!

LFW.png

追、@skotaroさんのアドバイスで、ax.imshow(image, cmap='Greys_r')とすると白黒写真のようになりました!cmapとはカラーマップ、_rはreverseの略で'Grey'だけだと白黒写真の反転バージョン、ということのようです!

4
5
4

Register as a new user and use Qiita more conveniently

  1. You get articles that match your needs
  2. You can efficiently read back useful information
  3. You can use dark theme
What you can do with signing up
4
5

Delete article

Deleted articles cannot be recovered.

Draft of this article would be also deleted.

Are you sure you want to delete this article?