2016-12-23 Jupyter / matplotlib > 2Dイメージをファイル保存する > plt.savefig("foo.png") / ただし、plt.show()があると保存に失敗する模様

動作環境

GeForce GTX 1070 (8GB)
ASRock Z170M Pro4S [Intel Z170chipset]
Ubuntu 14.04 LTS desktop amd64
TensorFlow v0.11
cuDNN v5.1 for Linux
CUDA v8.0
Python 2.7.6
IPython 5.1.0 -- An enhanced Interactive Python.

関連　http://qiita.com/7of9/items/639e4790759dc6fb2cf0

２Dイメージをファイル保存したい。
plt.savefig()を使うらしい。

import numpy as np
import matplotlib.pyplot as plt
import matplotlib.cm as cm

data_1d = np.random.rand(22500)
print(data_1d)
data_2d = np.reshape(data_1d, (150,150))

plt.imshow(data_2d, extent=(0,150,0,150),cmap=cm.gist_rainbow)
# plt.show()

plt.savefig("foo.png")

plt.show()はコメントアウトした。
これが実行されるとfoo.pngのファイルが正しく保存されないから。

バグなのか、仕様なのかは未調査。