python 零散知识汇总

求解两个列表中的不同元素

c = [x for x in set1 if x not in set2]  #在list1列表中而不在list2列表中
d = [y for y in set2 if y not in set1] 

得到当前文件夹中所有的文件名

os.listdir(r"E:\save_img\save_img")

根据某种筛选条件进行拆分

set([i.split(".")[0] for i in img_list])#根据.来切分当前数据 

将文件夹中的图片划分为训练集与测试集

list_dir=os.listdir(r"E:\save_img\save_img")
num=len(list_dir)
train_num=0.9*num
train_img=list_dir[:train_num]
test_img=list_dir[train_num:]
#之后对这两个数据进行随机翻转 即可得到训练集与测试集

读取当前文件夹下所有的文件

img_list=os.listdir(r"E:\save_img\save_img")

打开一个文件，不存在则创建

open(out_txt_path, ‘w’)

读取文件中所有的行

参考博客

https://blog.csdn.net/qq_37383691/article/details/78078736
这里是引用