非矩形热源Python

Question

我必须使热源形状像X，但它必须被定义为函数。在矩阵中，如下所示：

Gr=np.eye(10)*5000
for iGr in range(10):
    Gr[iGr,-iGr-1]=5000

def assert_heaters(M, Gr):
    M[20:30,10:20] = np.where(Gr > 0, Gr, M[20:30,10:20])
M=np.zeros([L,B])           
assert_heaters(M, Gr) 

我试着发挥作用，但没有成功。

def q(x,y,qqq):
    qq=0
    for k in range 10:
        if x>20 and x<30 and y>10 and y<20:

        qq=qqq
    return qq

我的问题是:我如何使它看起来像X而不是长方形？

Answer 1

一些简单的东西，如(未经测试的)：

def heat(x, y):
    x1 = x - 20
    y1 = y - 10
    if x1 < 0 or x1 > 9 or y1 < 0 or y1 > 9:    # out of range
        return 0
    if (x1 == y1) or (x1 == 9-y1):
        return 5000
    else:
        return 0