将查询结果存储在 plpython 函数中的列表变量中答案

【问题标题】：Storing the query result in a list variable in plpython function将查询结果存储在 plpython 函数中的列表变量中
【发布时间】：2013-12-06 19:45:26
【问题描述】：

我对 postgresql 和编写函数很陌生，所以请多多包涵。我需要将 Python 脚本转换为 postgresql 函数，并且我打算为此目的使用 PL/Python。但是我在这样做时遇到了一些问题。执行函数时收到错误：

错误：类型错误：+ 不支持的操作数类型：“int”和“dict” SQL状态：XX000 上下文：回溯（最近一次通话最后一次）： PL/Python 函数“椭圆”，第 5 行，在 meanX=float(sum(Xarray))/len(Xarray) if len(Xarray) > 0 else float('nan') PL/Python 函数“椭圆”

据我所知，查询将结果存储在字典中，然后导致此错误（因为我试图在脚本中使用列表进行操作）。至少我认为这可能是问题所在。所以我的问题是 - 有没有办法将查询结果存储在列表变量中？

CREATE OR REPLACE FUNCTION ellipse()
returns setof ellipse_param as $$

Xarray=plpy.execute("select laius from proov")
Yarray=plpy.execute("select pikkus from proov")
meanX=float(sum(Xarray))/len(Xarray) if len(Xarray) > 0 else float('nan')
meanY=float(sum(Yarray))/len(Yarray) if len(Yarray) > 0 else float('nan')
Xdevs=[]
Ydevs=[]
for x in Xarray:
    dev=x-meanX
    Xdevs.append(dev)
    dev=0
for y in Yarray:
    dev=y-meanY
    Ydevs.append(dev)
    dev=0

sumX=0 
sumY=0
for x in Xdevs:
    sumX+=x**2
for y in Ydevs:
    sumY+=y**2
Xaxes=sqrt(sumX/len(Xdevs))
Yaxes=sqrt(sumY/len(Ydevs))
A=sumX-sumY
B=sqrt(A**2+(((float(sum([a*b for a,b in zip(Xdevs,Ydevs)])))**2)*4))
C=float(sum([a*b for a,b in zip(Xdevs,Ydevs)]))*2
rotation=(atan(((A+B)/C)))
Sx=sqrt(((float(sum([(a*cos(rotation)-b*sin(rotation))**2 for a,b in  zip(Xdevs,Ydevs)])))/(len(Xdevs)-2))*2)
Sy=sqrt(((float(sum([(c*sin(rotation)+d*cos(rotation))**2 for c,d in zip(Xdevs,Ydevs)])))/(len(Xdevs)-2))*2)
return meanX, meanY, rotation, Xaxes, Yaxes
$$ LANGUAGE plpython3u;

【问题讨论】：

标签： postgresql plpython

【解决方案1】：

plpy.execute 会给你一个字典列表，所以你想要类似的东西

sum([x['laius'] for x in Xarray])

更多信息请参阅此处的文档http://www.postgresql.org/docs/devel/static/plpython-database.html

编辑：我读得太快了，浏览了你的整个函数——你可能想把列表构造函数放在更高的位置，可能就在执行查询之后，这样你就有一个值列表供以后使用（我没有没有注意到后面的代码中有多少假设数据在一个简单的列表中）。

【讨论】：

谢谢！你的回答真的很有帮助！