gpt4 book ai didi

python - 从 Pandas 数据框中提取非嵌套列表

转载 作者:行者123 更新时间:2023-12-05 02:06:34 25 4
gpt4 key购买 nike

df = pd.DataFrame({'x':['a','b','c','d'],'y':[[1],[3,4],[2],[5]]})

我想提取列表[3,4],所以我做了

df.loc[df.x=='b','y'].tolist()

但它返回一个嵌套列表。

[[3,4]]

如何从数据框中获取非嵌套列表?

最佳答案

你的 loc 表达式返回一个序列而不是一个标量,你只需要选择它的第一个也是唯一的元素:

df.loc[df.x=='b','y'].iloc[0]

关于python - 从 Pandas 数据框中提取非嵌套列表,我们在Stack Overflow上找到一个类似的问题: https://stackoverflow.com/questions/62724184/

25 4 0
Copyright 2021 - 2024 cfsdn All Rights Reserved 蜀ICP备2022000587号
广告合作:1813099741@qq.com 6ren.com