fig, [ax] = plt.subplots(1, 1,figsize=(1,1))とすると、TypeError: 'AxesSubplot' object is not iterable

Question

pythonのmatplotlibを利用してグラフを作っています。
以前、動画を二つ並べる方法を教えていただき、
そのコードを改造して、1つだけにしようとしたところで、つまずいています。
改造手出来たコードは下のようになっています。

import numpy as np
import matplotlib.pyplot as plt
from matplotlib.animation import FuncAnimation
from PIL import Image

def update(time, ax):
    x = np.empty((100), float)
    y = np.empty((100), float)
    for i in range(100):
        x[i] = 2 * np.pi * i / 100

    for i in range(100):
        y[i] = np.sin(x[i] + time * 0.1)
    ax.clear()
    ax.plot(x, y, color='black')

fig, [ax] = plt.subplots(1, 1,figsize=(1,1))
anim = FuncAnimation(fig, update, fargs=(ax), interval=100, frames=100)
anim.save('fig.gif', writer='imagemagick')

このコードを実行すると次のエラーが出ます。

QStandardPaths: XDG_RUNTIME_DIR not set, defaulting to '/tmp/runtime'
Traceback (most recent call last):
  File "test.py", line 17, in <module>
    fig, [ax] = plt.subplots(1, 1,figsize=(1,1))
TypeError: 'AxesSubplot' object is not iterable

原因・解決方法のわかる方、ご教授ください。

環境は次の通りです。

$ python
Python 3.6.5 |Anaconda, Inc.| (default, Apr 29 2018, 16:14:56)
[GCC 7.2.0] on linux
Type "help", "copyright", "credits" or "license" for more information.
>>> import matplotlib
>>> matplotlib.__version__
'2.2.2'

Answer 1

subplots() の２番目の戻り値の仕様は

https://matplotlib.org/api/_as_gen/matplotlib.pyplot.subplots.html

ax can be either a single Axes object or an array of Axes objects if more than one subplot was created.

となっております。つまり

plt.subplots(1, 1,figsize=(1,1))

の場合、２番目の戻り値は Axesの配列ではなく、Axes Objectとなりますので、記述方法は

fig, ax = plt.subplots(1, 1, figsize=(1,1))

となるかと思います。

Comments

[magichan]

被りましたね、すみません。

Answer 2

問題点

2点問題点があります。

1. Axes が1つの場合の subplots() の返り値

nrows=1, ncols=1 のように図が1つのときは、[Axes] ではなく、Axes オブジェクトを返すので、
fig, [ax] = plt.subplots(figsize=(5, 5)) のように受けたらエラーになります。

エラーが出たときはまず公式リファレンスを確認することをおすすめします。

matplotlib.pyplot.subplots()

ax can be either a single Axes object or an array of Axes objects if more than one subplot was created.

訳

ax は (1つの Axes を作成する場合は) 単一の Axes オブジェクト、2つ以上の Axes が作成する場合は Axes オブジェクトの配列になります。

ちなみに nrows=1, ncols=1 がデフォルト引数なので、Axes が1つのときは明示的に指定しなくてもよいです。

2. 要素が1つのタプルを作る場合はカンマが必要

要素が1つのタプルの場合、末尾にカンマが必要です。
カンマをつけない場合、(1) は 1 と同値になります。

print(type((1)))  # <class 'int'>
print(type((1,)))  # <class 'tuple'>

そのため、fargs=(ax) ではなく、fargs=(ax,) とします。

修正したコード

import matplotlib.pyplot as plt
import numpy as np
from matplotlib.animation import FuncAnimation
from PIL import Image

def update(time, ax):
    x = np.empty((100), float)
    y = np.empty((100), float)
    for i in range(100):
        x[i] = 2 * np.pi * i / 100

    for i in range(100):
        y[i] = np.sin(x[i] + time * 0.1)
    ax.clear()
    ax.plot(x, y, color='black')

fig, ax = plt.subplots(figsize=(5, 5))
anim = FuncAnimation(fig, update, fargs=(ax,), interval=100, frames=100)
anim.save('fig.gif', writer='imagemagick')