x 和 y 必须具有相同的第一维，但形状 (40,) 和 (1, 80) 错误

Question

我收到错误 x 和 y 必须具有相同的第一维，但具有形状 (40,) 和 (1, 80)，我尝试查找它并被告知使用 numpy.array 如下所示但它仍然无法正常工作。 我知道这是由于“- Stot”部分，但不确定该怎么做

import numpy
import matplotlib.pyplot as plt
import math
import scipy.special

kb = 1.380649*(10**-23)
Na = 20
Nb = 20
Ntot = int(Na + Nb)
qa = 0
qb = 41
Stot = 0
qaa = []
Saa = []
Sba = []
Stota = []
Snew = []
prob = 0
h = 1
pos = 0
suma = []
sumnew = []

#creates an array based on the value of qb starting from 0 to qb in increments of 1
for k in range (qa, qb-1):
    qa = qa + 1
    qb = qb - 1
    qaa += [qa]
    Sa = (qa + Na - 1) * (numpy.log(qa + Na - 1)) - (qa*numpy.log(qa)) - (Na*numpy.log(Na-1))
    Sb = (qb + Nb - 1) * (numpy.log(qb + Nb - 1)) - (qb*numpy.log(qb)) - (Nb*numpy.log(Nb-1))
    Stot = Sa + Sb + Stot
    Saa += [Sa]
    Sba += [Sb]
    
Prob = numpy.array([(Saa + Sba - Stot)])

plt.plot(qaa, Prob)
plt.title("Graph to show probability distribution of the macrostates, P(qa)")
plt.xlabel('qa')
plt.ylabel('Probability')

Answer 1

我不能确定 rhis 代码。 该消息似乎来自 plot 命令，但是具有表达式“（Saa + Sba - Stot）”的行应该会触发错误，因为您的 Saa 和 Sba 是列表，并且它们的附加结果是两个列表的串联. 列表减去浮点数（Stot 是浮点数）是不允许的操作如果要按成员添加列表的值，然后从每个项目中减去一个值，请在执行减法之前将每个列表转换为数组（数组减去允许浮动）

然而，这个熵计算似乎有一个错误。 Stota 的列表是做什么用的？