我想要一个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)
Run Code Online (Sandbox Code Playgroud)
然后迭代所有可能的对,但我确信有一个更好的方式......
我希望回来看起来像:
[[-5, -5],
[-5, -4.5],
[-5, -4],
...
[5, 5]]
Run Code Online (Sandbox Code Playgroud)
但顺序并不重要.
这就是我想要它做的:
import itertools
list(itertools.product(*[[0,1]]*4))
Run Code Online (Sandbox Code Playgroud)
输出:
[(0, 0, 0, 0),
(0, 0, 0, 1),
(0, 0, 1, 0),
(0, 0, 1, 1),
(0, 1, 0, 0),
(0, 1, 0, 1),
(0, 1, 1, 0),
(0, 1, 1, 1),
(1, 0, 0, 0),
(1, 0, 0, 1),
(1, 0, 1, 0),
(1, 0, 1, 1),
(1, 1, 0, 0),
(1, 1, 0, 1),
(1, 1, 1, 0),
(1, 1, 1, 1)]
Run Code Online (Sandbox Code Playgroud)
但它对我有点混淆.有更多的pythonic方式吗?