选择行，其中单个行的总和等于

Question

我有下面的数据帧。我想提取每个“InvoiceNo”的单个“ItemTotal”等于“InvoiceTotal”的行。 Data

The source data

在这个例子中。我应该可以提取1号和3号发票。因为，它的单项合计等于发票合计。你知道吗

我试过下面的代码。但它不起作用。你知道吗

import pandas as pd
import numpy as np

np.where(df.groupby('InvoiceNo')['ItemTotal'].sum() == df['InvoiceTotal'])

Answer 1

你可以用transform

np.where(df.groupby('InvoiceNo')['ItemTotal'].transform('sum') == df['InvoiceNo'])