numpy.broadcast_to对面

Question

是否与numpy.broadcast_to相反，例如numpy.project_to(array, function, axis)？

np.project_to(
  np.array([[1,1,1], [2,1,3], [2,2,2]], dtype=float),
  lambda x, y: x/y,
  axis=1)

会返回

array([ 1., 0.66666667, 0.5])

Answer 1

看起来你要求的是沿特定轴的缩减：这可以使用任何二进制ufunc的reduce方法来完成。例如：

>>> x = np.array([[1,1,1], [2,1,3], [2,2,2]], dtype=float)

>>> np.divide.reduce(x, axis=1)
array([ 1.        ,  0.66666667,  0.5       ])