python - 使用字符串的特定部分。 python中某些条件之前和之后的部分

Question

我有一组完整路径文件，例如:

a=['/dir1/europe_2013_info.csv', '/dir2/USA_2013_info.csv', '/dir3/Africa_2013_info.csv']

我想创建一个新文件并将此特征写入该文件中。例如:

f=open('all.csv','w')
f.write('port,europe\n')
f.close

f=open('all.csv','w')
f.write('port,USA\n')
f.close

等等。我对 python 相当陌生。有人可以帮助我如何创建这样的循环吗？

Answer 1

您不需要多次打开和关闭文件。

使用正则表达式 /(\w+)_\d{4} (或更明确的 /(\w+)_\d{4}_info\.csv ) 以提取列表中文件路径的所需部分和 csv.writer写入 csv 文件:

import re
import csv

a = ['/dir1/europe_2013_info.csv', '/dir2/USA_2013_info.csv', '/dir3/Africa_2013_info.csv']

regexp = '/(\w+)_\d{4}'
with open('all.csv', 'w') as f:
    writer = csv.writer(f)
    writer.writerows([['port', re.search(regexp, i).group(1)] for i in a])

那么，all.csv的内容将是:

port,europe
port,USA
port,Africa