我有一个非常基本的问题,但我在互联网上找不到任何“美丽”的解决方案.我想要的只是从有字符串和括号的行中提取数字.
例如,我有一个输出文本文件,如下所示：@H_404_3@
@H_404_3@
test: [1,50,200]
最佳答案
您可以使用非常幼稚的正则表达式：@H_404_3@
@H_404_3@
import numpy as np
import re
nums = map(int,re.findall(r'\d+','test: [1,200]'))
print(np.array([n for n in nums]))
# [ 1 50 200]
另一种选择是使用正则表达式提取整个列表,然后使用literal_eval：@H_404_3@
@H_404_3@
import numpy as np
import re
from ast import literal_eval
li = re.search(r'(\[.*\])',200]').group()
print(np.array(literal_eval(li)))
# [ 1 50 200]