时间戳

首页 > 解决方案 > 迭代列表并通过函数传递结果并将结果保存在数据框中

python - 迭代列表并通过函数传递结果并将结果保存在数据框中

问题描述

我有这个清单：

Peer_List_Members = [1967、2338、421、1275、2499、2240、881、2719、2894、2198、3120、3160、635]

我遍历这个列表如下：

for x in Peer_List_Members:
        x = ID_Search(str(x))
        print(x)

上面的函数通过函数“ID_Search(input_id)”传递每个元素，然后在数据库中查找数字的值并返回查找值：

def ID_Search(input_id): 
    returns_name = pd.read_sql_query("""
    SELECT BaseHoldingID
    FROM Holdings
    WHERE HoldingID=""" + input_id
    , conn)
    returns_name = pd.DataFrame(returns_name)
    BaseHoldingID = returns_name['BaseHoldingID'][0]
    return(BaseHoldingID)

print(x) 的输出如下：

1774
2134
421
1267
2295
2046
867
2517
2694
2010
2874
2913
629

但是当迭代并尝试使用以下方法保存到新列表（lst）时：

lst=[]
for x in Peer_List_Members:
    x = ID_Search(str(x))
    lst.append(x)
    print(lst)

我得到这个奇怪的输出！

[1774]
[1774, 2134]
[1774, 2134, 421]
[1774, 2134, 421, 1267]
[1774, 2134, 421, 1267, 2295]
[1774, 2134, 421, 1267, 2295, 2046]
[1774, 2134, 421, 1267, 2295, 2046, 867]
[1774, 2134, 421, 1267, 2295, 2046, 867, 2517]
...........
[1774, 2134, 421, 1267, 2295, 2046, 867, 2517, 2694, 2010, 2874, 2913, 629]

问题：首先我如何在附加时修复奇怪的输出。但理想情况下，我想将原始列表与新列表组合成一个数据框。

感谢您的帮助！非常感激。

标签： pythonpandaslistdataframefunction

解决方案

好的 - 我做了一些挖掘/谷歌搜索，并且能够找到一种方法......

def peerBaseHolding():
     return [ID_Search(str(number)) for number in Peer_List_Members]

peerBaseHolding = peerBaseHolding()
peerBaseHolding = pd.DataFrame(peerBaseHolding, columns={'BaseHoldingID'})
Peer_List_Members = pd.DataFrame(Peer_List_Members, columns={'HoldingID'})

peerList = Peer_List_Members.merge(peerBaseHolding, how='inner', left_index=True, right_index=True)
peerList

它有效，但有更好的方法吗？

推荐阅读