您好,欢迎来到欧得旅游网。
搜索
您的当前位置:首页【python代码技巧1】文件查找&数组变维

【python代码技巧1】文件查找&数组变维

来源:欧得旅游网

1. 遍历目录,根据正则匹配

glob

for subject_folder in glob.glob(os.path.join('/home/cv', '*', '*')):
    if os.path.isdir(subject_folder):
        # do something
        pass

2. 增一维,降一维

在处理数组中,总会出现各种匹配shape的情况,比如[3] -> [3,1],[256,256]->[1,1,256,256],除了增加,还有反过来的情况。

a = np.zeros(shape=(3))
a[:, np.newaxis].shape # (3,1)
b = np.zeros(shape=(256,256))
b[np.newaxis, np.newaxis].shape # (1,1,256,256)
aa = np.squeeze(a) # aa.shape = (3)
b2 = np.squeeze(b) # b2.shape = (256, 256)
b3 = np.squeeze(b, axis=0) # b3.shape=(1,256,256), 消灭第一维(only for dim whose size=1)

因篇幅问题不能全部显示,请点此查看更多更全内容

Copyright © 2019- ovod.cn 版权所有 湘ICP备2023023988号-4

违法及侵权请联系:TEL:199 1889 7713 E-MAIL:2724546146@qq.com

本站由北京市万商天勤律师事务所王兴未律师提供法律服务