@liguangsheng 的想法是對的, 可惜代碼有一點點問題
def getNumber(str):
for line in result.split('\n'):
group = line.split(':')
if str in group:
return group[0]
return False # <------------------------ 問題出在這裡
這會導致第一行沒有找到 str 就失敗, 比如說可以試一下:
print(getNumber('bbbbbb'))
這應該找的到的, 但會回傳 False
我小改了一下:
def find(data, keyword):
pairs = (line.strip().split(':') for line in result.split('\n'))
for i, words in pairs:
if keyword in words.split():
return i
return None # <--------- return should be here
if __name__=='__main__':
result = """32:aaaaaa
33:bbbbbb
34:ccccccc
35:ddddddd"""
print(find(result, 'aaaaaa'))
print(find(result, 'ccccccc'))
print(find(result, 'abcdefg'))
結果:
32
34
None