聲明4Dimensional np.array由2維np.array

-1

d[i, j, i] = s[i, j] - s[i]

d是4維np.array，s是2維np.array 使用NumPy的和沒有用於循環？

我已經試過如下：

d = np.zeros([10, 10, 10, 10]) 
s = np.ones([10, 10]) 
l1 = range(M) 
l2 = range(N) 
d[np.ix_(l1, l2, l1)] = s[np.ix_(l1, l2)] - s[np.ix_(l1)]

但我想這是行不通的。

d[i, j, k] = [1, 1, 1, 1, 1, 1, 1, 1, 1, 1]

但我想

d[i, j, k] = [0, 0, 0, 0, 0, 0, 0, 0, 0, 0] 

if i != k

來源

2017-05-09 Daniel

你能給與d，S和所需輸出一個更好的例子嗎？（並非全是0或1） – Allen

用那個'''，結果全部爲0。 – hpaulj

這不是很清楚.. –

給予i和JA不同範圍，以便更好地區別尺寸：

迭代溶液：

In [271]: d = np.zeros((3,4,3,4),int) 
In [272]: s = np.arange(12).reshape(3,4) 
In [273]: for i in range(3): 
    ...:  for j in range(4): 
    ...:   d[i,j,i,:] = s[i,j]-s[i,:]

數組溶液：

In [274]: d1 = np.zeros((3,4,3,4),int) 
In [281]: l1=np.arange(3) 
In [283]: d1[l1,:,l1,:] = (s[:,:,None]-s[:,None,:])

檢查

In [284]: np.allclose(d,d1) 
Out[284]: True

和

In [285]: d[0,1,0,:] 
Out[285]: array([ 1, 0, -1, -2]) 
In [286]: d[0,1,2,:] 
Out[286]: array([0, 0, 0, 0]) 
In [287]: d1[0,1,0,:] 
Out[287]: array([ 1, 0, -1, -2]) 
In [288]: d1[0,1,1,:] 
Out[288]: array([0, 0, 0, 0]) 
In [289]: d1[1,2,2,:] 
Out[289]: array([0, 0, 0, 0]) 
In [290]: d1[2,2,2,:] 
Out[290]: array([ 2, 1, 0, -1])

如果s = np.ones((3,4))，d是0所有值。

In [302]: d1[l1,:,l1,:].shape 
Out[302]: (3, 4, 4) 
In [303]: (s[:,:,None]-s[:,None,:]).shape 
Out[303]: (3, 4, 4)

這個索引的變體：

In [318]: I,J=np.ix_(l1,l2) 
In [319]: I 
Out[319]: 
array([[0], 
     [1], 
     [2]]) 
In [320]: J 
Out[320]: array([[0, 1, 2, 3]]) 
In [321]: d1[I,J,I,:].shape 
Out[321]: (3, 4, 4)

你前面的問題的解決方案是

In [323]: d1[I,J,I,J] = s

來源

2017-05-09 05:45:22 hpaulj

聲明4Dimensional np.array由2維np.array

回答

相關問題