2017-02-01 6 views
2

Ich habe einen Datenrahmen df1 wie folgt aus:Gruppenlisten verschiedener Reihen von mehreren Spalten Werte mit Pandas

import pandas as pd 
dic = {'A':[0,0,2,2,2,1,5,5],'B':[[1,5,3,8],[1,8,7,5],[7,8,9,5],[3],[1,5,9,3],[0,3,5],[],[4,2,3,1]],'C':['a','b','c','c','d','e','f','f'],'D':['0','8','7','6','4','5','2','2']} 
df1 = pd.DataFrame(dic) 

und sieht wie folgt aus:

#Initial dataframe 
    A    B C D 
0 0 [1, 5, 3, 8] a 0 
1 0 [1, 8, 7, 5] b 8 
2 2 [7, 8, 9, 5] c 7 
3 2   [3] c 6 
4 2 [1, 5, 9, 3] d 4 
5 1  [0, 3, 5] e 5 
6 5   [] f 2 
7 5 [4, 2, 3, 1] f 2 

Mein Ziel zu Gruppenzeilen, die die haben gleiche Werte in Spalte A und C und füge den Inhalt der Spalte B so zusammen, dass das Ergebnis wie folgt aussieht:

#My GOAL 
    A    B C 
0 0  [1, 5, 3, 8] a 
1 0  [1, 8, 7, 5] b 
2 2 [3, 7, 8, 9, 5] c 
3 2  [1, 5, 9, 3] d 
4 1  [0, 3, 5] e 
5 5  [4, 2, 3, 1] f 

Wie Sie sehen können, werden Zeilen mit den gleichen Elementen in den Spalten A und C zusammengeführt, und wenn mindestens eine davon unterschiedlich ist, bleiben sie unverändert. Meine Idee war es, die groupby und sum Funktionen wie folgt zu verwenden:

df1.groupby(by=['A','C'],as_index=False,sort=True).sum() 

aber Python eine Fehlermeldung zurück: Function does not reduce

Könnten Sie mir bitte sagen, was mit meinem Codezeile falsch ist? Was soll ich schreiben, um mein Ziel zu erreichen?

Hinweis: Es ist mir egal, was mit Spalte D passiert, die getrennt werden kann.

Antwort

1

Eine der Möglichkeiten wäre, die list von Listen zu glätten, bis sie mit Hilfe von itertools.chain(*iterables)

import itertools 
df1.groupby(['A', 'C'])['B'].apply(lambda x: list(itertools.chain(*x))).reset_index() 

(Oder)

Verwendung sum mit lambda erschöpft wird:

df1.groupby(by=['A','C'])['B'].apply(lambda x: x.sum()).reset_index() 

Beide Ausbeute:

enter image description here

sucht standardmäßig groupby().sum() für numerische Typen (Skalar) Werte Aggregation auszuführen und nicht eine Sammlung von Elementen wie list zum Beispiel.

1

Eine andere Möglichkeit:

df1.groupby(by=['A','C'],as_index=False,sort=True).agg({'B': lambda x: tuple(sum(x, []))}) 

Ergebnis:

A C    B 
0 0 a  (1, 5, 3, 8) 
1 0 b  (1, 8, 7, 5) 
2 1 e  (0, 3, 5) 
3 2 c (7, 8, 9, 5, 3) 
4 2 d  (1, 5, 9, 3) 
5 5 f  (4, 2, 3, 1) 

Sitz in this answer (es scheint, dass die Listen mit Aggregation nicht allzu gut funktionieren).

Verwandte Themen