[英]Getting values from pandas dataframe in a particular order
按照 mylist 的順序從下面的 dataframe 創建標簽列表的最有效方法是什么?
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
只需使用sort_values
:
>>> df.sort_values('filenames')
labels filenames
1 2 a1jpeg
0 1 a2jpeg
4 2 b1jpeg
3 1 b2jpeg
5 2 c1jpeg
2 1 c2jpeg
>>>
要轉換為列表:
>>> df['filenames'].sort_values().tolist()
['a1jpeg', 'a2jpeg', 'b1jpeg', 'b2jpeg', 'c1jpeg', 'c2jpeg']
>>>
首先使用Series.replace
,然后通過DataFrame.set_index
和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']
聲明:本站的技術帖子網頁,遵循CC BY-SA 4.0協議,如果您需要轉載,請注明本站網址或者原文地址。任何問題請咨詢:yoyou2525@163.com.