python グラフの曲線の色や凡例などの決定

Question

from pandas import DataFrame as df
import numpy as np
%matplotlib notebook
import matplotlib
import matplotlib.pyplot as plt

arr = np.random.randn(10,10)
name = list(range(0,10))

#グラフ描画
fig = plt.figure()
ax = fig.add_subplot(1, 1, 1)

#プロット
plt.plot(df(arr),label=name)

出力が以下です．

このグラフに凡例を追加したく，listなども与えてみましたがうまくいきませんでした．
また色や線種はすべてではなく1つずつ変えることはできますでしょうか．
通常のやり方では，すべてが同一の色になってしまいます．

以上，よろしくお願いいたします．

Answer 1

とりあえず、ラベル・色・線種などをリストで保持して、ループで書くのが簡単ではないでしょうか。

from pandas import DataFrame as df
import numpy as np
%matplotlib notebook
import matplotlib
import matplotlib.pyplot as plt

arr = np.random.randn(10,10)
name = list(range(0,10))

# ラベル
labels = ['a','b','c','d','e','f','g','h','i','j']
# 色
colors = ['blue','pink','green','orange','gray','black','yellow','cyan','purple','red']
# 線種
linestyles = ['solid','dashed','dashdot','dotted','solid','dashed','dashdot','dotted','solid','dashed']

# グラフ描画
fig = plt.figure()
ax = fig.add_subplot(1, 1, 1)

# プロット
for i in range(arr.shape[1]):
    ax.plot(arr[:,i], color=colors[i], label=labels[i], linestyle=linestyles[i])

# 凡例を表示
ax.legend()

Comments

[shougi]

返信が遅れ申し訳ありません．

やはり，forでやるのがはやいんですね
ありがとうございました．