初始化具有可变大小的2d数组

Question

使用以下方法:

myArray = [0,1] * NUM_ITEMS

期望的结果(2d数组):

[[0,1],[0,1],[0,1]...]

实际结果(扩展1d数组):

[0,1,0,1,0,1...]

如何在不使用numpy的情况下实现理想的结果？

Answer 1

列表理解应该可以解决问题:

>>> NUM_ITEMS = 5
>>> my_array = [[0, 1] for _ in range(NUM_ITEMS)]
>>> my_array
[[0, 1], [0, 1], [0, 1], [0, 1], [0, 1]]