博客
关于我
强烈建议你试试无所不能的chatGPT,快点击我
list(map(list,zip(*a)))
阅读量:3929 次
发布时间:2019-05-23

本文共 1251 字,大约阅读时间需要 4 分钟。

在Python的numpy中,对类似array=[[1,2,3],[4,5,6],[7,8,9]];的二维数组的转置,就一句话array.T。其实不使用numpy,单纯使用Python,代码也不长,同样也是一行。不过在此之前,先说明Python中map函数与zip(*)的使用。

一、map函数

首先Python中的map函数是很简单的。意为将第二个参数(一般是数组)中的每一个项,处理为第一个参数的类型。比如如下的代码,将a这个list的每一项都从int类型转化为str类型。

 

 
  1. #-*-coding:utf-8-*-

  2. a=[1,2,3];

  3. print map(str,a);

运行结果如下:

而如下的数组,则对a这个二维数组中的每一个项的一维数组进行求和操作,最后得出来自然是一个一维数组,因为每一个项的一维数组,转化为一个int。

 

 
  1. #-*-coding:utf-8-*-

  2. a=[[1,3,4],[2,3,2]];

  3. print map(sum,a);

运行结果如下:

二、zip(*)

在《【Python】用zip函数求欧氏距离、余弦相似度》()中已经介绍了zip的使用,

 

比如下述的一段代码:

 

 
  1. #-*-coding:utf-8-*-

  2. x=[1,2,3];

  3. y=[4,5,6];

  4. z=[7,8,9];

  5. print zip(x,y,z);

运行结果如下:

 

意思是取各个list的第x项,作为返回的二维数组的第x项中的一维数组中的元素。

其实zip还是反过来会这个二维数组操作,但要注意写成zip(*),表示这是一个zip的逆操作。

比如下述的一段代码:

 

 
  1. # -*-coding:utf-8-*-

  2. array=[[1,4,7],[2,5,8],[3,6,9]];

  3. x,y,z=zip(*array);

  4. print x,y,z;

xyz的值分别如下所示,恰好与zip函数形成一个逆操作。

三、python中二维数组的转置

这与数组转置有什么相关呢?

注意到,如果zip(*array)不经过x,y,z=zip(*array);拆分成x,y,z三个变量,那么[[1,4,7],[2,5,8],[3,6,9]];被zip(*array)之后的结果恰好是[(1, 2, 3), (4, 5, 6), (7, 8, 9)],刚好形成一个转置的关系。这对于所有array=[[1,2,3],[4,5,6],[7,8,9]];的二维数组都是一样的,不信可以试试。

当然[(1, 2, 3), (4, 5, 6), (7, 8, 9)]还不是我们需要的最后的结果,

因为只是一个存放tuple的list,我们要保持原来list是存list的一致性,所以要应用到上方的map函数。

因此对于一个数组的转置,代码如下:

 

 
  1. # -*-coding:utf-8-*-

  2. array = [[1, 4], [2, 5], [3, 6]];

  3. print map(list, zip(*array));

运行结果如下:

你可能感兴趣的文章
Scrum与XP极限编程的不同之处
查看>>
麦肯锡工作方法学习笔记(一)
查看>>
麦肯锡工作方法学习笔记(二)
查看>>
如何有效的进行测试用例评审
查看>>
如何设计测试用例
查看>>
敏捷测试工程师的十条法则
查看>>
制定目标时需要遵循的SMART原则
查看>>
管理学中的 Expectancy Theory - 期望理论
查看>>
Scrum联盟发布了2015年Scrum状态报告
查看>>
英语学习: The Greatest Law of Life
查看>>
约瑟夫问题求解
查看>>
数据结构概念解析之—时间复杂度
查看>>
数据结构概念解析之—空间复杂度
查看>>
常见的管理学中的原理
查看>>
管理学中的“金字塔上升”现象
查看>>
PMP笔记:解析质量管理的8项基本原则
查看>>
PMP笔记:质量管理的七个工具
查看>>
CEO的苦恼:内部提拔还是“空降”?
查看>>
悦读:重新定义公司-谷歌是如何运营的
查看>>
研发主管的烦恼:周一早晨延迟的项目会议
查看>>