selenium + python 獲取table數(shù)據(jù)的示例講解

<code class="language-python">""" 
根據(jù)table的id屬性和table中的某一個(gè)元素定位其在table中的位置 
table包括表頭，位置坐標(biāo)都是從1開始算 
tableId：table的id屬性 
queryContent：需要確定位置的內(nèi)容 
""" 
def get_table_content(tableId,queryContent): 
  arr = [] 
  arr1 = []  
  table_loc = (By.ID,tableId) 
  # 按行查詢表格的數(shù)據(jù)，取出的數(shù)據(jù)是一整行，按空格分隔每一列的數(shù)據(jù) 
  table_tr_list = driver.find_element(*table_loc).find_elements(By.TAG_NAME, "tr") 
  for tr in table_tr_list: 
    arr1 = (tr.text).split(" ") #以空格拆分成若干個(gè)(個(gè)數(shù)與列的個(gè)數(shù)相同)一維列表 
    # print(tr.text) 
    # print(arr1) 
    arr.append(arr1)  #將表格數(shù)據(jù)組成二維的列表 
 
  #循環(huán)遍歷table數(shù)據(jù)，確定查詢數(shù)據(jù)的位置 
  for i in range(len(arr)): 
    for j in range(len(arr[i])): 
      if queryContent== arr[i][j]: 
        print("%r坐標(biāo)為(%r,%r)" %(queryContent,i+1,j+1)) 
 
get_table_content("myTable","第二行第二列")</code> 

"""
根據(jù)table的id屬性和table中的某一個(gè)元素定位其在table中的位置
table包括表頭，位置坐標(biāo)都是從1開始算
tableId：table的id屬性
queryContent：需要確定位置的內(nèi)容
"""
def get_table_content(tableId,queryContent):
 
  # 按行查詢表格的數(shù)據(jù)，取出的數(shù)據(jù)是一整行，按空格分隔每一列的數(shù)據(jù)
  table_tr_list = driver.find_element(By.ID, tableId).find_elements(By.TAG_NAME, "tr")
  table_list = [] #存放table數(shù)據(jù)
  for tr in table_tr_list:  #遍歷每一個(gè)tr
    #將每一個(gè)tr的數(shù)據(jù)根據(jù)td查詢出來(lái)，返回結(jié)果為list對(duì)象
    table_td_list = tr.find_elements(By.TAG_NAME, "td")
    row_list = []
    print(table_td_list)
    for td in table_td_list:  #遍歷每一個(gè)td
      row_list.append(td.text)  #取出表格的數(shù)據(jù)，并放入行列表里
    table_list.append(row_list)
 
  # 循環(huán)遍歷table數(shù)據(jù)，確定查詢數(shù)據(jù)的位置
  for i in range(len(table_list)):
    for j in range(len(table_list[i])):
      if queryContent== table_list[i][j]:
        print("%r坐標(biāo)為(%r,%r)" %(queryContent,i+1,j+1))
 
get_table_content("myTable","第二行第二列")

"""
根據(jù)CSS選擇器和table中的某一個(gè)元素定位其在table中的位置
table包括表頭，位置坐標(biāo)都是從1開始算
cssSelector：table的CSS選擇器屬性
queryContent：需要確定位置的內(nèi)容
"""
def get_table_content(cssSelector,queryContent):
  locator = cssSelector + ">tbody>tr"
  table_list = []
  table_tr_list = driver.find_elements(By.CSS_SELECTOR, locator)[1:] #去掉表頭
  for tr in table_tr_list:
    row_list = []
    table_td_list = tr.find_elements(By.TAG_NAME,"td")
    for td in table_td_list:
      row_list.append(td.text)
    table_list.append(row_list)
 
  for i in range(len(table_list)):
    for j in range(len(table_list[i])):
      if queryContent== table_list[i][j]:
        print("%r坐標(biāo)為(%r,%r)" %(queryContent,i+1,j+1))
 
get_table_content("#myTable","第二行第二列")