python - 对二维数组和 nan 进行排序时的 np.argpartition

Question

2d数组排序的时候参考np.argpartition，能排除nan的影响吗？

Such as:
a = np.array([  1.,  -1.,  nan,   0.,  nan,   2.,   4.,  -2., -10.,  nan])

# Get effective Element Center
np.argpartition(a, (~np.isnan(a)).sum()//2)]

但是如果它是一个二维数组查询呢？

标签： pythonnumpypartitioning