我有一个数字列表，我想创建一个文本文件（使用python），其名称按特定顺序列出（对于使用mencoder创建的电影）。特别是在这里，图形名称包括月份（4月，8月......）。我想要Januray的第一个，然后是二月等等。
我知道我可以用丑陋的方式做到这一点，但我感兴趣的是一个既优雅（= pythonic？）又最终更通用的解决方案。
我的文件按自然顺序排列：
cld_for_April_EISopt_1000.png
cld_for_August_EISopt_1000.png
cld_for_December_EISopt_1000.png
cld_for_February_EISopt_1000.png
cld_for_January_EISopt_1000.png
cld_for_July_EISopt_1000.png
cld_for_June_EISopt_1000.png
cld_for_March_EISopt_1000.png
cld_for_May_EISopt_1000.png
cld_for_November_EISopt_1000.png
cld_for_October_EISopt_1000.png
cld_for_September_EISopt_1000.png
我希望在里面有一个文本文件：
cld_for_January_EISopt_1000.png
cld_for_February_EISopt_1000.png
cld_for_March_EISopt_1000.png
cld_for_April_EISopt_1000.png
cld_for_May_EISopt_1000.png
cld_for_June_EISopt_1000.png
cld_for_July_EISopt_1000.png
cld_for_August_EISopt_1000.png
cld_for_September_EISopt_1000.png
cld_for_October_EISopt_1000.png
cld_for_November_EISopt_1000.png
cld_for_December_EISopt_1000.png
或者更一般地说，如果我有一个列表或数组或类似的词典：
{'pattern1'：rank_in_output_list_1，...，'pattern12'：rank_in_output_list_12}
我该如何使用它来命令我的文件名？
到目前为止，我玩过：os.listdir，os.path.isfile，numpyp.ma.array，.compressed（）或.compress（）;但我没那么成功。
非常感谢。
克里斯托弗。
答案 0 :(得分：6)
作为排序的键，按_拆分并映射第三个元素。
sorted(filenames, key=lambda x: monthdict[x.split('_')[2]])
答案 1 :(得分：1)
订购序列的答案更长，但更灵活：
import collections
def rearrange(seq, order, keyfunc):
if not isinstance(order, collections.Mapping):
order = {v: i for i,v in enumerate(order)}
return sorted(seq, key=lambda x: order[keyfunc(x)])
if __name__ == '__main__':
filenames = """
cld_for_April_EISopt_1000.png cld_for_August_EISopt_1000.png
cld_for_December_EISopt_1000.png cld_for_February_EISopt_1000.png
cld_for_January_EISopt_1000.png cld_for_July_EISopt_1000.png
cld_for_June_EISopt_1000.png cld_for_March_EISopt_1000.png
cld_for_May_EISopt_1000.png cld_for_November_EISopt_1000.png
cld_for_October_EISopt_1000.png cld_for_September_EISopt_1000.png
""".split()
months = ['January', 'February', 'March', 'April', 'May', 'June', 'July',
'August', 'September', 'October', 'November', 'December']
def get_month_name(filename):
return filename.split('_')[2]
for filename in rearrange(filenames, months, get_month_name):
print(filename)
输出：
cld_for_January_EISopt_1000.png
cld_for_February_EISopt_1000.png
cld_for_March_EISopt_1000.png
cld_for_April_EISopt_1000.png
...
答案 2 :(得分：0)
这是我的最终版本。
我希望它更加pythonic。
请随意评论一般风格。
#!/usr/bin/env python
"""Demonstrate how to animate multiple figures into a climatology movie."""
#
# Imports
#
from os import system
from os.path import exists
from numpy import ma
from glob import glob
from copy import deepcopy
from logging import warning
#
# Parameters
#
month_list = ['January',
'February',
'March',
'April',
'May',
'June',
'July',
'August',
'September',
'October',
'November',
'December']
#
month_dict = {'January': 0,
'February': 1,
'March': 2,
'April': 3,
'May': 4,
'June': 5,
'July': 6,
'August': 7,
'September': 8,
'October': 9,
'November': 10,
'December': 11}
#
param_dict_default = {'fig_folder': 'figures/era40_correlations/',
'output_name': 'output_sol_2.mp4',
'fig_criteria': '*climatology*isccp*.png',
'order': month_list,
'input_file': 'list_fig_for_movie.txt',
'splitor': '_',
'fps': 10}
#
# Functions
#
def reorder(seq, extractor, order):
"Reorder 'seq' using 'extractor' and according to 'order'."""
rank = dict((v,i) for i,v in enumerate(order))
return sorted(seq, key=lambda v: rank.get(extractor(v),-1))
def get_ordering_key(filename, splitor, ind):
"""Get ordering key in 'filename' at 'ind' position for 'splitor'."""
return filename.split(splitor)[ind]
def prepareMovie(param_dict={}):
"""Return the command line to create the movie and the output filename.
Input
-----
param_dict : dictionary with parameters for creating the movie.
Keys: Default values:
* fig_criteria = '*climatology*isccp*.png'
* input_file = 'list_fig_for_movie.txt'
* splitor = '_'
* fps = 10
* fig_folder = 'figures/era40_correlations/'
* output_name = 'output_sol_2.mp4'
* order = ['January', 'February', 'March', 'April',
'May', 'June', 'July', 'August', 'September',
'October', 'November', 'December']
from:
for key in param_dict_default.keys():
print ' * ', key, '=', repr(param_dict_default[key])
Output
------
Command line to create the movie and output filename.
"""
# Update parameters
param_new = deepcopy(param_dict_default)
param_new.update(param_dict)
# List all selected figures
file_names = glob(param_new['fig_folder'] + param_new['fig_criteria'])
# Find position of ordering key in name
for item in param_new['order']:
try:
ind = file_names[0].split(param_new['splitor']).index(item)
except:
pass
# Sort all selected figures
if isinstance(param_new['order'], list):
file_names = sorted(file_names, key=lambda x: param_new['order'].\
index(x.split(param_new['splitor'])[ind]))
elif isinstance(param_new['order'], dict):
file_names = sorted(file_names, key=lambda x: param_new['order']\
[x.split(param_new['splitor'])[ind]])
else:
raise ValueError("param_dict['order'] must be list or dictionary.")
# Remove input file for mencoder if already exists
if exists(param_new['input_file']):
warning(" '%s' existed and has been overwritten."
%param_new['input_file'])
system("rm %s" %param_new['input_file'])
# Write input file for mencoder
f = open(param_new['input_file'], "w")
for item in file_names:
f.write(item+'\n')
f.close()
# Create command for mencoder
command = ['mencoder',
'mf://@' + param_new['input_file'],
'-mf',
"type=png:w=800:h=600:fps=%s" %param_new['fps'],
'-ovc',
'lavc',
'-lavcopts',
'vcodec=mpeg4',
'-oac',
'copy',
'-o',
param_new['output_name']]
# Make it one line white-spaced and return it
command = ''.join([item + ' ' for item in command])
return command, param_new['output_name']
#
# Main
#
if __name__ == "__main__":
[command, output_name] = prepareMovie()
# Create the movie
system(command)
# Open it
system("mplayer %s" %output_name)