python selenium xpath定位时使用变量

driver.find_element_by_xpath(input[@id = "kw"])

  上面的代码，相信很多学习selenium + python 的朋友都很熟悉，就是定位百度首页搜索框的代码，如果我们想把"kw"，用一个变量来表示该怎么操作呢？

目前，我知道的有两种方法，如下，是定位百度搜索框，点击搜索的代码，在XPath定位的过程中，用了变量：

# encoding:utf-8
from selenium import webdriver
import time
driver = webdriver.Firefox()
driver.get("http://www.baidu.com")

# 下面这一段使用的是python中变量的方法，简单吧！
def enter(var):
    driver.find_element_by_xpath("//input[@id = '%s']" % var).send_keys("haha")
enter('kw')
time.sleep(3)

#下面这一段使用的好像是一种固定格式，在JS中有看到
def enter1(var1):
    driver.find_element_by_xpath("//input[@id = '" + var1 + "']").click()
enter1('su')
time.sleep(3)
driver.quit()