Pythonでリストの中に特定の文字列があった場合その次の文字列をprintしたい

Question

やりたいこと

文字だと分かりずらいので以下に示します。
仮に特定の文字列が'183'だとすると

['34', '1800', '183', '5348']

から

5384

をプリントしたい。

今できていること

特定の文字列（以下ルームと呼ぶ）の場所が特定の位置でないと上手く作動しないプログラムしかかけていません。

import re
t='''
3月4日
18:00
デュオ
????183
????5348
'''
room='183'
r=re.sub(r'[^\d\n]', '', t)
p=r.split()
print(p)
if room in p:
	key=p[3]
else:
	key=p[0]
print(key)

ルームが[2]にないと上手くキー（ルームの次の文字列）を取り出せません。ルームの位置が特定出来ていなくてもルームの隣のキーをとりだせるようにするにはどうすればよいでしょうか（キーはリストの位置で言うとルームの右隣に来ることが前提）。どなたか回答お願いします。

Answer 1

p.py

def get_next_item(data, name):
    return [data[p + 1]  for p in range(0, len(data) - 1) if data[p] == name]

data = ['34', '1800', '183', '5348']
print(get_next_item(data, '183'))

data = ['34', '1800', '183']
print(data, " -> ", get_next_item(data, '183'))

data = ['34', '1800']
print(data, " -> ",  get_next_item(data, '183'))

data = ['34', '1800', '183', '5348', '183', '999']
print(data,  " -> ", get_next_item(data, '183'))

実行例

"183" が含まれなかった場合、"183" が複数ふくまれていた場合, list の最後が "183" だった場合の対処も考慮してみています。

Comments

[kosei.]

ほんとにありがとうございます。無事機能追加することができました。

Answer 2

indexで、指定要素の位置が分かります。

    key=p[p.index(room)+1]

上記だと、最終要素だった場合に+1するとリストの範囲外になるので、必要ならそこの考慮も必要。

Comments

[kosei.]

できました！何度も本当にありがとうございます。