Разбиение списка на Python

Question

Я хотел бы узнать, как изменить содержимое списка, в частности, имена файлов, которые были возвращены os.listdir().

Имена файлов состоят из нескольких дубликатов имен, за которыми следует символ подчеркивания, затем суффикс, а затем расширение файла. Я пытаюсь вырезать элементы в списке вниз только первую часть имени файла, перед '_', так:

apple_d.jpg 
apple_si.jpg 
apple_sg.jpg 

становится только одну запись в списке, 'apple'.

Я могу избавиться от дубликатов и повторной alphabetise с

list(sorted(set(t))) 

но избавившись от всего, от подчёркивания года оказывается сложнее. Я попытался сделать это через .rpartition("_")[0]. но это, по-видимому, не работает для списков. Так что мне интересно, как это сделать?

Редактировать: хорошо, не работает здесь. Все еще получаются дубликаты и не раскалываются.

objects = os.listdir(dir) 





    for object in objects: 
     sorted(set(object.split('_', 1)[0])) 
     cmds.menuItem(label = object, parent = "objectMenu") 

(Последняя команда - команда майя, которая заполняет меню опций). Очень устал, поэтому мне придется это сделать позже. Но большое спасибо за помощь на сегодняшний день. Конечно, скоро это получится.

Answer 1

Использование str.split() или str.rsplit() с лимитом, а затем выбрать первый элемент:

filename.split('_', 1)[0] 

.rsplit('_', 1) разделится на последнем подчеркиванием, .split() на первом. Выберите тот, который лучше всего подходит вам.

Это дает вам все до первого или последнего подчеркивания для этого имени файла.

Используя это в комплекте понимания, с sorted() возвращает список из этого набора:

unique_prefixes = sorted({filename.split('_', 1)[0] for filename in os.listdir(somedir)}) 

Для Python 2.6 и более ранних версий, где вы еще не имеют синтаксис набор понимания, следующее выражение генератора с set() будет работать тоже:

unique_prefixes = sorted(set(filename.split('_', 1)[0] for filename in os.listdir(somedir))) 

Answer 2

из моего понимания вопроса, вы пытались позвонить rpartition непосредственно в списке (возможно, как os.listdir('path').rparition('_')[0]). Я считаю, что использовать rpartition так, как вы планируете, вам нужно применить его к отдельным элементам списка. Список понимание является наиболее рекомендуемый способ сделать это:

file_prefixes = [f.rpartition('_')[0] for f in os.listdir('path')] 

Превращение его в набор понимания как Martijin Питерс предлагает исключит дубликаты:

file_prefixes = {f.rpartition('_')[0] for f in os.listdir('path')} 

Примечание: Это должно быть на самом деле эквивалентно Martijin Питерс 'ответ, просто используя rpartition, как упоминалось в вопросе, а не rsplit. (В основном сохраняет параметр 1.)

Answer 3

Это работало для меня guys-

labels = os.listdir(dir) 




labels_processed=[] 
for label in labels: 
    labels_processed.append(label.split('_')[0]) 



labprocessed = set(labels_processed) 



for finallab in labprocessed:   
    cmds.menuItem(label = finallab, parent = "objectMenu") 

Приветствия,

S Тозера