按2列python分组

Question

我有一个形状的数据框：

 id | V
 1    0
 2    1
 3    0
 4    0
 1    0
 2    0
 4    1

其中 V 仅为 1 或 0 并且 id 可以有重复项。我想要的是每个 id 计算 0 和 1。上面的例子是

    1 : ('0':2,'1':0)
    4 : ('0':1,'1':1)
    3 : ('0':1,'1':0)
    2 : ('0':1,'1':1)

我不关心结果的格式，这只是一个例子。它可以是一个数据框，其中 0 和 1 作为列或其他。

有什么想法吗？

Answer 1

您可以使用value_counts 方法。

例子：

df.groupby("id").V.value_counts()

Answer 2

获取 Json 格式。你可以这样做。

df.groupby(['id','V'])['V'].count().unstack().T.fillna(0).astype(int).to_json()

输出：

{
  "1":{"0":2,"1":0},
  "2":{"0":1,"1":1},
  "3":{"0":1,"1":0},
  "4":{"0":1,"1":1}
}