筛选dataframe值为1的数据

假设你有一个名为 `df` 的 DataFrame，其中包含一些值为 0 或 1 的数据。你可以使用以下代码来筛选出所有值为 1 的行：

df[df == 1]

这将返回一个与原始 DataFrame 结构相同的 DataFrame，其中值为 1 的单元格保留，而其他单元格则被替换为 NaN。如果你只想获取所有包含至少一个 1 的行，可以使用以下代码：

df.loc[(df == 1).any(axis=1)]

这将返回一个 DataFrame，其中包含所有至少包含一个值为 1 的行。