次元数の異なるnumpy配列のインデックス付け

Question

私はnumpyを使用しており、計算結果を3つのまたは 4次元行列に格納したいとします。

result[:, indices, iteration] = some_function() # won't work 

寸法はindicesタプルの長さから理解されている：私はこのような何かをしたいです。 indicesは、たとえば、(6,)または(8, 5)に等しいのであれば、私は通常、私はタプルに座標のすべてを入れて、としてそれを使用したい

result[:, 6, iteration] = some_function() 

または

result[:, 8, 5, iteration] = some_function() 

と同等のものを持ちたいですインデックスがありますが、:の存在により、私はそれを実行できません。

resultのディメンションは、正しいディメンションの前に設定されているため、確認することはできますが、長さはindicesとしてください。これを達成するための方法はありますか？

Answer 1

お試しください。result[(slice(None),) + indices + (iteration,)]

Answer 2

あなたは、インデックスの長さを知る必要がある場合：

len(indices) 

をしかし、それは、このあなたの質問ですか？