matplotlibでのグラフの描画に関して

Question

pythonにてMatplotlib.pyplot を用いてグラフの描画をしようとしているのですが、plt.textが入るとグラフが暗転？してしまいグラフが表示されなくなります。解決方法を教えていただけると幸いです。

import numpy as np
import matplotlib.pyplot as plt

n = np.array([i for i in range(100)])
y1 = 8 * n ** 2
y2 = 64 * n * np.log2(n)

idx = np.argwhere(np.diff(np.sign(y1 - y2)) != 0)
plt.plot(n[idx], y1[idx], 'ms', ms=5, label='Intersection', color='green')
plt.xlabel("n")
plt.ylabel("Execution Time")
plt.text(n[idx], y1[idx], "Intersection", fontsize=12)
plt.plot(n, y1, color='red', label='Insertion-sort')
plt.plot(n, y2, color='blue', label='Merge-sort')
plt.legend
plt.grid()
plt.show()

plt.text項を削除した場合のグラフ

plt.text項を残したまま実行した場合
またこの時、グラフ画像の保存さえできなかったです。
よろしくお願いします。

Answer 1

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

を見る限りmatplotlib.pyplot.textには「複数のテキストを一度に生成する」機能がないように見えます。

x, y : scalars <==スカラーを指定せよって書いてあるから

The position to place the text. By default, this is in data coordinates. The coordinate system can be changed using the transform parameter.

素朴な対処としては「個々の交差点に対するテキスト生成をループで行う」方法があります。idxは二次元配列になってますのでiがスカラーになるよう、idx.ravel()としている点にご注意ください。

# plt.text(n[idx], y1[idx], "Intersection", fontsize=12)
for i in idx.ravel():
    plt.text(n[i], y1[i], "Intersection", fontsize=12)

---

ご質問とは無関係ですが凡例が表示できてないと思います。
誤:plt.legend
正:plt.legend()

Comments

[daifukuzaka]

ご回答ありがとうございました！
うまく実行できた上に、errorの意味も理解でき非常にありがたかったです。