Python Pandas : column with ordered file names based on given sequence

Question

我想在数据框中创建一个列，其中包含基于特定序列号的文件名列表。

我的主数据框如下所示:

  num letter 
0    0    I   
1    1    P  
2    2    I  
2    3    B  
2    4    P  
2    5    B  
2    6    I  

我的文件名是:I-0001.jpg、I-0002.jpg、I-0003.jpg、P-0001.jpg、P-0002.jpg、B-0001.jpg、B -0002.jpg

我想要的最终结果是:

   num  letter  file
0    0    I   I-0001.jpg
1    1    P   P-0001.jpg
2    2    I   I-0002.jpg
2    3    B   B-0001.jpg
2    4    P   P-0002.jpg
2    5    B   B-0002.jpg
2    6    I   I-0003.jpg

所以基本上，列文件应该是这样的，只要在“字母”列中找到 I、P 或 B，它就会根据中的编号选择文件名中包含 I、P 或 B 的文件名每个文件。

有什么帮助吗？

Answer 1

您可以将letter列与GroupBy.cumcount创建的Series连接起来对于每个字母的计数器，添加1，转换为字符串，如有必要，添加str.zfill :

df['file'] = (df['letter'] + '-' +
              df.groupby('letter').cumcount().add(1).astype(str).str.zfill(4) + '.jpg')
print (df)
   num letter        file
0    0      I  I-0001.jpg
1    1      P  P-0001.jpg
2    2      I  I-0002.jpg
2    3      B  B-0001.jpg
2    4      P  P-0002.jpg
2    5      B  B-0002.jpg
2    6      I  I-0003.jpg

详细信息:

print (df.groupby('letter').cumcount())
0    0
1    0
2    1
2    0
2    1
2    1
2    2
dtype: int64

print (df.groupby('letter').cumcount().add(1).astype(str).str.zfill(4))
0    0001
1    0001
2    0002
2    0001
2    0002
2    0002
2    0003
dtype: object