python - 根据元素位置从另一列中提取某些元素

Question

我在 DataFrame 中有两列，crewname 是制作电影的工作人员列表。 Director_loc 是导演列表中的位置。

我想创建一个包含导演姓名的新专栏。

    crewname                                           Director_loc

    [John Lasseter, Joss Whedon, Andrew Stanton, J...  0
    [Larry J. Franco, Jonathan Hensleigh, James Ho...  3
    [Howard Deutch, Mark Steven Johnson, Mark Stev...  0
    [Forest Whitaker, Ronald Bass, Ronald Bass, Ez...  0
    [Alan Silvestri, Elliot Davis, Nancy Meyers, N...  5
    [Michael Mann, Michael Mann, Art Linson, Micha...  0
    [Sydney Pollack, Barbara Benedek, Sydney Polla...  0
    [David Loughery, Stephen Sommers, Peter Hewitt...  2
    [Peter Hyams, Karen Elise Baldwin, Gene Quinta...  0
    [Martin Campbell, Ian Fleming, Jeffrey Caine, ...  0

我已经尝试了一些使用列表理解、枚举等的代码。我有点不好意思把它们放在这里。

任何帮助将不胜感激。

Answer 1

使用带有列表理解的索引:

df['name'] = [a[b] for a , b in zip(df['crewname'], df['Director_loc'])]
print (df)
                                         crewname  Director_loc  \
0  [John Lasseter,  Joss Whedon,  Andrew Stanton]             2   
1          [Larry J. Franco,  Jonathan Hensleigh]             1   

                  name  
0       Andrew Stanton  
1   Jonathan Hensleigh