python selenium for文

Question

実現したいこと

1-30までのidが連番になっている要素があるとします。
id1,id2,id3,id4......
この各要素をelem1,elem2,elem3...へテキストを代入していき、そのテキストの中から"日本"というテキストの要素を抜き出したいです。

発生している問題・分からないこと

初心者につきいまのところ以下の案しか思い浮かばないのですが、elemは30まであるので大変です。何かうまく纏める文をご教授いただかますでしょうか。

elem1 = driver.find_element(By.ID, "id1").text
elem2 = driver.find_element(By.ID, "id2").text
elem3 = driver.find_element(By.ID, "id3").text
...

elems ＝　[elem1,elem2,elem3....]

for i in elems,
if i ＝＝ '日本'
wantelem ＝i
else
pass

該当のソースコード

特になし

試したこと・調べたこと

• teratailやGoogle等で検索した
• ソースコードを自分なりに変更した
• 知人に聞いた
• その他

上記の詳細・結果

for文で＋1をしていくのかと思いますがその先がわかりません。

補足

特になし

Answer 1

CSS selector を利用して以下の様に書くこともできます。

import re

elems = driver.find_elements(By.CSS_SELECTOR, '[id^="id"]')
elems = [e.text for e in elems if re.match(r'id\d+$', e.get_attribute('id'))]
print(elems)

Answer 2

文字列の操作で変数の値を含めることができます。
いくつか方法はありますが、f文字列を使うと簡単です。

質問の例だとこんな感じでしょうか。

elems ＝　[]
for i in range(10):
    id = f'id{i + 1}'
    new_elem = driver.find_element(By.ID, id).text
    elems.append(new_elem)

少し冗長に書いています。
これで、id1 から id10 までの取得結果が配列elemsに入ります。

---

コメントを受けて追記。 一部コメントから転載

lookupするものが日本だけでなく、米国イタリアなど30カ国程度ある場合

もし、「 i が 日本,米国,...の何れかであれば」という判定にしたいのであれば、 if i in countries のように in 演算子を使うといいでしょう。

例えば、countriesに日本があれば、そのidはnihon_id、米国があればbeikoku_id、、、、、と数十か国ある場合はどのように記述するのがよいのでしょうか。

質問の意味が不明ですね。 countriesには日本も米国も入っているはずなんですが。
たぶんinで判定した後、idが日本だったら、米国だったら、ということなのかと思います。であれば、日本語の文字列を英語の文字列に変換したいということだと思うので、対応させるための辞書を作るのはどうでしょうか。

romaji_id = {
    '日本': 'nihon_id',
    '米国': 'beikocu_id',
    ....
}

これで、変数idに日本や米国が入っているときに、romaji_id[id]で nihon_idやbeikoku_idが取得できます。

"NoSuchElementException"が発生した際にfor文を抜ける(途中までの要素は取得した状態）には、 for文の下にifbreakとなりますでしょうか？それともtryexceptをfor文の最初に入れるべきでしょうか

例外が発生した場合に処理を継続するのであればtry_except一択ですね。exceptのところで、ループを抜ければ、それまでのelemsにデータが入っています。
質問にあるようなif分の構文では例外はつかまえられません。

ちなみに、例外が発生するということは、idがいくつまであるかわからないということですよね。そのような場合は、melianさんの回答のように、idが「id」で始まるべてものを全て取得するという方式のほうがすっきりしますね。

Comments

[kashinomori]

毎度ありがとうございます。後ほどベストアンサーとさせていただきます。後続処理の追加となり恐縮なのですが、取得したelemsに対してlookupするものが日本だけでなく、米国イタリアなど30カ国程度ある場合は以下のコードが冗長なく良いのでしょうか？

countries ＝(日本,米国,....)
for i in elems,
if i ＝＝ countries,
wantelem ＝i
else

[TakaiY]

lookup の意味がわかりません。

少なくとも、 elemsの要素であるiには取得したテキストが入り、countriesはリストですので、「if i ＝＝ countries」のように比較はできません。

もし、「 i が 日本,米国,...の何れかであれば」という判定にしたいのであれば、
if i in countries
のように in 演算子を使うといいでしょう。

[kashinomori]

ありがとうございます。すみません、lookupは忘れてください。
例えば、countriesに日本があれば、そのidはnihon_id、米国があればbeikoku_id、、、、、と数十か国ある場合はどのように記述するのがよいのでしょうか。

[kashinomori]

また一つ疑問が出てまいりましたが、以下の要素を取得している中で
"NoSuchElementException"が発生した際にfor文を抜ける(途中までの要素は取得した状態）には、
for文の下にif~breakとなりますでしょうか？それともtry~exceptをfor文の最初に入れるべきでしょうか？
elems ＝　[]
for i in range(10):
id = f'id{i + 1}'
new_elem = driver.find_element(By.ID, id).text
elems.append(new_elem)
if new_elem = NoSuchElementException:
break

[TakaiY]

長くなるし、コードもあるので、回答に追記しました。

[kashinomori]

大変ありがとうございます。はい、idがいくつまであるかわからないということです。（しかも３行ずつしか画面表示されませんので下にスクロールもしなければいけませんのでそちらのコードも織り込まないといけません）　
idがいくつまであるかわからないということですよね。
まずmedian様の回答を参考にしてみたいと思いますが、基本的なところで
３行目の[id^="id"]
5行目のr'id\d+$'
の意味がよくわかりませんでした。。

[TakaiY]

参考まで
- [id^="id"] : 属性セレクタ(attribute selector)
https://developer.mozilla.org/ja/docs/Web/CSS/Attribute_selectors
- r'id\d+$' : raw文字列 正規表現
検索してみてください。

[kashinomori]

どうもありがとうございます。追加追加で聞いてしまい申し訳ありませんでした。また何かありましたらよろしくお願い申し上げます。