如何不使用try-except块将Dataframe过滤无结果时设为0?
不用try-except的解决方案
首先得明确:你原来的try-except块其实根本不会触发——用布尔索引过滤DataFrame,哪怕没有匹配项,也只会返回一个空的DataFrame,不会抛出任何异常,所以那个except分支永远不会执行。
要实现「没找到匹配结果就设为0」的逻辑,直接判断过滤后的结果是否为空即可,不用try-except:
方案1:空结果返回数值0
filtered = df[df['a'] == "ABC"] op = filtered if not filtered.empty else 0
方案2:空结果返回同结构的全0 DataFrame
如果需要保持op为DataFrame类型,只是空的时候填充0,可以这么写:
filtered = df[df['a'] == "ABC"] op = filtered if not filtered.empty else filtered.assign(**{col:0 for col in filtered.columns})
或者更直观的写法:
op = df[df['a'] == "ABC"] if op.empty: op = pd.DataFrame([0]*len(df.columns), columns=df.columns)
内容的提问来源于stack exchange,提问作者ranadeep
相关产品推荐
相关产品推荐

