2015-03-31 11 views
22

Mam następujące DataFrame, gdzie Track ID jest indeksem wiersza. Jak mogę podzielić ciąg znaków w kolumnie stats na 5 kolumn liczb?Podziel ciągi w krotkach w kolumny, w pandach

Track ID stats 
14.0 (-0.00924175824176, 0.41, -0.742016492568, 0.0036830094242, 0.00251748449963) 
28.0 (0.0411538461538, 0.318230769231, 0.758717081514, 0.00264000622468, 0.0106535783677) 
42.0 (-0.0144351648352, 0.168438461538, -0.80870348637, 0.000816872566404, 0.00316572586742) 
56.0 (0.0343461538462, 0.288730769231, 0.950844962874, 6.1608706775e-07, 0.00337262030771) 
70.0 (0.00905164835165, 0.151030769231, 0.670257006716, 0.0121790506745, 0.00302182567957) 
84.0 (-0.0047967032967, 0.171615384615, -0.552879463981, 0.0500316517755, 0.00217970256969) 
+0

Uczynisz rzeczy znacznie łatwiejszymi dla innych, umieszczając df.to_dict() w ramce danych zamiast jej reprezentacji. – chthonicdaemon

+1

Kolumna statystyk, czy zawiera łańcuch, który wygląda jak krotka, czy zawiera krotki? – joris

+0

możliwy duplikat [Konwertuj pandy "Seria par macierzowych" na "dwukolumnowe DataFrame"?] (Http://stackoverflow.com/questions/29346512/convert-a-pandas-series-of-par-grach do dwukolumnowej ramki danych) – TheBlackCat

Odpowiedz

43

i za drugim przypadku, zakładając, że są ciągi, które wyglądają jak krotek:

In [74]: df['stats'].str[1:-1].str.split(',', expand=True).astype(float) 
Out[74]: 
      0   1   2   3   4 
0 -0.009242 0.410000 -0.742016 0.003683 0.002517 
1 0.041154 0.318231 0.758717 0.002640 0.010654 
2 -0.014435 0.168438 -0.808703 0.000817 0.003166 
3 0.034346 0.288731 0.950845 0.000001 0.003373 
4 0.009052 0.151031 0.670257 0.012179 0.003022 
5 -0.004797 0.171615 -0.552879 0.050032 0.002180 

(uwaga: dla starszych wersji pandy (< 0.16.1), trzeba użyć return_type='frame' zamiast słowa kluczowego ekspansji)

Nawiasem mówiąc, jeśli nie jest krotki i ciągi, wy can po prostu wykonaj następujące czynności:

df['stats'].apply(pd.Series) 

Ten ostatni można również łączyć z powyższych (jeśli return_type='frame' nie jest jeszcze dostępna):

df['stats'].str[1:-1].str.split(',').apply(pd.Series).astype(float) 
+0

Dzięki za wejście. Jestem nowy w pandach, pythonach i skryptach w ogóle, więc wciąż orientuję się w podstawach. Dane w kolumnie "statystyki" były rzeczywiście krotkami. Tworzenie nowej DataFrame z 'df2 = df ['stats']. Apply (pd.Series)' znacznie mnie poruszyło. Dzięki jeszcze raz. –

+6

'Zwrot_typu' jest przestarzałe na rzecz' expand = True'. – tacaswell

+0

Chciałbym móc to powtórzyć – Private

0

Zakładając, że masz kolumnę, która zawiera krotki (jak wydaje się w swoim przykładzie) zamiast strun, to będzie działać:

df = pandas.DataFrame({'Track ID': [14, 28, 42], 'stats': [(1, 2, 3, 4, 5), (1, 2, 3, 4, 5), (1, 2, 3, 4, 5)]}).set_index("Track ID") 

from operator import itemgetter 
for i in range(5): 
    df["Col {}".format(i)] = df.stats.apply(itemgetter(i)) 

Jeśli rzeczywiście mają ciągi, które wyglądają jak krotek, można analizowania ich w pierwszej kolejności, a następnie zastosować ten sam wzór jak wyżej:

df = df2 = pandas.DataFrame({'Track ID': [14, 28, 42], 'stats': ["(1, 2, 3, 4, 5)", "(1, 2, 3, 4, 5)", "(1, 2, 3, 4, 5)"]}).set_index("Track ID") 
df.stats = df2.stats.str.strip("()").str.split(", ") 
1

Jeśli masz sekwencję krotek, a nie strun, a chcesz je jako kolumny DataFrame, to jest najprostsze podejście:

df = pd.concat([df['Track ID'],pd.DataFrame(df['stats'].values.tolist())], axis=1) 

Jeśli to jest rzeczywiście struny, można najpierw przekształcić ją w listach jak tak, następnie zastosuj powyższą operację:

dfpart = pd.DataFrame(df['stats'].apply(lambda x: x.strip('()').split(', ')).values.tolist()).astype(float) 
df = pd.concat([df['Track ID'], dfpart], axis=1)