Python一题多解学思路：指定列前置

前言

熟读 Python 基本数据结构，但实际应用时却大脑短路，无法解决问题？

熟悉 pandas ，但遇到没有内置的方法，就不知所措？

如果你有这种情况，那么你缺少的是解决问题的思路。

经常向我提问的同学应该知道，我一般不会直接给出代码，而是给你提供思路。本系列主打思路，基于同一思路，给出多种不同的解决方案，让你举一反三解决问题。

问题

有位小伙伴提了一个简单的问题，数据如下：

• 我希望把列"key1,2,3"调整到表头的最左边，其他的列顺序我不关心
• 别让我把所有列名写出来，因为我有些表可能有几十上百的列

前提

我们需要知道一些前提知识。

pandas 中怎么指定列顺序？非常简单：

• 往 df[] 中指定多个列名的 list 即可
• 显然，提问者就是不希望手工输入所有的列表

pandas 中怎么获取表的所有列名：

• df.columns 即可。
• 如果希望得到列表，只需要 list(df.columns) 即可

如何拼接 Python 中的2个列表：

理解这两点后，现在的问题其实不是什么 pandas 或 啥表头的顺序问题，而是一个列表构造问题：

• 行4：是解决问题的关键

接下来，基于这些前提条件和思路，看看能做出多少种解法

解法1

这是最直接的方式：

• 行4-7：遍历所有列，把非目标列名去掉即可

如果你不熟悉推导式，也可以用 for in 遍历

"我不懂啥推导式呀，遍历也用得不熟悉，有没有不用遍历的方式？"

解法2

"Python 不是一直声称语法简单吗，为什么不可以把 2 个列表相减，就得到剩余列名？"

这想法还挺有道理的，为什么不呢？

因为列表中的元素是可以重复的，如果 2 个列表能相减，他就要考虑有重复元素与没有重复元素的歧义。

因此，列表是不能相减，但是 Python 中的确有一种东西与列表相似，但他里面的元素不能重复—— set(集)

由此我们可以得到第二种解法：

• 行4：用 set 函数得到 set 结构，即可相减。
• 为什么需要在外面再套一个 list 呢？因为下一步我们需要使用 "列表相加" 的骚操作

细心的你可能注意到，出来的结果中，其他列的顺序与解法1不一样！

因为 set 是不关心元素顺序的，2 个 set 相减后，顺序变化了

"有没有不同遍历，有可以保留其他列的原始前后顺序的？"

解法3

前面的解法，本质上都在做去重工作。pandas 中不也有去重功能吗，我们也可以用上。

• 行4：pd.Series 中传入有重复元素的列表，就能返回一个 Series。使用他的去重方法即可完成

"有问题啊，new_cols 是列表？，不应该是一个 Series 吗？为什么 df[一个Series] 也能改变列顺序？"

实际上，df[] 里面可以接受大多数序列，比如 set 或 numpy 的数组等等，唯独元组不行。

这涉及到多层索引的东西，超纲了，可参考：

思考

上面 3 种解法都很好解决了问题，并且解法1和3还能保证其他列的顺序不变。

但是，如果目标列 key1,2,3 可能会出现缺失，比如某个表只有 key1 和 key2，那么今天的 3 种方法都不能执行。

不妨用今天教你的思路，想想怎么解决这个难题

公众号回复"数据处理"获取源码