python - 在 Python 中使用循环创建数组的交替列表

Question

我正在尝试使用循环生成交替的数组列表，但我无法弄清楚语法。目前我正在使用以下代码(作为示例):

[ numpy.array([i,4,5]),numpy.array([31,4,i]) for i in range(5)) ]

它给出了以下错误:

"SyntaxError: invalid syntax"

我已经尝试过 、 + 和 concatenate 但它似乎不起作用。

所需的输出是以下带有交替数组条目的列表:

[array([0, 4, 5]),
 array([31,  4,  0]),
 array([1, 4, 5]),
 array([31,  4,  1]),
 array([2, 4, 5]),
 array([31,  4,  2]),
 array([3, 4, 5]),
 array([31,  4,  3]),
 array([4, 4, 5]),
 array([31,  4,  4])]

谢谢大家!

Answer 1

你想生产什么？

In [3]: [ (numpy.array([i,4,5]),numpy.array([31,4,i])) for i in range(5) ]
Out[3]:
[(array([0, 4, 5]), array([31,  4,  0])),
 (array([1, 4, 5]), array([31,  4,  1])),
 (array([2, 4, 5]), array([31,  4,  2])),
 (array([3, 4, 5]), array([31,  4,  3])),
 (array([4, 4, 5]), array([31,  4,  4]))]

使用for循环，同样的事情:

myList = []
for i in range(5):
   item = ( numpy.array([i,4,5]),numpy.array([31,4,i]) )
   myList.append(item)
print(myList)