python 二维列表取某一列 或者计算每一列相加,由于不想数组那样可以直接取,所以麻烦一点

取某一列

方法1
rel_matrix = [[5,3,4,2,3],[1,5,2,2,3],[3,2,5,2,1],[1,4,2,5,2],[1,2,3,1,5]]
k = 2  # 你要取的那一列的下标
col = [row[k] for row in rel_matrix]
方法2
rel_matrix = [[5,3,4,2,3],[1,5,2,2,3],[3,2,5,2,1],[1,4,2,5,2],[1,2,3,1,5]]
k = 2  # 你要取的那一列的下标
col = list(list(zip(*rel_matrix))[k])    

        为啥要用两个list是这样的

zip() 函数用于将可迭代的对象作为参数,将对象中对应的元素打包成一个个元组,然后返回由这些元组组成的列表。
在 Python 3.x中为了减少内存,zip() 返回的是一个对象。如需展示列表,需手动 list() 转换。

        所以就清楚了,第一个list只是为了把zip()返回转换成一个列表,方便我们用下标访问,但是访问得到的是个元组,如果你想做的操作列表操作那就再用 list 转一下,当然,如果你要做的操作用元组也能返回一样的值,那就不用转了,就比如下面的求和,sum可以对元组求和,也可以对列表求和。所以就不用转

计算每一列相加

方法1
rel_matrix = [[5,3,4,2,3],[1,5,2,2,3],[3,2,5,2,1],[1,4,2,5,2],[1,2,3,1,5]]
n = len(rel_matrix)
res = []
for i in range(n):
    col = [row[i] for row in rel_matrix]
    res.append(sum(col))
number = res.index(max(res))   # 返回列求和的最大值对应的下标
方法2
rel_matrix = [[5,3,4,2,3],[1,5,2,2,3],[3,2,5,2,1],[1,4,2,5,2],[1,2,3,1,5]]
n = len(rel_matrix)
res = []
for col in zip(*rel_matrix):    # 二维列表取列
    res.append(sum(col))
number = res.index(max(res))    # 返回列求和的最大值对应的下标
评论 6
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包

打赏作者

中南自动化学院至渝

你的鼓励将是我创作的最大动力

¥1 ¥2 ¥4 ¥6 ¥10 ¥20
扫码支付:¥1
获取中
扫码支付

您的余额不足,请更换扫码支付或充值

打赏作者

实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值