python 猜数字 1-100

从1到100的自然数。如果任意设定数字，用户必须反复输入数字，直到匹配。如果你猜对了数字，我想告诉你你猜对了几次。
条件:数字是任意设定的。程序会反复运行，直到用户猜对数字。当用户猜对数字时，告诉用户只猜了几次。

import random
b = random.randint(1,100)
count = 0
while True:
  count += 1
  a = int(input())
  if a == b:
    break
print(count)