画像を表示させて選択を行い選択別に処理を行う方法

Question

前提・実現したいこと

画像を２つ提示して好きな方を選択してもらい
その選択毎に別の処理をしていきたいです。

また、選択を決めるまでの時間も図りたいです。

プログラミング初心者です。

該当のソースコード

from PIL import Image
from matplotlib import pyplot as plt
import numpy as np

%matplotlib inline 

sampleA = Image.open('sampleA.jpg', 'r')
sampleB = Image.open('sampleB.jpg', 'r')

import time


# 時間計測開始
time1_sta = time.perf_counter()


a = "sampleA"
b = "sampleB" 

print (a)
plt.imshow(np.asarray(sampleA))
print (b)
plt.imshow(np.asarray(sampleB))

while True:
    x = input("好きな方を選んでください＞＞＞＞")

    if x == ("a" or "A"):
        print(a)
        break
    elif x == ("b" or "B"):
        print(b)
        break
    else:
        print("もう一度")

# 時間計測終了
time_end = time.perf_counter()

# 経過時間（秒）
tim = time_end- time_sta


# 結果
print(tim)
    

補足情報（FW/ツールのバージョンなど）

Jupyternotebookでpython使っています。

Answer 1

問題点

matplotlib の使い方

この部分ですが、Figure を分割して Axes を2つ作り、その Axes に対して imshow() を呼び出すようにしないと2つの画像を並べて表示することはできないですよ。

plt.imshow(np.asarray(sampleA))
plt.imshow(np.asarray(sampleB))

また、デフォルトでは、図が表示されるのは最後までセルが実行されたタイミングになるので、途中で表示したい場合、そのタイミングで plt.show() を呼び出す必要があります。

x == ("a" or "A")

x が "a" または "A" か判定したい場合は、x in ["a", "A"] です。
x == ("a" or "A") だと、括弧内が先に評価されて、x == "a" になるので、"A" の場合は if 文に引っかかりません。

サンプルコード

import time

import matplotlib.pyplot as plt
from PIL import Image

img1 = Image.open("sample.png")
img2 = Image.open("sample2.jpg")
fig, [ax1, ax2] = plt.subplots(1, 2)
ax1.set_title("A")
ax1.set_axis_off()
ax1.imshow(img1)
ax2.set_title("B")
ax2.set_axis_off()
ax2.imshow(img2)
plt.show()

start = time.perf_counter()
while True:
    ret = input("好きな方を選んでください＞＞＞＞").lower()

    if ret in ["a", "b"]:
        break
    else:
        print("もう一度")
elapsed = time.perf_counter() - start
print(f"{elapsed:.2f} (secs)")