plt.contourの線の色の変更の仕方

Question

実現したいこと

「該当のソースコード」はこのページを元にしたものですが、私の質問内容の具体例として挙げただけで、深い意味はありません。6、7行目のデータはこちらのCase BのCase 5: low particle density and small particleです。

「該当のソースコード」を実行すると、以下の図が出てきます。(test1.png)

ここで、等高線を描きたい量（displacement）は整数値しかとらないことに気づいたため、等高線をcontourfからcontourに変更しようと思いました。（48行目のcontourf→contour）
その実行結果が以下の図です。(test2.png)

test2.pngの色が少し見づらいので、それを設定する方法を教えてください。
test1.pngは4色、test2.pngは5色必要なので、そもそもtest1.pngから自分で1色増やさなければいけません。

例えば、test2.pngの色を、上から順に紫、赤、黄、緑、青と設定したいです。

該当のソースコード

import matplotlib.pyplot as plt
import numpy as np
from skimage.io import imread
from scipy.signal import correlate

path1 = r'.\B005_1.tif'
path2 = r'.\B005_2.tif'

a = imread(path1)
b = imread(path2)

win_size = 32

def vel_field(curr_frame, next_frame, win_size):
    ys = np.arange(0, curr_frame.shape[0], win_size)
    xs = np.arange(0, curr_frame.shape[1], win_size)
    dys = np.zeros((len(ys), len(xs)))
    dxs = np.zeros((len(ys), len(xs)))
    
    for iy, y in enumerate(ys):
        for ix, x in enumerate(xs):
            int_win = curr_frame[y : y + win_size, x : x + win_size]
            search_win = next_frame[y : y + win_size, x : x + win_size]
            cross_corr = correlate(
                search_win - search_win.mean(), int_win - int_win.mean(), method="fft"
            )
            dys[iy, ix], dxs[iy, ix] = (    
                np.unravel_index(np.argmax(cross_corr), cross_corr.shape)
                - np.array([win_size, win_size])
                + 1                        
            )
                      
    # draw velocity vectors from the center of each window
    ys = ys + win_size / 2
    xs = xs + win_size / 2

    return xs, ys, dxs, dys
   
xs, ys, dxs, dys = vel_field(a, b, win_size)

x = xs
y = ys[::-1]
x, y = np.meshgrid(x, y)
z = dys

levels1 = np.linspace(-10,10,5)
levels2= np.linspace(-10,10,5)
ax2 = plt.contourf(x,y,z,cmap='jet',levels=levels2)
ax2.set_clim(vmin=min(levels2), vmax=max(levels2))
cb = plt.colorbar(ticks=levels1)
cb.set_label("y-direction displacement[pixel]", fontsize=16)
cb.ax.tick_params(labelsize=12)
plt.xlabel("x[pixel]", fontsize=12)
plt.ylabel("y[pixel]", fontsize=12)
plt.xticks(fontsize=12)
plt.yticks(fontsize=12)
plt.savefig('test1.png')

試したこと

ネットで色々調べましたが、よく分かりませんでした。

「該当のソースコード」の48行目を
ax2 = plt.contour(x,y,z,cmap='jet',levels=levels2,colors=['purple', 'r', 'y', 'g', 'b'])
としてみましたが、下記のようなエラーメッセージが出てしまいました。

Traceback (most recent call last):
  File "・・・\testCV228-2.py", line 69, in <module>
    ax2 = plt.contour(x,y,z,cmap='jet',levels=levels2,colors=['purple', 'r', 'y', 'g', 'b'])
          ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
  File "・・・\Python311\Lib\site-packages\matplotlib\pyplot.py", line 2527, in contour
    __ret = gca().contour(
            ^^^^^^^^^^^^^^
  File "・・・\Python311\Lib\site-packages\matplotlib\__init__.py", line 1461, in inner
    return func(ax, *map(sanitize_sequence, args), **kwargs)
           ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
  File "・・・\Python311\Lib\site-packages\matplotlib\axes\_axes.py", line 6449, in contour
    contours = mcontour.QuadContourSet(self, *args, **kwargs)
               ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
  File "・・・\Python311\Lib\site-packages\matplotlib\contour.py", line 756, in __init__
    raise ValueError('Either colors or cmap must be None')
ValueError: Either colors or cmap must be None

Comments

[8524ba23]

https://teratail.com/help/question-tips は読みましたか？
読んだ結果を本文に反映ください。解決のヒントを探れるかもしれません。
「調べましたが、よく分かりません」の内容が分からないので
「～は調べましたか？」→「それはすでに調べましたが分かりませんでした」
といった無駄なやり取りが発生する可能性があります。

[yyicp]

>can110様
コメントありがとうございます。
「試したこと」を加筆しましたのでご確認ください。

[melian]

> Either colors or cmap must be None

と表示されているので、colors だけ指定してみてはどうでしょうか。

ax2 = plt.contour(x,y,z,levels=levels2,colors=['purple', 'r', 'y', 'g', 'b'])

[yyicp]

>melian様
できました。
ありがとうございました。

ベストアンサーに選びたいので、解答欄に記入していただきますようお願いします。
（図の添付は不要です）

なお、その際はcolorsの中の色の順番を逆にしてください。（私のミスです。申し訳ありません）
ax2 = plt.contour(x,y,z,levels=levels2,colors=['b', 'g', 'y', 'r', 'purple'])

[8524ba23]

解決おめでとうございます！
解決方法をまとめて、「自己解決」として回答してください！
同じように困っている人の貴重な情報になると思います。

[yyicp]

>can110様
ありがとうございます。
melian様に回答欄に記入するようお願いしました。

[8524ba23]

解決おめでとうございます！
解決方法をまとめて、「自己解決」として回答してください！
同じように困っている人の貴重な情報になると思います。
https://teratail.com/help/question-tips#questionTips42

Answer 1

※ コメントから転記

Either colors or cmap must be None

と表示されていますので、colors だけ指定するとよいかと思います。

ax2 = plt.contour(x,y,z,levels=levels2,colors=['b', 'g', 'y', 'r', 'purple'])

Comments

[yyicp]

できました。
どうもありがとうございました。