def indices(m, pole, dlim=75.0): print ("Determining polar cap pixels.") if pole == 'north': p = np.where((m.lath > dlim) & (m.rg < m.rsun*np.sin(75.0*np.pi/180))) vp = np.where((m.rg < m.rsun) & (m.lath > dlim) & M.isfinite(m.im_corr)) posp = np.where((m.rg < m.rsun) & (m.lath > dlim) & (m.im_corr > 0.0)) negp = np.where((m.rg < m.rsun) & (m.lath > dlim) & (m.im_corr < 0.0)) else: p = np.where((m.lath < -dlim) & (m.rg < m.rsun*np.sin(75.0*np.pi/180))) vp = np.where((m.rg < m.rsun) & (m.lath < -dlim) & M.isfinite(m.im_corr)) posp = np.where((m.rg < m.rsun) & (m.lath < -dlim) & (m.im_corr > 0.0)) negp = np.where((m.rg < m.rsun) & (m.lath < -dlim) & (m.im_corr < 0.0)) return p, vp, posp, negp