python 按条件修改数组的值

yolov5输出的txt文件，修改classes序号。

import os
import numpy as np

paths = '/data/runs/detect/exp/labels'
filename = os.listdir(paths)
savepath = '/data/my_set/labels'

if not os.path.exists(savepath):
    os.makedirs(savepath)
    print('创建文件夹 ' + savepath)

for i in filename:
    filepath = os.path.join(paths,i)
    with open(filepath) as lines:
        print('打开文件 ' + filepath)
        a = np.loadtxt(filepath)
        a[a==2] = 1
        a[a==7] = 2
        print(a)
    with open(os.path.join(savepath,i),'w') as f:
        np.savetxt(f,a,fmt='%g')
        print('写入成功 ' + os.path.join(savepath,i))