import numpy as np
def _mask(r_width, r_hight):
mask = np.zeros((r_hight, r_width), dtype=int)
x_mid = int(r_width / 2)
y_mid = int(r_hight / 2)
x_step = (r_width-1) / (r_hight-1)
for y in range(y_mid):
mask[y][x_mid - int(x_step * y) - 1 : x_mid + int(x_step * y) + 1] = 1
for y in range(y_mid, r_hight):
mask[y][x_mid - int(x_step * (r_hight - y)) : x_mid + int(x_step * (r_hight - y)) + 1] = 1
return mask
print(_mask(6,7), _mask(7,7), _mask(8,8), sep="\n\n")
aW1wb3J0IG51bXB5IGFzIG5wCgpkZWYgX21hc2socl93aWR0aCwgcl9oaWdodCk6CiAgICBtYXNrID0gbnAuemVyb3MoKHJfaGlnaHQsIHJfd2lkdGgpLCBkdHlwZT1pbnQpCiAgICB4X21pZCA9IGludChyX3dpZHRoIC8gMikKICAgIHlfbWlkID0gaW50KHJfaGlnaHQgLyAyKQogICAgeF9zdGVwID0gKHJfd2lkdGgtMSkgLyAocl9oaWdodC0xKQoKICAgIGZvciB5IGluIHJhbmdlKHlfbWlkKToKICAgICAgICBtYXNrW3ldW3hfbWlkIC0gaW50KHhfc3RlcCAqIHkpIC0gMSA6IHhfbWlkICsgaW50KHhfc3RlcCAqIHkpICsgMV0gPSAxCiAgICBmb3IgeSBpbiByYW5nZSh5X21pZCwgcl9oaWdodCk6CiAgICAgICAgIG1hc2tbeV1beF9taWQgLSBpbnQoeF9zdGVwICogKHJfaGlnaHQgLSB5KSkgOiB4X21pZCArIGludCh4X3N0ZXAgKiAocl9oaWdodCAtIHkpKSArIDFdID0gMQogICAgCiAgICByZXR1cm4gbWFzawoKcHJpbnQoX21hc2soNiw3KSwgX21hc2soNyw3KSwgX21hc2soOCw4KSwgc2VwPSJcblxuIikK