当我在“搜索”按钮中输入数据时，如何使用selenium编写逻辑来检查数据是否显示

from selenium import webdriver
from selenium.webdriver.common.by import By
from selenium.webdriver.support.ui import WebDriverWait
from selenium.webdriver.support import expected_conditions as EC

driver = webdriver.Chrome()  # 这里使用Chrome浏览器，你也可以选择其他浏览器

driver.get("https://example.com")  # 替换成你想要打开的网页链接
search_box = driver.find_element_by_id("search-box")  # 替换成搜索框的HTML元素ID或其他定位方法
search_box.send_keys("要搜索的数据")  # 输入要搜索的数据

search_button = driver.find_element_by_id("search-button")  # 替换成搜索按钮的HTML元素ID或其他定位方法
search_button.click()  # 点击搜索按钮

wait = WebDriverWait(driver, 10)  # 最多等待10秒钟，你可以根据实际情况调整
data_element = wait.until(EC.visibility_of_element_located((By.ID, "data-element")))  # 替换成显示数据的HTML元素ID或其他定位方法

if data_element.is_displayed():
    print("数据已显示")
else:
    print("数据未显示")

driver.quit()