2017-10-18 6 views
0

J'ai le code ci-dessous qui transforme un fichier CSV avec deux colonnes dans un dictionnaire:Dictionnaire des données au format CSV avec Pandas en Python 3

import pandas as pd 

df = pd.read_csv('file.csv', sep=',') 
dct = df.groupby('group').ip.apply(lambda x: x.tolist()).to_dict() 

print(dct) 
{'A': ['192.168.1.1', '192.168.1.4'], 
'B': ['192.168.1.2', '192.168.1.5'], 
'C': ['192.168.1.3', '192.168.1.6']} 

Comment pourrais-je aller à faire l'inverse de cela? Je dois ce dictionnaire:

dct = {'A': ['192.168.1.1', '192.168.1.4'], 
'B': ['192.168.1.2', '192.168.1.5'], 
'C': ['192.168.1.3', '192.168.1.6']} 

et que vous voulez le transformer au format CSV avec deux colonnes comme celui-ci:

ip,group 
192.168.1.1,A 
192.168.1.2,B 
192.168.1.3,C 
192.168.1.4,A 
192.168.1.5,B 
192.168.1.6,C 

utilisateur @ cᴏʟᴅsᴘᴇᴇᴅ m'a aidé pour la première partie et Pandas est au-dessus de mes compétences en python à le moment. Merci pour toute aide!

Répondre

1

pd.DataFrame + melt.

df = pd.DataFrame(dct).melt(var_name='group', value_name='ip') 
print(df) 
    group   ip 
0  A 192.168.1.1 
1  A 192.168.1.4 
2  B 192.168.1.2 
3  B 192.168.1.5 
4  C 192.168.1.3 
5  C 192.168.1.6 

Enregistrer dans un fichier CSV en utilisant df.to_csv:

df.to_csv('out.csv', sep=',', index=False) 
+0

@coldspeed Je suis super impressionné comment cela fonctionne avec 2 lignes de code! Un moyen de passer la première colonne avec le 0-5? Je dois trouver un bon livre et apprendre les pandas! – user1781482

+1

@ user1781482 Modifié, définissez 'index = False' :-) –