Rozdeliť dataframe stĺpec s pandy série do viacerých stĺpcov

0

Otázka

Mám pandy dataframe s veľa stĺpcov. Jedným z nich je séria. Som si chcú rozdeliť, že stĺpec do súboru boolean stĺpce. Takže, ak hodnota v riadku je ['Červená','Hot','Letné'], chcem, 3 stĺpce: Červená (so hodnota 1), Horúce (s hodnotou 1) a Letný (s hodnotu 1).

Príklad:

df = pd.DataFrame({'Owner': ['Bob', 'Jane', 'Amy'],
               'Make': ['Ford', 'Ford', 'Jeep'],
               'Model': ['Bronco', 'Bronco', 'Wrangler'],
               'Sentiment': [['Meh','Red','Dirty'], ['Rusty','Sturdy'], ['Dirty','Red']],
               'Max Speed': [80, 150, 69],
              'Customer Rating': [90, 50, 91]})

dáva nám:

enter image description here

Teraz chcem, aby tento výstup: enter image description here (True/False by mohli byť tie a núl, príliš. Rovnako dobre).

poznámka: pozrel som sa na tento príspevok nižšie: Rozdeliť Pandy stĺpec uvádza do viacerých stĺpcov ale len priamo funguje, ak je váš seriál nie je už súčasťou DF.

akákoľvek pomoc ocenia!

dataframe pandas python
2021-11-24 02:59:52
2

Najlepšiu odpoveď

2

Skúste explode potom crosstab a join

s = df.Sentiment.explode()
out = df.join(pd.crosstab(s.index,s).astype(bool))
out
  Owner  Make     Model          Sentiment  ...    Meh    Red  Rusty  Sturdy
0   Bob  Ford    Bronco  [Meh, Red, Dirty]  ...   True   True  False   False
1  Jane  Ford    Bronco    [Rusty, Sturdy]  ...  False  False   True    True
2   Amy  Jeep  Wrangler       [Dirty, Red]  ...  False   True  False   False
[3 rows x 11 columns]
2021-11-24 14:07:56
0

Skúste toto:

df = pd.concat([df, pd.get_dummies(df['Sentiment'].explode())], axis=1)

Výstup:

>>> df
  Owner  Make     Model             Sentiment  Max Speed  Customer Rating  AWESOME  Dirty  LOVE  Meh  Red  Rusty  Sturdy
0   Bob  Ford    Bronco     [Meh, Red, Dirty]         80               90        0      0     0    1    0      0       0
0   Bob  Ford    Bronco     [Meh, Red, Dirty]         80               90        0      0     0    0    1      0       0
0   Bob  Ford    Bronco     [Meh, Red, Dirty]         80               90        0      1     0    0    0      0       0
1  Jane  Ford    Bronco       [Rusty, Sturdy]        150               50        0      0     0    0    0      1       0
1  Jane  Ford    Bronco       [Rusty, Sturdy]        150               50        0      0     0    0    0      0       1
2   Amy  Jeep  Wrangler  [LOVE, AWESOME, Red]         69               91        0      0     1    0    0      0       0
2   Amy  Jeep  Wrangler  [LOVE, AWESOME, Red]         69               91        1      0     0    0    0      0       0
2   Amy  Jeep  Wrangler  [LOVE, AWESOME, Red]         69               91        0      0     0    0    1      0       0

Ako to funguje

To, čo ste hľadali sa zvyčajne nazýva one-hot kódovanie, a tam je metóda, v pandy len pre to, že: get_dummies(). To trvá niekoľko (alebo DataFrame) a vytvorí nový stĺpec pre každý jedinečnú hodnotu v tom, že Rad (alebo DataFrame).

df['Sentiment'].explode() vytvorí nový stĺpec, ktorý obsahuje všetky jednotlivé hodnoty všetkých zoznamov v vybratého stĺpca(ov).

2021-11-24 03:25:47

V iných jazykoch

Táto stránka je v iných jazykoch

Русский
..................................................................................................................
Italiano
..................................................................................................................
Polski
..................................................................................................................
Română
..................................................................................................................
한국어
..................................................................................................................
हिन्दी
..................................................................................................................
Français
..................................................................................................................
Türk
..................................................................................................................
Česk
..................................................................................................................
Português
..................................................................................................................
ไทย
..................................................................................................................
中文
..................................................................................................................
Español
..................................................................................................................