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

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

1.问题背景

2.解决方法

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

2.2.构建Series数据

2.3.将排序的test.txt读取进来

2.4.将Series转换成DataFrame

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

---

1.问题背景

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

        但是问题来了，在test这个文件夹中图片的顺序和test.txt的顺序居然不一样！，并且就算是一样的，在使用keras的flow_from_directory函数批量读取图片的时候，顺序也会发生变化！

        因此，必须实现一种方法，将预测结果按照test.txt的顺序进行排列。一张图概括如下：

2.解决方法

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

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

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

2.2.构建Series数据

s = pd.Series(my_dict)

2.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']

2.4.将Series转换成DataFrame

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

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

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