python - 修改函数输出并保存列表

Question

我正在尝试从大约 1000 个 csv 文件名中返回单元编号列表。我可以读入它们，然后让 python 删除它们周围的所有垃圾，并替换第五个字符以按照我需要的方式对其进行格式化。我想返回所有单元号的列表，例如 ['6726-0501', '6826-1144']。我目前得到的是它一一打印出单元号并且不保存它们。我已经查看了之前的问题，但似乎无法获得创建列表然后将单元编号附加到列表并将该列表保存到变量中的模式以进行工作。有谁知道简单修改它以输出列表并保存列表以供以后使用的好方法？

谢谢

罗宾

file_names = ['job_1106_unit_672600501_las_PN23074.LAS.csv', 'job_1108_unit_682601144_las_PN23072.LAS.csv']


def change(file_names):
    for comps in file_names:
         comps_of_comps = list(comps)
         unit_num = comps_of_comps[14:23] #[672600501]
         a = (unit_num[0:4])  #[6726]
         b = (unit_num[5:9])   #[0501]
         unit_num = a + list('-') + b  #[6,7,2,6,-,0,5,0,1]
         unit_num = ''.join(unit_num) #6726-0501
         print unit_num

change(file_names)

Answer 1

您可以初始化一个新列表并向其追加并返回该列表。喜欢

file_names = ['job_1106_unit_672600501_las_PN23074.LAS.csv', 'job_1108_unit_682601144_las_PN23072.LAS.csv']
def change(file_names):
    result = []
    for comps in file_names:
         comps_of_comps = list(comps)
         unit_num = comps_of_comps[14:23] #[672600501]
         a = (unit_num[0:4])  #[6726]
         b = (unit_num[5:9])   #[0501]
         unit_num = a + list('-') + b  #[6,7,2,6,-,0,5,0,1]
         unit_num = ''.join(unit_num) #6726-0501
         result.append(unit_num)
    return result

print change(file_names)

或者

import re

def change(file_names):
    result = []
    for i in file_names:
        s = re.match('.*unit_(.*)_las.*', i).group(1)
        result.append(s[:len(s)/2]+"-"+s[(len(s)/2)+1:])
    return result