4

を使用して、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...caseC/C++ を使用して実装できますが、軸が 100 個ある場合はどうでしょうか。それを実装する方法は?

4

1 に答える 1