Change information in a CSV file using info from the first one in python -


i'm trying edit csv file using informations first one. doesn't seem simple me should filter multiple things. let's explain problem.

i have 2 csv files, let's patch.csv , origin.csv. output csv file should have same pattern origin.csv, corrected values.

i want replace trip_headsign column fields in origin.csv using forward_line_name column in patch.csv if direction_id field in origin.csv row 0, or using backward_line_name if direction_id 1.

i want if part of line_id value in patch.csv between ":" , ":" symbols same part of route_id value in origin.csv before ":" symbol.

i know how replace whole line, not parts, have part of value.

here sample of origin.csv:

route_id,service_id,trip_id,trip_headsign,direction_id,block_id  210210109:001,2913,70405957139549,70405957,0, 210210109:001,2916,70405961139553,70405961,1,

and sample of patch.csv:

line_id,line_code,line_name,forward_line_name,forward_direction,backward_line_name,backward_direction,line_color,line_sort,network_id,commercial_mode_id,contributor_id,geometry_id,line_opening_time,line_closing_time  oif:100110010:10oif439,10,boulogne pont de saint-cloud - gare d'austerlitz,boulogne / pont de st cloud - gare d'austerlitz,oif:sa:8754700,gare d'austerlitz - boulogne / pont de st cloud,oif:sa:59400,dfb039,91,oif:439,metro,oif,geometry:line:100110010:10,05:30:00,25:47:00 oif:210210109:001oif30,001,ffourches longueville provins,place mérot - gare de longueville,,gare de longueville - place mérot,oif:sa:63:49,000000   1,oif:30,bus,oif,,05:39:00,19:50:00

each file has hundred of lines need parse , edit way.

based on mhopeng answer, obtained code:

#!/usr/bin/env python2 __future__ import print_function import fileinput import sys  # first route info patch.csv f = open(sys.argv[1]) d = open(sys.argv[2]) # ignore header line #line1 = f.readline() #line2 = d.readline() # line of data line1 in f.readline():     line1 = f.readline().split(',')     route_id = line1[0].split(':')[1] # '210210109'     route_forward = line1[3]     route_backward = line1[5]     line_code = line1[1]  # process origin.csv , replace lines in-place     line in fileinput.input(sys.argv[2], inplace=1):         line2 = d.readline().split(',')         num_route = line2[0].split(':')[0] # prevent lines same route_id different code considered same line          if line.startswith(route_id) , (num_route == line_code):         if line.startswith(route_id):             newline = line.split(',')             if newline[4] == 0:                 newline[3] = route_backward             else:                 newline[3] = route_forward             print('\t'.join(newline),end="")         else:             print(line,end="")

but unfortunately, doesn't push right forward or backward_line_name in trip_headsign (always forward), , triggers error, before finishing parsing file:

traceback (most recent call last): file "./gtfs_enhancer_headsigns.py", line 28, in if newline[4] == 0: indexerror: list index out of range

thanks on this.

pandas convenient handling csv files. use this:

import pandas pd  origin = pd.read_csv('origin.csv',index_col=none) patch  = pd.read_csv('patch.csv', index_col=none)  # create match_keys matching origin.csv patch.line_id  patch['match_key'] = [x.split(':')[1] x in patch.line_id.values] origin['match_key'] = [x.split(':')[0] x in origin.route_id.values]  i,key in enumerate(origin.match_key.values):     p = patch[patch.match_key == key]     if len(p) == 1:         if (origin.direction_id[i] == 0):             origin.trip_headsign[i] = p.forward_line_name.values[0]         elif (origin.direction_id[i] == 1):             origin.trip_headsign[i] = p.backward_line_name.values[0]  origin.to_csv('new_origin.csv',index=false)

Comments

Post a Comment

Popular posts from this blog

java - pagination of xlsx file to XSSFworkbook using apache POI -

right in code, reading xlsx file, xssfworkbook, , writing database. but, when size of xlsx file increases, causes outofmemory error. can not increase server size, or divide xlsx file pieces. tried loading workbook using file (instead of inputstream), didn't either. i looking way read 10k rows @ time (instead of entire file @ once) , iteratively write workbook , database. is there way apache poi? poi contains called "eventmodel" designed purpose. it's mentioned in faq : the ss eventmodel package api reading excel files without loading whole spreadsheet memory. require more knowledge on part of user, reduces memory consumption more tenfold. based on awt event model in combination sax. if need read-only access, best way it. however, may want double check first if issue somewhere else. check out this item : i think poi using memory! can do? 1 comes quite lot, reason isn't might think. so, first thing check - what's source of prob...
Read more

Unlimited choices in BASH case statement -

consider simple bash snippet: case $option in 1) image=${options[0]%.tar} ;; 2) image=${options[1]%.tar} ;; 3) image=${options[2]%.tar} ;; 4) image=${options[3]%.tar} ;; *) echo "invalid option" exit 1 esac in real script numbers goes 30. makes pretty long. can somehow specify cases variable? something this: case $option in $i) image=${options[$(($i-1))]%.tar} any pointer appreciated. you can match against multiple patterns in single clause: case $option in 1|2|3: echo "$option one, 2 or three" ;; esac if still typing you, can use simple pattern matching: case $option in # following matches # - single-digit numbers 0-9 # - two-digit numbers starting either 1 or 2 # - number 30 [0-9]|[12][0-9]|30) image=${options[$(($option-1))]%.tar} ;; *) echo "invalid option" 1>&2 exit 1 esac
Read more

apache - How do I stop my index.php being run twice for every user -

we've got webapp initial html served using apache , php. i noticed index.php being run twice every request. seems caused / effected rewrite rules in our .htaccess - rewritecond %{http_host} ^(www\.|hotels\.)ourdomain\.com$ [nc,or] rewritecond %{server_name} ^(www\.|hotels\.)ourdomain\.com$ [nc] rewriterule (.*) http://ourdomain.com/$1 [r=301] rewriterule ^hotels/([^/]+)/?\??(.*)$ ?d=$1&$2 [qsa] the final rule moving parameter url path query string. i.e. http://ourdomain.com/hotels/vegas?someparam=1 becomes http://ourdomain.com?d=vegas&someparam=1 if go directly query string version index.php run once. if use url redirected, index.php run twice (i'm checking adding error_log('end of index.php') file). so example going http://ourdomain.com/hotels/paris hits file twice http://ourdomain.com?d=paris hits once. i've seen this question , had @ blog mentioned , , can't find empty string url's (i've tried using yslow process). ...
Read more