要求沿维度的最小条目数

Question

我有一个函数fct可以处理3维坐标点：

def fct(p):
  # does something with p

points = np.array((n_points, 3))
fct(points)

现在，也可能发生点是二维的。然而，该函数应该能够使用它们并使用它们，就像它们是三维的，z轴等于零。

目前，我这样做：

def fct(p):
  p3 = np.zeros((p.shape[0], 3))
  p3[:, 0 : p.shape[1]] = p # accepts 2 or 3 dims in p
  # does something with p3

points = np.array((n_points, 2))
fct(points)

所以这需要一份副本。 是否有内置，更有效的方式？