z = np.mgrid[a:X:m, b:y:n]
z = np.mgrid[a:X:cj, b:y:dj]
利用到广播机制进行扩充
# z = np.mgrid[a:x:m, b:y:n] 两个参数,生成一个三维空间向量由两个数组z[0]和z[1]组成,步长分别为m和n, [a, x) [b, y)
# 第二种:z = np.mgrid[a:x:cj, b:y:dj] 生成平均分为c/d个元素,左闭右闭,[a, x] [b, y]
# 例如第一种:
# 两个数组的行数由函数第一个参数决定(2 3 4),共三行;列数由第二个参数决定(1 2)共两列;默认步长为1,左闭右开;分成的元素由mn或cd决定
# 第一个数组z[0]的元素由第一个参数决定(2 3 4),共三种元素;第二个数组z[1]的元素由第二个参数决定(1 2),共两种元素
# 第一个数组z[0]的元素行相同;第二个数组z[1]的元素列相同
z = np.mgrid[2:5, 1:3] # 默认步长为1,左闭右开
x, y = z[0], z[1]
print('x=', x)
print('y=', y)
# 结果:
# x= [[2 2]
# # [3 3]
# # [4 4]]
# # y= [[1 2]
# # [1 2]
# # [1 2]]
z = np.mgrid[2:5, 3:4]
x, y = z[0], z[1]
print('x=', x)
print('y=', y)
# 结果:
# x= [[2]
# [3]
# [4]]
# y= [[3]
# [3]
# [3]]