[Tutor] trying to split or rpartition the contents of a list

[Amit Saha]

Hello,

On Wed, May 22, 2013 at 5:49 PM, Stuart Tozer <stutozer at gmail.com> wrote:
> Hi everyone.
>
> I'm stuck on a problem while developing a small tool for Maya. Basically, I
> have a folder with some filenames in it which have been returned by
> os.listdir(). These filenames look something like...
>
>
> apple_d.jpg
> apple_si.jpg
> apple_sg.jpg
> box_d.jpg
> box_si.jpg
> pumpkin_d.jpg
>
>
> Right now, I'm using os.listdir to get the filenames, and then each filename
> becomes a selectable option in an option menu in maya. The code is...
>
> objects = os.listdir(dir)
>
>
>
>
>
>     for object in objects:
>         cmds.menuItem(label = object, parent = "objectMenu")
>
>
>
> My problem is that I don't want duplicates of the same object in the menu
> and I also want to truncate everything past and including the underscore. So
> for eg, if my filenames are the same as the example above, I want to change
> the list so I have this instead...
>
> apple
> box
> pumpkin
>
>
> I have tried splitting the filenames, but so far haven't had any luck.
>
> for object in objects:
>         sorted(set(object.split('_', 1)[0]))
>         cmds.menuItem(label = object, parent = "objectMenu")

You are almost there.

Something like this should work:

>>> labels = ['apple_d.jpg','apple_si.jpg','apple_sg.jpg','box_si.jpg']

Now, create a list of only the names (upto _):

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

Remove the duplicates:

>>> set(labels_processed)
set(['box', 'apple'])

If you have any doubts about this, please ask.

Best,
Amit.



--
http://echorand.me