根据需要添加多个np.newaxis?

2024-10-01 13:44:57 发布

您现在位置:Python中文网/ 问答频道 /正文

我想成对比较(使用<=)两个NumPyndarraysAB的所有元素,其中两个数组都可以有任意维m和n,这样结果就是维m+n的数组

我知道如何在给定的B维度上进行

  1. 标量:A <= B

  2. 一维:A[..., np.newaxis] <= B

  3. 二维:A[..., np.newaxis, np.newaxis] <= B

基本上,我正在寻找一种方法,在第二个数组中插入尽可能多的np.newaxis维数

是否有类似np.newaxis * B.ndim的语法,或者其他方式


Tags: 方法元素np方式语法数组标量维数