声明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

回答

相关问题