在 Python 中按多个条件对数据进行分组

Question

我想我有一个快速的问题，但我没有找到用简单的话谷歌它的方法。

我有一个这样的原始数据集：

Number of account     Value
123               100
456               300
789               400
910               100
674               250

我有一个方法表来整合这些原始数据，形成有用的东西。它看起来像：

Variable              Number of account
"a"                  123, 456, 910
"b"                    789,674

所以，最后我想得到这样的表格：

Variable              Number of account
"a"                  Sum of values for(123, 456, 910)
"b"                  Sum of values for(789,674)

我最初的想法是执行以下操作：对于方法表中的每一行，对于方法表中的每个帐户数，原始数据中的值求和。

两个问题：

1. 巩固它的最佳方法是什么？
2. 如果在方法表中帐户数是逗号分隔的字符串怎么办？("123,456,910").我可以在熊猫数据帧中的一个单元格中存储多个数字吗

Answer 1

假设我在两个数据帧中有数据：

df是：

Number_of_account     Value
123               100
456               300
789               400
910               100
674               250

table_2是：

Variable              Number_of_account
"a"                  123,456,910
"b"                    789,674

首先，我将从 table2 创建一个查找表：

lookup_table = pd.concat([pd.Series(row['Variable'], row['Number_of_account'].split(','))              
for _, row in table_2.iterrows()]).reset_index()
lookup_table.columns = ["Number_of_account", "variable"]
lookup_table.Number_of_account = pd.to_numeric(lookup_table.Number_of_account)

结果是：

Number_of_account variable
0                123        a
1                456        a
2                910        a
3                789        b
4                674        b

然后，我将主数据帧(df(与查找表合并，并使用groupby来计算值的总和。

df = pd.merge(df, lookup_table, on="Number_of_account")
df.groupby("variable")["Value"].sum()

结果是：

variable
a    500
b    650