如何修改这个 Python 定义函数代码？

Question

有人可以看看下面的 Python 函数代码吗？

import numpy as np
import scipy.stats as stats
from scipy.stats import poisson, norm

cs = 100
co = 300
mu = 4.7

G = poisson(mu)
p = G.pmf(np.arange(3*mu))

# Define Z(Q) function 

def Z(Q):
    for i in range(len(p)):
        return sum(p[i]*cs*max((Q-i), 0) + p[i]*co*max((i-Q), 0))

# Plot Q and (Q)

import pylab as pl

x = []
y = []

for Q in range(0, 12):
    x.append(Q)
    y.append(Z(Q))

pl.plot(x, y, '-o')
pl.show()

错误消息显示在最后一个“绘图”过程中： 在 Z(Q) 中，'numpy.float64' 对象不可迭代。

我希望 Z(Q) 返回的值对范围 (0, len(p)) 中的每个 i 求和，这是一个带有变量 Q 的函数。最后绘图是 Q(X Axis) 和 Z(Q )（Y 轴）可以绘制每个 Q。

如何修改 Z(Q)？谢谢！

如果我想输出 [Q, Z(Q)] 我该怎么做呢？我的代码：

   with open('opt.csv', 'wb') as fd:
        a = csv.writer(fd, delimiter=',')
        data1 = [['Order_Number', 'Cost'],
                 [Q, Z(Q)]]
        a.writerows(data1) 

Answer 1

如果您希望 Z(Q) 返回多个值，您可能需要使用“yield”语句：

def Z(Q):
    for i in range(len(p)):
        yield sum(p[i]*cs*max((Q-i), 0) + p[i]*co*max((i-Q), 0))

这将允许它被迭代并返回每次运行 for 循环的总和。

如果您需要将输出作为列表而不是通用可迭代对象，则需要在该调用中构造列表，或者将对 Z(Q) 的调用包装在列表中：list(Z(Q))

---

编辑：

仔细查看（并安装了正确的库...）后，我的原始答案似乎不正确。堆栈跟踪（理想情况下将包含在您的问题中:)）指向 Z 本身的返回行。这是因为您的原始代码在每次迭代 range(len(p)) 时都尝试运行对 sum 的调用，而不是在所有结果上运行一次。

您可以调整您的代码，通过将总和移到函数之外 来正确返回单个总和值，即：

import numpy as np
import scipy.stats as stats
from scipy.stats import poisson, norm

cs = 100
co = 300
mu = 4.7

G = poisson(mu)
p = G.pmf(np.arange(3*mu))

# Define Z(Q) function 

def Z(Q):
    for i in range(len(p)):
        yield p[i]*cs*max((Q-i), 0) + p[i]*co*max((i-Q), 0)

# Plot Q and (Q)

import pylab as pl

x = []
y = []

for Q in range(0, 12):
    x.append(Q)
    y.append(sum(Z(Q)))

pl.plot(x, y, '-o')
pl.show()

总结变化：

• Z(Q) 现在按所述生成函数的每次迭代
• sum 操作现在在Z(Q) 的结果上运行

希望对你有帮助！