リストに列を追加したい

Question

前提・実現したいこと

99x40のリスト:predataがあります．
これに指定の行に1x40の配列を挿入して最終的には105x40の配列:resDにしたいと考えています．
配列を挿入したい列が，最終的な配列の1,7,96,100,104,105行:zeroRowです．
どのようにしたらよいかご教示いただけないでしょうか．
よろしくお願いいたします．

発生している問題・エラーメッセージ

xの繰り返し回数を99でforループをまわした場合
resD繰り返し回数が105回まで達せず，resDのサイズが102x40となってしまう

xの繰り返し回数を105でforループをまわした場合
predataのサイズを繰り返し回数が超えてしまい下記のエラーが発生する
IndexError: list index out of range

該当のソースコード

predata = np.arange(0, 3960).reshape(99, 40).tolist()
resD = []
zeroRow = [0, 6, 95, 99, 103, 104]
zeroarr = np.linspace(0, 0, 40).tolist()
for x in range(np.array(predata).shape[0]):
    rr = np.array(resD).shape[0]
    print(rr)
    if x in zeroRow:
        resD.insert(rr, zeroarr)
        resD.insert(rr+1, predata[x])
        
    else:
        resD.insert(rr, predata[x])

試したこと

繰り返し回数を変えてやってみましたが，上記のようなエラーが発生してしまっております．

補足情報（FW/ツールのバージョンなど）

開発環境　spyder 4
Pythonバージョン　3.8
OS　Win 10

Comments

[meg_]

リストとnumpy配列と両方使用していますが、あえてですか？

[Kodanosuke]

当方は初心者なものでとにかく動く様にと書いてしまっております。
もし可能でしたらその辺りも含めてご指導いただけると幸いです。
よろしくお願いします。

[meg_]

> xの繰り返し回数を99でforループをまわした場合
> resD繰り返し回数が105回まで達せず，resDのサイズが102x40となってしまう
全く検証していないのですが、99で102になるなら xの繰り返し回数をを102にするとどうなるのでしょうか？

[Kodanosuke]

回答していただきありがとうございました．
無事に解決することができました．

Answer 1

以下のように挿入前の配列を基準とした挿入位置を算出することで、numpy配列のまま処理できます。
参考：Inserting rows and columns into a numpy array

import numpy as np

predata = np.arange(20).reshape(5,4)+1
zeroRow = [0, 3, 7]
idx = np.array(zeroRow) - np.arange(len(zeroRow)) # 挿入前での位置
print(idx) # [0 2 5]
predata = np.insert( predata, idx, 0, axis=0) # すべて０の行を追加するなら第三引数は０でよい
print(predata)
"""
[[ 0  0  0  0]
 [ 1  2  3  4]
 [ 5  6  7  8]
 [ 0  0  0  0]
 [ 9 10 11 12]
 [13 14 15 16]
 [17 18 19 20]
 [ 0  0  0  0]]
"""

Comments

[Kodanosuke]

回答いただきありがとうございます．
今回はリストでの操作を想定していましたので，上記の回答をベストアンサーにさせていただきました．
ただ，このような方法もあると知ることができ非常に助かりました．

Answer 2

リストを使ってやる場合、挿入すると考えると難しいので、入れ物を作っておいて入れていく方がわかりやすいです。

predata = np.arange(0, 3960).reshape(99, 40).tolist()
zeroRow = [0, 6, 95, 99, 103, 104]
resD  = [None]*(len(predata) + len(zeroRow))
zeroarr = np.linspace(0, 0, 40).tolist()

offset = 0
for i in range(len(resD)):
    if i in zeroRow:
        resD[i] = zeroarr.copy()
        offset += 1
    else:
        resD[i] = predata[i - offset]

print(resD[5])
print(resD[6])
print(resD[7])

resD = [None]*(len(predata) + len(zeroRow)) で長さ105のリストを作っています。
あとは、割と単純に入れていくだけです。

実行結果

>>> print(resD[5])
[160, 161, 162, 163, 164, 165, 166, 167, 168, 169, 170, 171, 172, 173, 174, 175, 176, 177, 178, 179, 180, 181, 182, 183, 184, 185, 186, 187, 188, 189, 190, 191, 192, 193, 194, 195, 196, 197, 198, 199]
>>> print(resD[6])
[0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0]
>>> print(resD[7])
[200, 201, 202, 203, 204, 205, 206, 207, 208, 209, 210, 211, 212, 213, 214, 215, 216, 217, 218, 219, 220, 221, 222, 223, 224, 225, 226, 227, 228, 229, 230, 231, 232, 233, 234, 235, 236, 237, 238, 239]

Comments

[Kodanosuke]

回答いただきありがとうございます．
今回はリストでの操作を想定していたので，こちらの回答をベストアンサーにさせていただきます．