Convertir la série Pandas en DataFrame


92

J'ai une série Pandas sf:

email
email1@email.com    [1.0, 0.0, 0.0]
email2@email.com    [2.0, 0.0, 0.0]
email3@email.com    [1.0, 0.0, 0.0]
email4@email.com    [4.0, 0.0, 0.0]
email5@email.com    [1.0, 0.0, 3.0]
email6@email.com    [1.0, 5.0, 0.0]

Et je voudrais le transformer en DataFrame suivant:

index | email             | list
_____________________________________________
0     | email1@email.com  | [1.0, 0.0, 0.0]
1     | email2@email.com  | [2.0, 0.0, 0.0]
2     | email3@email.com  | [1.0, 0.0, 0.0]
3     | email4@email.com  | [4.0, 0.0, 0.0]
4     | email5@email.com  | [1.0, 0.0, 3.0]
5     | email6@email.com  | [1.0, 5.0, 0.0]

J'ai trouvé un moyen de le faire, mais je doute que ce soit le plus efficace:

df1 = pd.DataFrame(data=sf.index, columns=['email'])
df2 = pd.DataFrame(data=sf.values, columns=['list'])
df = pd.merge(df1, df2, left_index=True, right_index=True)

4
Dans les versions plus récentes de pandas, cela peut être réalisé avec un seul reset_indexappel .
cs95

Réponses:


137

Plutôt que de créer 2 dfs temporaires, vous pouvez simplement les transmettre en tant que paramètres dans un dict en utilisant le constructeur DataFrame:

pd.DataFrame({'email':sf.index, 'list':sf.values})

Il existe de nombreuses façons de construire un df, voir la documentation


une autre excellente option est de concaténer si votre série a les mêmes axespd.concat([sf.index, sf.values], axis=1)
Lauren

63

to_frame () :

En commençant par la série suivante, df:

email
email1@email.com    A
email2@email.com    B
email3@email.com    C
dtype: int64

J'utilise to_frame pour convertir la série en DataFrame:

df = df.to_frame().reset_index()

    email               0
0   email1@email.com    A
1   email2@email.com    B
2   email3@email.com    C
3   email4@email.com    D

Il ne vous reste plus qu'à renommer le nom de la colonne et à nommer la colonne d'index:

df = df.rename(columns= {0: 'list'})
df.index.name = 'index'

Votre DataFrame est prêt pour une analyse plus approfondie.

Mise à jour: Je viens de tomber sur ce lien où les réponses sont étonnamment similaires aux miennes ici.


1
series_obj.to_frame()travaux! Je produis ce type de classe<class 'pandas.core.frame.DataFrame'>
Johnny Zhang

1
Pourquoi utiliser to_frame().reset_index()plutôt que juste reset_index? Vous pourriez même faire justereset_index(name='list')
dumbledad

17

Series.reset_indexavec nameargument

Souvent, le cas d'utilisation survient où une série doit être promue en DataFrame. Mais si la série n'a pas de nom, il reset_indexen résultera quelque chose comme,

s = pd.Series([1, 2, 3], index=['a', 'b', 'c']).rename_axis('A')
s

A
a    1
b    2
c    3
dtype: int64

s.reset_index()

   A  0
0  a  1
1  b  2
2  c  3

Là où vous voyez le nom de la colonne est "0". Nous pouvons résoudre ce problème en spécifiant un nameparamètre.

s.reset_index(name='B')

   A  B
0  a  1
1  b  2
2  c  3

s.reset_index(name='list')

   A  list
0  a     1
1  b     2
2  c     3

Series.to_frame

Si vous souhaitez créer un DataFrame sans promouvoir l'index dans une colonne, utilisez Series.to_frame, comme suggéré dans cette réponse . Cela prend également en charge un paramètre de nom.

s.to_frame(name='B')

   B
A   
a  1
b  2
c  3

pd.DataFrame Constructeur

Vous pouvez également faire la même chose qu'en Series.to_framespécifiant un columnsparamètre:

pd.DataFrame(s, columns=['B'])

   B
A   
a  1
b  2
c  3

Je me demandais pourquoi on pourrait utiliser à la to_frameplace de reset_index, mais y a-t-il jamais une bonne raison d'utiliser les deux? ici
dumbledad

@dumbledad principalement utilitaire. Si vous voulez une seule dataframe col avec index, utilisez to_frame (). Si vous avez besoin de deux colonnes (une de l'index de la série et l'autre des valeurs de la série elle-même), utilisez reset_index ().
cs95

Et si je veux convertir Series en DataFrame avec l'index Seires utilisé comme noms de colonnes DataFrame (c'est-à-dire transposés)? to_framene semble pas avoir d'argument pour faire cela. Merci.
Confondu le

@Confounded utilise to_frame (). T pour le transposer
cs95

17

La réponse en une ligne serait

myseries.to_frame(name='my_column_name')

Ou

myseries.reset_index(drop=True, inplace=True)  # As needed

4

Series.to_framepeut être utilisé pour convertir un Seriesen DataFrame.

# The provided name (columnName) will substitute the series name
df = series.to_frame('columnName')

Par exemple,

s = pd.Series(["a", "b", "c"], name="vals")
df = s.to_frame('newCol')
print(df)

   newCol
0    a
1    b
2    c

1

probablement noté comme une manière non pythonique de le faire, mais cela donnera le résultat que vous voulez dans une ligne:

new_df = pd.DataFrame(zip(email,list))

Résultat:

               email               list
0   email1@email.com    [1.0, 0.0, 0.0]
1   email2@email.com    [2.0, 0.0, 0.0]
2   email3@email.com    [1.0, 0.0, 0.0]
3   email4@email.com    [4.0, 0.0, 3.0]
4   email5@email.com    [1.0, 5.0, 0.0]
En utilisant notre site, vous reconnaissez avoir lu et compris notre politique liée aux cookies et notre politique de confidentialité.
Licensed under cc by-sa 3.0 with attribution required.