高效陣列創建在numpy的

Question

我試圖創建一個三維陣列，Tusing numpy的定義如下：

T_ {I，J，K} = \ delta_ {I，K} - \增量{Ĵ， k}

其中\ delta_ {i，j}是Kronecker delta函數（當i = j時爲1，否則爲0）。我想知道用numpy做這個最有效的方法。我可以使用for循環創建兩個三維數組並將其減去。但我懷疑有一種更快更自然的方法。非常感激任何的幫助。

Answer 1

相當於delta是在numpy的eye：

delta = numpy.eye(5) 
T = delta[:,None,:] - delta[None,:,:] 

的None創建<虛擬>尺寸（不帶任何附加存儲器）用於在numpy的廣播。