Skip to content

Instantly share code, notes, and snippets.

@nrweir
Created March 1, 2018 15:32
Show Gist options
  • Star 0 You must be signed in to star a gist
  • Fork 0 You must be signed in to fork a gist
  • Save nrweir/22e5d8dca88ed27440af8ae0788a88cb to your computer and use it in GitHub Desktop.
Save nrweir/22e5d8dca88ed27440af8ae0788a88cb to your computer and use it in GitHub Desktop.
os.chdir(img_dir)
flist = os.listdir()
imgs = [f for f in flist if '.tif' in f.lower()]
pex_imgs = [im for im in imgs if '594' in im]
mito_imgs = [im for im in imgs if '447' in im]
pex_imgs.sort()
mito_imgs.sort()
if len(pex_imgs) != len(mito_imgs):
raise ValueError('Length of pex and mito img sets do not match.')
ims_per_job = int(len(pex_imgs)/array_l)
split_pex_list = []
split_mito_list = []
for i in range(0, len(pex_imgs), ims_per_job):
split_pex_list.append(pex_imgs[i:i+ims_per_job])
split_mito_list.append(mito_imgs[i:i+ims_per_job])
n_leftover = len(pex_imgs)%array_l
if n_leftover != 0:
leftover_pex = pex_imgs[-n_leftover:]
leftover_mito = mito_imgs[-n_leftover:]
for x in range(0,len(leftover_pex)):
split_pex_list[x].append(leftover_pex[x])
split_mito_list[x].append(leftover_mito[x])
mito_list = split_mito_list[array_n]
pex_list = split_pex_list[array_n]
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment