如何在 numpy 数组 pandas 列中使用 groupby 进行值计数

Question

我有一个 pandas dataframe 像这样：

import pandas as pd
foo = pd.DataFrame({'s': ['a','b','a','a'], 'versions':[['1','2'],'2',['1','2'],'2']})

我想value_counts() s按 group 列versions

我努力了

foo.groupby('s')['versions'].value_counts()

但我收到一个错误TypeError: unhashable type: 'list' ，有什么想法吗？

Answer 1

您可以将列表/数组转换为可散列的元组，但因为还有字符串需要if-else语句来避免将字符串100拆分为元组(1,0,0) ：

foo['versions'] = foo['versions'].apply(lambda x: tuple(x) if isinstance(x, list) else (x,))
print (foo)
   s versions
0  a   (1, 2)
1  b     (2,)
2  a   (1, 2)
3  a     (2,)

s = foo.groupby('s')['versions'].value_counts()
print (s)
s  versions
a  (1, 2)      2
   (2,)        1
b  (2,)        1
Name: versions, dtype: int64