numpyスライスを一般的に書くことができますか？

Question

私は

xとRHSは、サイズの3D numpyのアレイ（T、L、S）です

x[i, :, :] = (rhs[i, :, :]-diag[i] * x[i+1, :, :])/diag[i] 

のような何かをしたいです。 diagはサイズTの1D配列です。

これは適切に放送されます。

しかし、2D配列やその他の次元数でも同様の機能を使いたいと思います。どのようにしてこれを一般的に記述すれば、最初の次元がTの配列でも動作するようになります。関数内にこれらの種類の行がたくさんあるので、コードの重複を避けたい。

Answer 1

x[i] = (rhs[i] - diag[i] * x[i+1])/diag[i] 

これらのコロンは完全に不要です。