Python使用正则表达式提取文本指定内容,保存为npy

import re
import numpy as np

# 读入文件
with open('filename.txt') as f:
    file = f.read()
    print(file)
    # 使用正则表达式匹配相应的内容, 使用findall匹配,返回列表类型,但是属于字符数组
    file = re.findall('G/g_loss: (\d\S\d+)', file, re.S)

    # print(file)


# 将字符数组转为数字型数组
new_file = []
for n in file:
    new_file.append(float(n))

print(new_file)

# 将已有数组转换为数组,并保存
a = np.asarray(new_file)
np.save('filename.npy', a)

p = np.load('filename.npy')
print(p)

你可能感兴趣的:(numpy)