如何在python中打印字符串分割的第一部分

2024-10-03 04:33:51 发布

您现在位置:Python中文网/ 问答频道 /正文

我的csv数据如下:

 Minneapolis-St. Paul-Bloomington   MN-WI--33460
Charlotte-Gastonia-Rock Hill     NC-SC--16740
Las Vegas-Paradise   NV--29820
New York-White Plains-Wayne  NY-NJ  (MSAD)--35620

我们如何才能只打印列中之前的数据?在

^{pr2}$

代码:

   #!/usr/bin/python
   import csv
   import re

   sample="""columnA,ColumnB,columnC
   1,Cleveland-Elyria-Mentor    OH--17460
   2,Dallas-Plano-Irving    TX  (MSAD)--19100
   3,ASJDFJKDJ-kD-JE       WA--21092"""


   open('sample.csv','w').write(sample)
   with open('sample.csv') as sample, open('final_output.csv','w') as output:
   reader = csv.reader(sample)
   writer = csv.writer(output)
   # discard input header
   next(reader)
   # write output header
   writer.writerow(['col1','col2','col3'])
   #process rows
   for row in reader:
        if row:
              for fipsplit in row[1].split('\n'):
               fipsval = re.findall(r'--(\d+)$', fipsplit)
               writer.writerow([fipsplit])
    print open('final_output.csv').read()

Tags: csv数据sampleimportreoutputasopen