[Python]大きさの異なる行列を、繰り返しを用いて計算したい。

Question

前提・実現したいこと

計算をイメージしやすいように画像を添付いたします。
率直に、numpyのnadarrayであるデータAとBがあったとき、画像のようなデータが欲しいです。
ただ、データAとデータBの行と列が大きくなった時も利用できるようにしたいので、
具体的な数値はこちらで指定せず、（１や２など）
元のデータから取得できる情報のみで結果を得られるようにしたいです。（データAやBの行数、列数など。）
もちろん、数値を使わないに越したことはないのですが・・・

import numpy as np

A = np.array([[1, 5], [2,6], [3, 7],[4, 8]])
B = np.array([[1, 3], [2,4]])
print(A)
print(B)
#[[1 5]
# [2 6]
# [3 7]
# [4 8]]
#[[1 3]
# [2 4]]

#ここから先がわかりません
Result = []
for i in B:
    for j in A:
        Result.append(i*j)
print(np.array(Result))
#[[ 1 15]
# [ 2 18]
# [ 3 21]
# [ 4 24]
# [ 2 20]
# [ 4 24]
# [ 6 28]
# [ 8 32]]

試したこと

自分の書いたものだと8×2になってしまいました。
なんとか、for文など用いて画像の結果のようにならないでしょうか？

データが大きくなった場合の追記です。

Comments

[nto]

データAの量が増えた場合にはデータBの量も必ず比例するのでしょうか？
比例しない場合はありうるのでしょうか？

[ozwk]

>データAとデータBの行と列が大きくなった時も利用できるようにしたい

ということであれば
A:4x2, B:2x2以外のときどういう計算をしたいのかがよくわかりません。
例とは説明の補足のためにあるものです。
どういう計算をしたいのかを例だけに頼らず説明してください

[sabakan4040]

ご質問ありがとうございます。
例えばBが3×3になると、
Aはたて9よこ3になります。
Bを基準にして、横はBとそろったまま、縦は2乗になります。

[meg_]

Aの行数とBの行数は必ず2対1で、Aの列数とBの列数は必ず同じですか？

[sabakan4040]

※完全に私の説明不足です。
お手を煩わせて申し訳ありません

[sabakan4040]

Aの列数とBの列数は必ず同じです。
行数は２乗されます。

Answer 1

こういうことですかね。

import numpy as np

A = np.array([[1, 5], [2,6], [3, 7],[4, 8]])
B = np.array([[1, 3], [2,4]])

assert B.shape[0] == B.shape[1]          # Bは正方行列
assert A.shape[1] == B.shape[1]          # AとBの列数は同じ
assert A.shape[0] == B.shape[0] ** 2     # Aの行数はBの行数の二乗


C = A.copy()
for offset in range(B.shape[0]):
    C[offset::B.shape[0]] *= B
    
print(C)

これでもいけそうです。

B2 = np.hstack([B] * B.shape[0]).reshape(-1, B.shape[0])
C = A * B2

print(C)

Comments

[sabakan4040]

LouiS0616さん
回答ありがとうございます。
自分がやりたいことはBが正方行列でなく、A,Bの列数がそろったままそれぞれ列数が大きくなるのですが、いただいたコードはとても参考になります。
このコードを使って自分のためしたい場合にも援用でき次第、ベストアンサーとさせていただきます。