Положение элемента в списке в рамках Pandas DataFrame - PullRequest
0 голосов
/ 25 ноября 2018

новичок здесь

У меня есть Pandas DataFrame, который содержит в себе столбец экипажа, как показано ниже.Похоже, что каждая строка является списком.

    crewjob
    [Director, Screenplay, Screenplay, Screenplay,...
    [Executive Producer, Screenplay, Original Musi...
    [Director, Characters, Writer, Sound Recordist]
    [Director, Screenplay, Producer, Producer, Pro...
    [Original Music Composer, Director of Photogra...
    [Director, Screenplay, Producer, Producer, Ori...
    [Director, Screenplay, Producer, Original Musi...
    [Screenplay, Screenplay, Director, Novel]
    [Director, Screenplay, Screenplay, Producer, P...

Я хотел бы извлечь для каждой строки позицию директора в списке.Я попробовал метод перечисления, и он выбрасывает пустой список.

   indices = [i for i, x in enumerate(creditsdf['crewjob']) if x == "Director"]

Я также попробовал метод .index

   creditsdf['test'] = creditsdf['crewjob'].index("Director")

, который дает следующую ошибку;

TypeError: 'RangeIndex' object is not callable

1 Ответ

0 голосов
/ 25 ноября 2018

Использовать понимание вложенного списка:

indices = [[i for i, x in enumerate(y) if x == "Director"] for y in increditsdf['crewjob']]
print (indices)
[[0], [], [0], [0], [], [0], [0], [2], [0]]
Добро пожаловать на сайт PullRequest, где вы можете задавать вопросы и получать ответы от других членов сообщества.
...