在python中，重复2d数组不同的次数。

Question

我在python中工作。我有一个具有形状(3, 11, 16)的数组stud和另一个数组times = np.array[740, 560, 600]。我想重复a (0 ,11 ,16)的第一个切片740次。然后，我想将a (1, 11, 16)的下一个切片重复560次，第三个切片也是如此。

这是我的尝试

new_array = []
for i in times:
    for j in range(len(stud)):
        rep = np.repeat(stud[j,:,:], i, axis=0)

这里的问题是所有切片j都重复i次。我想让它工作，这样，对于第一个i，只有第一个j被执行。对于第二个i，只执行第二个j，依此类推。你知道怎么做吗？

Answer 1

我相信你要找的是：

new_array = []
for i, j in zip(times, range(len(stud))):
    rep = np.repeat(stud[j,:,:], i, axis=0)

这将经历3次循环：

第一次，i=740, j=0

第二次，i=560, j=1

第三次，i=600, j=2

如果我误解了你的问题，请纠正我。