python - 在矩阵中重复每一行 n 次，并在每个部分旁边添加一列 1*n

Question

我想重复矩阵的每一行并在其旁边添加另一列。

想象这是我的矩阵

A = [[11, 12], [13, 14], [15, 16], [17, 18]]

我想每行重复2次，那么结果将是

B = [[1, 11, 12], [2, 11, 12], [1, 13, 14], [2, 13, 14], [1, 15, 16], [2, 15, 16], [1, 17, 18], [2, 17, 18]]

我已经尝试过下面的代码

k = 2
B = [A] * k

这给我的完整代码带来了内存错误

我不知道如何使用 panda，我正在使用 numpy。有没有办法以有效的方式使用 numpy 而不会面临内存错误，就我而言，

并得到正确的答复？

P.S.:我没有添加代码，因为我正在处理巨大的数据集，而且它只是整个代码的一小部分!

Answer 1

你应该这样做

k = 2
B = A * k 

而不是

k = 2
B = [A] * k

要添加新的索引列，您可以这样做

for i, sub_list in B:
    sub_list.insert(i, i+1) # starting from index 1 instead of 0

由于列表是可变的，因此无需重新分配 B。