时间戳

首页 > 解决方案 > 以特定顺序从熊猫数据框中获取值

python - 以特定顺序从熊猫数据框中获取值

问题描述

按 mylist 的顺序从下面的数据框中创建标签列表的最有效方法是什么？

import numpy as np
import pandas as pd    

mylist = ['a1.jpeg','a2.jpeg','b1.jpeg','b2.jpeg','c1.jpeg','c2.jpeg']
    
dat = np.array([(1, 2, 1, 1, 2, 2), ('a2jpeg', 'a1jpeg', 'c2jpeg', 'b2jpeg', 'b1jpeg' , 'c1jpeg')])
df = pd.DataFrame(np.transpose(dat), columns=['labels', 'filenames'])
df
>>labels    filenames
0   1   a2.jpeg
1   2   a1.jpeg
2   1   c2.jpeg
3   1   b2.jpeg
4   2   b1.jpeg
5   2   c1.jpeg

标签： pythonpandasnumpy

解决方案

首先使用Series.replace，然后通过DataFrame.set_indexwith更改顺序Series.reindex：

L = (df.assign(filenames = df['filenames'].replace('jpeg','.jpeg', regex=True))
       .set_index('filenames')['labels']
       .reindex(mylist)
       .tolist())
print (L)
['2', '1', '2', '1', '2', '1']

或者：

df['filenames'] = pd.Categorical(df['filenames'].replace('jpeg','.jpeg', regex=True), 
                                 ordered=True, 
                                 categories=mylist)

L = df.sort_values(by='filenames')['labels'].tolist()
print (L)
['2', '1', '2', '1', '2', '1']

如果可能，通过排序简化解决方案DataFrame.sort_values：

L = df.sort_values(by='filenames')['labels'].tolist()
print (L)
['2', '1', '2', '1', '2', '1']

推荐阅读