Python使用遇到的一些小技巧汇总《一》

1、np.expand_dims(a,axis=0)：用于扩展数组的形状

2、np.argmax()：返回沿轴最大值的索引值

3、os模块

import os 

print(os.getcwd()) #获取当前工作目录路径

os.chdir(path)  #修改当前工作目录路径

os.path.join('file1','file2','file3')  #合并路径，得到file1/file2/file3

4、判断文件是否存在

#不存在则抛出异常，
if not tf.gfile.Exists(filedir):
    raise ValueError('Failed to find file: ' + filedir)

5、创建文件夹

# 创建文件夹cifar10_data/raw/
if not os.path.exists('cifar10_data/raw/'):
    os.makedirs('cifar10_data/raw/')

6、维度旋转

myarray_tra = tf.transpose(myarray, [0, 2, 1])

'''
    三维myarray的shape为[i,j,k]，可以看成i个[j,k]的二维数组，
    参数[0,2,1]，表示将二维和三维进行旋转
    若参数为[1,0,2]，表示将一维和二维进行旋转