を使用して、sum
特定の軸に沿ってすべての要素を取得できます。numpy.sum
>>> a = numpy.array([[1,2], [3,4]])
>>> numpy.sum(a, 1)
array([3, 7])
つまりsum along row
、各列の要素を 1 つずつ追加します。
軸が 2 つまたは 3 つしかない場合は、if...elif
またはswith...case
C/C++ を使用して実装できますが、軸が 100 個ある場合はどうでしょうか。それを実装する方法は?