python学习手册

1. 交并集

1.获取两个list 的并集：print list(set(a).union(set(b)))
2. 获取两个 list 的差集：print list(set(b).difference(set(a))) # b中有而a中没有的

2. csv处理
   https://blog.csdn.net/lbj1260200629/article/details/89600055
   https://www.cnblogs.com/dengyg200891/p/4890962.html
   自己写的代码：可以借鉴
   def write_csv():
   path_train = r"/data/CTA_HeadNeck/04_train_test/train/"
   path_test = r"/data/CTA_HeadNeck/04_train_test/test/"
   train = r"/data/CTA_HeadNeck/04_train_test/train.csv"
   test = r"/data/CTA_HeadNeck/04_train_test/test.csv"

   csv_train_file = open(train, ‘w’, newline=’’)
   csv_train_writer = csv.writer(csv_train_file)
   csv_test_file = open(test, ‘w’, newline=’’)
   csv_test_writer = csv.writer(csv_test_file)

   for i, dirs in enumerate(os.listdir(path_train)):
   new_dir = os.path.join(path_train, dirs)
   # mask = os.path.join(new_dir, “mask.hdr”)
   a_file = glob.glob("{0}/CTA.nii.gz".format(new_dir))
   mask = glob.glob("{0}/Vessel_BinaryLabel.nii.gz".format(new_dir))
   csv_train_writer.writerow([a_file[0]])
   csv_train_writer.writerow([mask[0]])

   for i, dirs in enumerate(os.listdir(path_test)):
   new_dir = os.path.join(path_test, dirs)
   a_file = glob.glob("{0}/CTA.nii.gz".format(new_dir))
   s = dirs + ’ ’ + a_file[0]
   csv_test_writer.writerow([s + “\n”])