Wie teilt man CSV in Teile mit Spaltennamen?

Question

Ich habe eine große CSV-Datei, die ich Spalten spalten möchte. Das heißt, einige angegebene Spalten kommen in einen Teil, andere in einen anderen Teil. Ich möchte auch mehr als 2 Teile erstellen können. Wie kann ich das in Python machen? Gibt es auch eine Bibliothek in Python, um viele Datenformate zu verarbeiten?

Eingangsformat:

policyID statecode county eq_site_limit hu_site_limit fl_site_limit fr_site_limit tiv_2011 tiv_2012 eq_site_deductible hu_site_deductible fl_site_deductible fr_site_deductible point_latitude point_longitude line construction point_granularity 

119736 FL CLAY COUNTY 498960 498960 498960 498960 498960 792148.9 0 9979.2 0 0 30.102261 -81.711777 Residential Masonry 1 
448094 FL CLAY COUNTY 1322376.3 1322376.3 1322376.3 1322376.3 1322376.3 1438163.57 0 0 0 0 30.063936 -81.707664 Residential Masonry 3 
206893 FL CLAY COUNTY 190724.4 190724.4 190724.4 190724.4 190724.4 192476.78 0 0 0 0 30.089579 -81.700455 Residential Wood 1 
333743 FL CLAY COUNTY 0 79520.76 0 0 79520.76 86854.48 0 0 0 0 30.063236 -81.707703 Residential Wood 3 
172534 FL CLAY COUNTY 0 254281.5 0 254281.5 254281.5 246144.49 0 0 0 0 30.060614 -81.702675 Residential Wood 1 

Eingangsformat Spalten:

policyID statecode county eq_site_limit hu_site_limit fl_site_limit fr_site_limit tiv_2011 tiv_2012 eq_site_deductible hu_site_deductible fl_site_deductible fr_site_deductible point_latitude point_longitude line construction point_granularity 

Ausgabeformat Spalten:

Part A: ['policyID', 'statecode', 'county', 'eq_site_limit', 'hu_site_limit']

Teil B: ['fl_site_limit', 'fr_site_limit', 'tiv_2011', 'tiv_2012', 'eq_site_deductible', 'hu_site_deductible', 'fl_site_deductible', 'fr_site_deductible', 'point_latitude', 'point_longitude', 'line', 'construction', 'point_granularity']

Code:

import csv 
import pandas as pd 

df = pd.read_csv("FL_insurance_sample.csv") 
cl_list = list(df.columns.values) 
a = cl_list[:5] 
b = cl_list[5:] 

with open('data1.csv', 'w') as datafile: 
    for x in a: 
     saved_column = df[x] 
     datafile.write(saved_column) 

with open('data2.csv', 'w') as datafile: 
    for x in b: 
     saved_column = df[x] 
     datafile.write(saved_column) 

Answer 1

Ich gehe davon aus, dass Sie bestimmte Spalten von Ihrem ursprünglichen Datenrahmen auf neue Datenrahmen und dann auf csv aufteilen möchten.
Lassen Sie mich wissen, wenn diese Annahme nicht korrekt ist, da die Antwort darauf basiert.

OK, also Sie lesen csv zu einem Pandas Datenrahmen (df)

import csv 
import pandas as pd 

df = pd.read_csv("FL_insurance_sample.csv") 

Erstellen Sie dann ein neues df auf Ihre Bedürfnisse (Ihrerseits A nehmen hier)

>>> part_A = df.filter(['policyID', 'statecode', 'county', 'eq_site_limit', 'hu_site_limit'], axis=1) 

>>> part_A 
    policyID statecode  county eq_site_limit hu_site_limit 
0  NaN  NaN   NaN   NaN   NaN 
1 119736.0  FL CLAY COUNTY  498960.0  498960.00 
2 448094.0  FL CLAY COUNTY  1322376.3  1322376.30 
3 206893.0  FL CLAY COUNTY  190724.4  190724.40 
4 333743.0  FL CLAY COUNTY   0.0  79520.76 
5 172534.0  FL CLAY COUNTY   0.0  254281.50 

senden TEIL_A df Daten

>>> part_A.to_csv("part_A.csv", index=False, encoding='utf-8') 

erstellen csv Ebenso eine neue df für part_B

>>> part_B = df.filter(['fl_site_limit', 'fr_site_limit', 'tiv_2011', 'tiv_2012', 'eq_site_deductible', 'hu_site_deductible', 'fl_site_deductible', 'fr_site_deductible', 'point_latitude', 'point_longitude', 'line', 'construction', 'point_granularity'], axis=1) 

Dann senden Sie part_B df nach csv.

>>> part_B.to_csv("part_B.csv", index=False, encoding='utf-8') 

So können Sie Spalten aufgeteilt csv basierend auf Ihren Bedürfnissen und senden.

Answer 2

Um eine Liste der Spalten in eine CSV-Datei, verwenden Sie die Funktion to_csv() zu schreiben:

df = pd.read_csv("FL_insurance_sample.csv") 

df.iloc[:,:5].to_csv("data1.csv") 
df.iloc[:,5:].to_csv("data2.csv") 

Wenn Sie eine Liste der Spalten übergeben möchten direkt:

df[a].to_csv("data1.csv") 
df[b].to_csv("data2.csv")