if_else_comprehension.py

pedobjects=[] #list of pedobjects, represents lines in a pedfile
pedfh=open(options.pedfile, 'r')
for line in pedfh:
  fields=line.strip().split('\t')
  (fid,iid,pid,mid,sex,phenotype)=fields[0:6]
  phenotype=int(phenotype)
  pedobjects.append( Ped(fid,iid,pid,mid,sex,phenotype) )

#the phenotype status is set to 2 if the sample is affected: http://pngu.mgh.harvard.edu/~purcell/plink/data.shtml#ped
affecteds=[ pedobj.getid() for pedobj in pedobjects if pedobj.getpheno() == 2  ]
unaffecteds=[ pedobj.getid() for pedobj in pedobjects if pedobj.getpheno() == 1  ]