Python：使用pandas按某一给定的txt进行排序

最近参加了百度的一个深度学习方面的图像分类比赛，在结果提交的过程中，要求按照所给的TXT进行排序,然后提交结果:

但是问题来了，在test这个文件夹中图片的顺序和test.txt的顺序居然不一样！，并且就算是一样的，在使用keras的flow_from_directory函数批量读取图片的时候，顺序也会发生变化！因此，必须实现一种方法，将预测结果按照test.txt的顺序进行排列。一张图概括如下：

思考了一下，pandas应该可以解决这个问题，查询了一下用法，最后汇总如下：

1.将index和values通过字典捆绑在一起：

my_dict = {'a':1,'b':2,'c':3})

2.构建Series数据：

s = pd.Series(my_dict)

3.将排序的test.txt读取进来：

with open('test.txt', 'r') as fread: 	# 打开文件，读取模式
    line = fread.readlines()              	# 一次读取所有行

    for i in line:
        list_custom.append(i.strip())

这时，我们得到了一个list:

list_custom = ['b', 'a', 'c']

4.将Series转换成DataFrame

df = pd.DataFrame(s)
df = df.reset_index()
df.columns = ['fileName', 'label']

5.设置成“category”数据类型并按照list排序：

df['fileName'] = df['fileName'].astype('category')
df['fileName'].cat.reorder_categories(list_custom, inplace=True)
df.sort_values('fileName', inplace=True)