博客
关于我
强烈建议你试试无所不能的chatGPT,快点击我
用python实现矩阵转置
阅读量:4451 次
发布时间:2019-06-07

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

前几天群里有同学提出了一个问题:手头现在有个列表,列表里面两个元素,比如[1, 2],之后不断的添加新的列表,往原来相应位置添加。例如添加[3, 4]使原列表扩充为[[1, 3], [2, 4]],再添加[5, 6]扩充为[[1, 3, 5], [2, 4, 6]]等等。

其实不动脑筋的话,用个二重循环很容易写出来:

def trans(m):    a = [[] for i in m[0]]    for i in m:        for j in range(len(i)):            a[j].append(i[j])    return am = [[1, 2], [3, 4], [5, 6]]    # 想象第一个列表是原始的,后面的是往里添加的print trans(m)    # result:[[1, 3, 5], [ 2, 4, 6]]

然而不管怎么看这种代码都很丑。

仔细看了一下m这种结构。等等,这不是字典的iteritems()的结果么?如果dict(m),那么结果——不就是keys()和values()么?

于是利用字典转换一下:

def trans(m):    d = dict(m)    return [d.keys(), d.values()]

可是再仔细想想,这里面有bug。如果添加列表的第一个元素相同,也就是转化之后dict的key相同,那肯定就不行了呀!况且,如果原始列表不是两个,而是多个,肯定不能用字典的呀!于是这种方法作罢,还是好好看看列表的形状。

然后又是一个不小心的发现:

 

这种转置矩阵的即时感是怎么回事?

没错,这个问题的本质就是求解转置矩阵。于是就简单了,还是用个不动脑筋的办法:

def trans(m):    for i in range(len(m)):        for j in range(i):            m[i][j], m[j][i] = m[j][i], m[i][j]    return mm = [[1, 2, 3], [4, 5, 6], [7, 8, 9]]print trans(m)

其实还是有点bug的,看起来是好用的,然而这个矩阵要求行列长度相同才行。

最后,群里某大神说:如果只是转置矩阵的话,直接zip就好了。这才想起来zip的本质就是这样的,取出列表中的对应位置的元素,组成新列表,正是这个题目要做的。

所以最终,这个题目(转置矩阵)的python解法就相当奇妙了:

def trans(m):    return zip(*d)

没错,就这么简单。python的魅力。

转载于:https://www.cnblogs.com/anpengapple/p/5427367.html

你可能感兴趣的文章
SharePoint文档库,如何在新窗口打开中的文件
查看>>
Xmind日常操作
查看>>
java: -source 1.6 中不支持 switch 中存在字符串
查看>>
Card Stacking 队列模拟
查看>>
抽象类和接口的关系与区别哦
查看>>
【C语言】C语言函数
查看>>
为什么用到混合支付?
查看>>
4.STL六大组件
查看>>
java学习之—栈
查看>>
1.5 重点
查看>>
子序列的按位或 Bitwise ORs of Subarrays
查看>>
IN语句改写EXISTS
查看>>
C#-WinForm-用户控件如何获取父级窗体
查看>>
STL_vector
查看>>
Dev中GridView——背景颜色改变
查看>>
socket编程2
查看>>
web开发中的MVC框架与django框架的MTV模式
查看>>
django添加导包路径
查看>>
java基础知识—变量、数据类型和运算符
查看>>
hadoop队列管理(指定queue跑程序)
查看>>