我有多個具有相同后綴 ( _01_020301.txt) 和不同數字前綴的檔案。目標是按連續前綴將它們分組到串列中。
這是一個例子：
09456_01_020301.txt
09457_01_020301.txt
09458_01_020301.txt
09459_01_020301.txt
09460_01_020301.txt
09465_01_020301.txt
09466_01_020301.txt
09467_01_020301.txt
09468_01_020301.txt
因為 09456~09460 和 09465~09468 是兩個連續的組，所以結果應該是：
[['09456_01_020301.txt',
'09457_01_020301.txt',
'09458_01_020301.txt',
'09459_01_020301.txt',
'09460_01_020301.txt'],
['09465_01_020301.txt',
'09466_01_020301.txt',
'09467_01_020301.txt',
'09468_01_020301.txt']]
uj5u.com熱心網友回復：
這是使用簡單回圈的一種方法：
l = ['09456_01_020301.txt',
'09457_01_020301.txt',
'09458_01_020301.txt',
'09459_01_020301.txt',
'09460_01_020301.txt',
'09465_01_020301.txt',
'09466_01_020301.txt',
'09467_01_020301.txt',
'09468_01_020301.txt']
out = []
prev = float('-inf')
for i in l:
prefix = int(i.partition('_')[0])
if prefix == prev 1:
out[-1].append(i)
else:
out.append([i])
prev = prefix
輸出：
[['09456_01_020301.txt',
'09457_01_020301.txt',
'09458_01_020301.txt',
'09459_01_020301.txt',
'09460_01_020301.txt'],
['09465_01_020301.txt',
'09466_01_020301.txt',
'09467_01_020301.txt',
'09468_01_020301.txt']]
uj5u.com熱心網友回復：
假設輸入是一個 str（例如，.csv檔案），我通過比較以前的后綴和當前后綴來創建以下代碼。
input = """09456_01_020301.txt
09457_01_020301.txt
09458_01_020301.txt
09459_01_020301.txt
09460_01_020301.txt
09465_01_020301.txt
09466_01_020301.txt
09467_01_020301.txt
09468_01_020301.txt"""
result = [] # for the final result
tmp_list = [] # for a temporary sub_list for consecutie suffix files
lines = input.split('\n')
tmp_list.append(lines[0])
prev_suffix = int(lines[0][:5].lstrip('0')) # lstrip('0') to remove leading zero (for example, 09456 to 9456)
for line in lines[1:]:
current_suffix = int(line[:5].lstrip('0'))
if current_suffix == (prev_suffix 1): # 9457 == 0946 1 ?
tmp_list.append(line)
else:
result.append(tmp_list)
tmp_list = [line]
prev_suffix = current_suffix
result.append(tmp_list)
print(result)
# [['09456_01_020301.txt', '09457_01_020301.txt', '09458_01_020301.txt', '09459_01_020301.txt', '09460_01_020301.txt'], ['09465_01_020301.txt', '09466_01_020301.txt', '09467_01_020301.txt', '09468_01_020301.txt']]
uj5u.com熱心網友回復：
另一種解決方案，使用itertools.groupby：
from itertools import groupby
lst = [
"09456_01_020301.txt",
"09457_01_020301.txt",
"09458_01_020301.txt",
"09459_01_020301.txt",
"09460_01_020301.txt",
"09465_01_020301.txt",
"09466_01_020301.txt",
"09467_01_020301.txt",
"09468_01_020301.txt",
]
out = []
for _, g in groupby(enumerate(lst), lambda k: int(k[1].split("_")[0]) - k[0]):
out.append([v for _, v in g])
print(out)
印刷：
[
[
"09456_01_020301.txt",
"09457_01_020301.txt",
"09458_01_020301.txt",
"09459_01_020301.txt",
"09460_01_020301.txt",
],
[
"09465_01_020301.txt",
"09466_01_020301.txt",
"09467_01_020301.txt",
"09468_01_020301.txt",
],
]
轉載請註明出處，本文鏈接：https://www.uj5u.com/qianduan/466027.html