如何修复 Pandas 提取数据？

Question

我在 pandas 中使用此规则在特定列的数据框中查找值，然后 return 返回一些列。如果失败。我得到了错误。

unpack = dfzips.loc[dfzips.zip == element, ['city', 'county_name', 'state_name', "state_id"]].values[0]

错误是：

IndexError: index 0 is out of bounds for axis 0 with size 0

怎么装？

Answer 1

您可能需要检查是否至少有一行在 zip 列中包含元素 value。如果至少有一行 zip 列包含 element 值，那么运行你的代码，否则什么都不做（例如，如果你在循环中做，请使用 continue 命令).

例如，

unpack = []
for element in list_of_element:
    if (dfzips.zip == element).sum() > 0:
        unpack += [dfzips.loc[dfzips.zip == element, ['city', 'county_name', 'state_name', "state_id"]].values[0]]
    else:
        continue

如何修复 Pandas 提取数据？

How to fix Pandas extract data?

python

pandas