Question

计算三维网格上的点的坐标，并存储在以下数组中获得的值：

import numpy as np

x = np.zeros(4*4*8, float)
y = np.zeros(4*4*8, float)
z = np.zeros(4*4*8, float)

for l in range(4*4*8):
    x[l], y[l], z[l] = np.unravel_index(l, (8, 4, 4))

我将矩阵定义为

def getMatrix(kappa, x, y, z):


   nxnynz = 4*4*8
   nrange_x = np.arange(nxnynz)
   nrange_y = nrange_x
   w, r = np.meshgrid(nrange_x, nrange_y)
   delta_r = np.sqrt(((x[w]-x[r])**2)+((y[w]-y[r])**2)+((z[w]-z[r])**2)*)

   matrix = np.zeros(delta_r.shape)

   matrix[delta_r == 0] = 4.*kappa/(nxnynz**3.)
   matrix[delta_r != 0] = np.sin(kappa*2.*np.pi*delta_r[delta_r != 0])/(delta_r[delta_r != 0]*float(nxnynz))

其中关键点是delta_r == 0和delta_r != 0时这些值的不同定义。现在，除了这个规范之外，我还需要添加一个，即条件（在编写错误的代码中，只是为了给出这个想法）

if ((abs(w-r)) % 8 ) != 0:
    matrix[w][r] = 0.

其中w和r与之前使用的索引相同。如何在以前的矩阵定义中拟合这个新条件？

Answer 1

这应该是布尔索引的直接使用。例如：

In [401]: w,r=np.meshgrid(np.arange(3),np.arange(2))
In [402]: w
Out[402]: 
array([[0, 1, 2],
       [0, 1, 2]])
In [403]: r
Out[403]: 
array([[0, 0, 0],
       [1, 1, 1]])
In [404]: mat = np.ones(w.shape,int)
In [405]: mat
Out[405]: 
array([[1, 1, 1],
       [1, 1, 1]])

In [406]: abs(w-r)%8
Out[406]: 
array([[0, 1, 2],
       [1, 0, 1]], dtype=int32)
In [407]: (abs(w-r)%8)!=0
Out[407]: 
array([[False,  True,  True],
       [ True, False,  True]], dtype=bool)

In [408]: mat[(abs(w-r)%8)!=0]=0
In [409]: mat
Out[409]: 
array([[1, 0, 0],
       [0, 1, 0]])

如果没有充分研究您的代码，我认为您的matrix（错误名称）与w和r具有相同的形状。我可以构造一个满足条件的布尔数组，并轻松应用matrix。

在矩阵上添加新条件 - Python

1 个答案: