python常用操作

1.获取文件夹内每个文件名

函数：os.listdir()

import os
rootdir = '/home/xx/image'
lists = os.listdir(rootdir）

结果：

以此方式获取的文件名是无序的，那么需要调用sort()将其进行排序

lists.sort()	# 此时输入的lists就是有序的了

2.截去字符串中指定字符

函数：strip()

s1 = '123456.jpg'
s1_ = s1.rstrip('.jpg')# 截去右边的.jpg
print(s1_)

s2 = '_111'
s2_ = s2.lstrip('_')
print(s2_)

s3 = '_11_'
s3_ = s3.strip('_')
print(s3_)

结果：

3.创建一个新的文件夹

函数：os.mkdir()

import os
# 判断桌面上是否有a这个文件夹，如果没有则创建一个
if 'a' not in os.listdir('~/桌面/')
	os.mkdir('~/桌面/' + 'a') # 创建一个名字为a的文件夹，一定要记得添加a文件夹所要创建的路径哦，不然会被创建到当前代码所在的目录下。

4.numpy求每一列/行的最大/小值

import numpy as np
# 生成4*3的随机数组
a = np.random.randint(0,20,size=[4,3])
# 求每一列的最大值
'''得到的max0[0]是第一列的最大值，max0[1]是第二列的最大值，以此类推，下面几个同理'''
max0 = np.max(a,axis=0)
# 求每一行的最大值
max1 = np.max(a,axis=1)
# 求每一列的最小值
min0 = np.min(a,axis=0)
# 求每一行的最小值
min1 = np.min(a,axis=1)

5.按行/列连接

# 按列连接：第一个维度必须保持一致
b = np.random.randint(0,10,size=[2,3])
c = np.random.randint(11,20,size=[2,5])
bc = np.c_[b,c]

# 按列连接：第二个维度必须保持一致
b = np.random.randint(0,10,size=[3,2])
c = np.random.randint(11,20,size=[5,2])
bc = np.r_[b,c]
print(b)
print(c)
print(bc)

6.加入一个新的数/数组到list中

a = []
a.append(1)

7.查看变量的类型

print(type(a))

8.list和numpy的相互转换

# list转numpy
b = np.array(a)
# numpy转list
c = b.tolist()

9.list求最大值

a = [1,2,3,14]	# 创建一个list类型的变量
print(max(a))	# 输出为：14

10.求list中某个值的索引

a = [1,2,3,14]	# 创建一个list类型的变量
index_ = a.index(2)	# 输出为：1