配列（リスト）の埋め込む(置き換え) 長さが異なる

Question

配列（リスト）の埋め込み

現状のコード

l = [[0 for i in range(5)] for j in range(5)]
r=[[1],[1,1],[1,1,1]]

# x=座標[横] y=座標[縦] z=埋め込む配列
def Sample_Demo_New_Bloku1(x,y,z):
	if l[x][y]==0:l[x][y]=z

r配列の１番目なら動作しますが、２番め以降はリストの長さが異なるので、
上手く行きません。

どうしたら、良いででしょうか？

#元の配列

[[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]]

加工後　パターンA

x=0,y=0,z=r[1]の場合

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

加工後　パターンB

x=1,y=1,z=r[2]の場合

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

#　追記
昨日の夜は大分疲れてた見たいです。
朝起きたら普通にできました。
np系も欲しかったのでベストアンサーに選んで置きます。

from pprint import pprint
#pprint(l)

l = [[0 for i in range(5)] for j in range(5)]
r=[[1],[1,1],[1,1,1]]

# x=座標x y=座標y z=埋め込む配列

def Sample_Demo_New_Bloku2(x,y,z):
	for i in range(len(z)):
		l[x][i+y]=r[r.index(z)][i]

Sample_Demo_New_Bloku2(0,0,r[1])


pprint(l)

Comments

[mather]

Sample_Demo_New_Bloku1 関数を実行する部分の記載が抜けてます。具体的にどう使うつもりなのですか？

Answer 1

import numpy as np

l = np.array(l)
r = [np.array(v) for v in r]

~

if l[x][y]==0:
  l[x][y:y+z.shape[0]] = z

とかでしょうか？
numpyにしないといけんかった…

---

何がやりたいのか定かじゃないけど、別に起きうるエラーの対策を先にやっておく。

#必numpy
y_ = min(z.shape[0], l.shape[1])
if np.all(l[x][y:y+y_]==0):
  l[x][y:y+y_] = z[:y_]

---

リストでがんばるなら。

y_ = min(len(z), len(l[y:]))
if all([v==0 for v in l[y:y_]]):
  for i in range(y_):
    l[x][y+i] = z[i]