Python строка, найти конкретное слово, а затем скопировать слово после

Question

У меня проблема делать такую ​​операцию, скажем, у нас есть строка

teststring = "This is a test of number, number: 525, number: 585, number2: 559" 

Я хочу, чтобы хранить 525 и 585 в список, как я могу сделать это?

Я сделал это очень глупо, работает, но там должен быть более эффективными способами

teststring = teststring.split() 
found = False 
    for word in teststring: 
     if found: 
      templist.append(word) 
      found = False 
     if word is "number:": 
      found = True 

Существуют ли решения с регулярным выражением?

Followup: Что делать, если я хочу хранить 525, 585 и 559?

Answer 1

re Используйте модуль:

>>> re.findall(r'number\d*: (\d+)',teststring) 
['525', '585', '559'] 

\d любая цифра [0-9]
* означает от 0 до бесконечности у раза
() означает, что захват
+ средства от 1 до времен бесконечности

Если вам нужно преобразовать сгенерированные строки в int с, используйте map:

>>> map(int, ['525', '585', '559']) 
[525, 585, 559] 

или

list comprehension:

>>> [int(s) for s in ['525', '585', '559']] 
[525, 585, 559] 

Answer 2

Это не самый эффективный код в мире, но это, вероятно, еще лучше, чем регулярное выражение:

tokens = teststring.split() 
numlist = [val for key, val in zip(tokens, tokens[1:]) if key == 'number:'] 

для последующей деятельности и более общих запросов:

def find_next_tokens(teststring, test): 
    tokens = teststring.split() 
    return [val for key, val in zip(tokens, tokens[1:]) if test(key)] 

, которые могли бы можно назвать:

find_next_tokens(teststring, lambda s: s.startswith('number') and s.endswith(':')) 

Это поможет, если ключи для поиска поступают от пользователя:

find_next_tokens(teststring, lambda s: s in valid_keys) 

Answer 3

Вы можете использовать группы регулярных выражений, чтобы выполнить это. Вот некоторые примеры кода:

import re 
teststring = "This is a test of number, number: 525, number: 585, number2: 559" 
groups = re.findall(r"number2?: (\d{3})", teststring) 

groups затем содержит номера. Этот синтаксис использует группы регулярных выражений.

Answer 4

Вы можете попробовать это:

import re 
[int(x) for x in re.findall(r' \d+', teststring)] 

, который даст вам:

[525, 585, 559] 

Answer 5

Я предлагаю:

teststring = "This is a test of number, number: 525, number: 585, number2: 559" 
# The following does: "This is a test of number, number: 525, number: 585, number2: 559" -> ["525, number", "585, number2", "559"] 
a = teststring.split(': ')[1:] 
# The following does: ["525, number", "585, number2", "559"] -> ["525", " number", "585", " number2", "559"] 
b = [i.split(',') for i in a] 
# The following does: [["525", " number"], ["585", " number2"], ["559"]] -> ["525", "585", "559"] 
c = [i[0] for i in b] 
>>> c 
['525', '585', '559']