numpy用数组填充数组

Question

我想在Kronecker产品下合并未指定（有限）的矩阵。 为了做到这一点，我想将矩阵保存在数组中，但是我不知道该怎么做。 目前，我有：

for i in range(LNew-2):
    for j in range(LNew-2):
        Bulk = np.empty(shape=(LNew-1,LNew-1))
        if i == j:
            Bulk[i,j]=H2
        else:
            Bulk[i,j]=idm

这里H2和idm都是矩阵，我想在Kronecker产品下合并它们。 但是由于Bulk是一个ndarray对象，我想它不会接受其中的arraylike对象。

编辑：

这是我要使用此功能的功能。 我正在用它为量子自旋链建立哈密顿矩阵。 因此，H2是两个粒子链的哈密顿量，H2是4x4矩阵，idm是2x2恒等矩阵。

现在三个粒子链是np.kron（H2，idm）+ np.kron（idm，H2）

对于四个粒子np.kron（np.kron（H2，idm），idm）+ np.kron（idm，np.kron（H2，idm））+ np.kron（idm，np.kron（idm，H2） ）等。

def ExpandHN(LNew):
idm = np.identity(2)
H2 = GetH(2,'N')
HNew = H2
for i in range(LNew-2):
    for j in range(LNew-2):
        Bulk = np.empty(shape=(LNew-1,LNew-1))
        if i == j:
            Bulk[i,j]=H2
        else:
            Bulk[i,j]=idm
i = 0
for i in range(LNew-2):
    for j in range(LNew-3):
        HNew += np.kron(Bulk[i,j],Bulk[i,j+1]) #something like this

return HNew

如您所见，第二组for循环尚未制定。

话虽这么说，如果某人有一个完全不同但可行的解决方案，我也会对此感到满意。

Answer 1

如果我理解正确，那么您的问题可以归结为如何使用numpy创建数组的数组。 我建议使用标准的python对象dict ：

Bulk = dict()
for i in range(LNew-2):
    for j in range(LNew-2):
        if i == j:
            Bulk[(i,j)]=H2
        else:
            Bulk[(i,j)]=idm

通过使用元组作为键，您可以维护矩阵的类似数组的索引。 还要注意，您应该在两个for循环之外定义Bulk（无论如何）。

HTH