pythonで2次元配列にデータを挿入するやり方

Question

pythonを勉強中のものです。
0～11の数をdataという配列にいれた後、xという二次元配列にdata配列をstartの値（下記のコードで言う"-7"）まで遡って、それからnum個（下記のコードで言う"4"）格納し、それをfor分スライドさせて二次元配列に格納したいというコードになります。
しかしstart=-7,num=4としたとき、最後の要素（data[-4:0]）のとき、エラーになります。
実現したいことは二次元配列xがこのようになることです。
[ [ 5. 6. 7. 8.]
[ 6. 7. 8. 9.]
[ 7. 8. 9. 10.]
[ 8. 9. 10. 11.]]←これが現したい

どうすればよろしいでしょうか。

何卒ご教授よろしくお願いいたします。

import numpy as np

num = 4
start = -7

data = []
for i in range(12):
    data.append(i)

x = np.zeros((num, num))

for i in range(num):
    j = start + i
    k = j + num
    x[i] = data[j:k]

print(x)

【自分の環境】
windows10
visual studio 2019 community
python 3

Answer 1

for ループの最後、i が3のとき、 jが -4 で k が 0 となるので、data[j:k] は、data[-4:0]となりますが、これは意図している、[8, 9, 10, 11] にはならず、空のリスト [] になるため、

x[i] = data[j:k]

の代入ができず、画像のとおり

ValueError: cannot copy sequence with size 0 to array axis with dimension 4

というエラーが発生します。ここを修正するには、k が0のとき、x[i] に代入するのは data[j:k]ではなく、kをスライスの終端に使用しないdata[j:]にすればよいので、下記のように修正してみるといかがでしょうか。

- x[i] = data[j:k]
+ x[i] = data[j:k] if k < 0 else data[j:]

追記

以下は dataを使わないで、目的の x を作るコードです。

import numpy as np

num = 4
start = 5

x = np.zeros((num, num))

for i in range(num):
    j = start + i
    k = j + num
    x[i] = range(j, k)


print(x)

Comments

[Yusuke_nomecchi]

解決しました！
ありがとうございます！

[jun68ykt]

どういたしまして。
> 解決しました！
とのことでよかったです????