是否有一个多维版本的arange/linspace？
是否有一个多维版本的arange/linspace？
提问于 2018-03-26 06:30:23
我想要一个2d numpy数组(x，y)的列表，其中每个x都在{-5，-4.5，-4，-3.5，…，3.5，4，4.5，5}中，而y是相同的。
我可以
x = np.arange(-5, 5.1, 0.5)
y = np.arange(-5, 5.1, 0.5)然后遍历所有可能的配对，但我相信还有更好的方法。
我想要一些看起来像：
[[-5, -5],
[-5, -4.5],
[-5, -4],
...
[5, 5]]顺序并不重要。
Stack Overflow用户
发布于 2018-03-26 15:34:23
可以使用np.meshgrid:
import numpy as np
x = np.arange(-5, 5.1, 0.5)
y = np.arange(-5, 5.1, 0.5)
X,Y = np.meshgrid(x,y)您可以将其转换为所需的输出。
XY=np.array([X.flatten(),Y.flatten()]).T
print XY
array([[-5. , -5. ],
[-4.5, -5. ],
[-4. , -5. ],
[-3.5, -5. ],
[-3. , -5. ],
[-2.5, -5. ],
....
[ 3. , 5. ],
[ 3.5, 5. ],
[ 4. , 5. ],
[ 4.5, 5. ],
[ 5. , 5. ]])页面原文内容由Stack Overflow提供。腾讯云小微IT领域专用引擎提供翻译支持
原文链接：
https://stackoverflow.com/questions/-100004307
复制相关文章
相似问题