Question

我有一个带有一列列表的pandas数据框，我想找到一种方法来返回一个数据框，其中一列中的列表和另一列中的总计数。我的问题是找到一种方法来将包含相同值的列表添加到一起，例如我想找到[＆＃39; a＆＃39;，＆＃39; b＆＃39;]和[＆＃39;]的总和。 b＆＃39;，＆＃39; a＆＃39;]最后。

例如数据框：

Lists          Count
['a','b']        2
['a','c']        4
['b','a']        3

将返回：

Lists          Count
['a','b']        5
['a','c']        4

Answer 1

列表不可用。所以，排序并转换为元组，

In [80]: df
Out[80]: 
   count   lists
0      2  [a, b]
1      4  [a, c]
2      3  [b, a]

In [82]: df['lists'] = df['lists'].map(lambda x: tuple(sorted(x)))

In [83]: df
Out[83]: 
   count   lists
0      2  (a, b)
1      4  (a, c)
2      3  (a, b)

In [76]: df.groupby('lists').sum()
Out[76]: 
    count
lists        
(a, b)      5
(a, c)      4

Answer 2

你也可以使用集合（在将它们强制转换为字符串之后）。

df = pd.DataFrame({'Lists': [['a', 'b'], ['a', 'c'], ['b', 'a']], 
                   'Value': [2, 4, 3]})

df['Sets'] = df.Lists.apply(set).astype(str)

>>> df.groupby(df.Sets).Value.sum()
Sets
set(['a', 'b'])    5
set(['a', 'c'])    4
Name: Value, dtype: int64

计算数据框中的唯一列表

2 个答案: