Pandas中DataFrame按列条件生成JSON文件报错求助

错误根源 #

你遇到的The truth value of a DataFrame is ambiguous错误，是因为df['Row2'] == '10'返回的是一个布尔Series（每一行对应一个True/False值），而if语句需要单个布尔值来判断逻辑，Python无法确定该用Series中的哪个值作为判断依据，因此抛出歧义错误。

正确实现方式 #

根据你的需求，分两种场景给出解决方案：

场景1：拆分数据导出（符合条件的行一个文件，其余行另一个文件） #

这应该是你最可能的需求——把Row2等于'10'的行导出到指定文件，剩余行导出到另一个文件：

# 筛选符合条件的子集
df_match = df[df['Row2'] == '10']
# 筛选不符合条件的子集
df_no_match = df[df['Row2'] != '10']

# 分别导出JSON
df_match.to_json(filelocation)
df_no_match.to_json(diff_filelocation)

场景2：判断整体是否存在/全符合条件后导出整个DataFrame #

如果你的需求是只要数据中存在至少一行Row2等于'10'，就把整个DataFrame导出到第一个文件，否则导出到第二个，可以用.any()或.all()明确生成单个布尔值：

# 判断是否存在至少一行符合条件
if (df['Row2'] == '10').any():
    df.to_json(filelocation)
else:
    df.to_json(diff_filelocation)

# 或者判断是否所有行都符合条件
if (df['Row2'] == '10').all():
    df.to_json(filelocation)
else:
    df.to_json(diff_filelocation)

关于df[df["Row2"]]能运行的原因 #

df[df["Row2"]]是合法的pandas索引操作：它会把Row2列中非空、非零、非False的值对应的行筛选出来，这是pandas允许的布尔索引用法。但if语句需要的是单个布尔值，而非布尔Series，所以直接把布尔Series放到if里就会触发歧义错误。

内容的提问来源于stack exchange，提问作者Rav86