python-3.x - 计算 0 和 1 列的高度以及全高的 Bool 值

Question

迫切需要帮助！我尝试过遍历二维数组并使用 row.count() 但到目前为止没有任何效果。

Python不确定如何开始我想找到1列的高度。输入 [[0,1,1,0],[1,1,1,0],[1,1,1,1]]

输出 [2,3,3,1]

制作一个函数，该函数采用如下例所示的二维数组并给出真或假值。确实只有 2 会占据一列的全长。[[2,1,1,0],[2,1,1,0],[2,1,1,0]

[ 真，假，假，假]

标签： python-3.x

Answer 1

你可以使用numpy：这个

import numpy as np
M=[[0,1,1,0],[1,1,1,0],[1,1,1,1]]
a=np.sum(M,axis=0)
print(a)----> output = [2 3 3 1]

a_list = a.tolist()
a_max = max(a_list)
print((a > a_max-1).tolist()) ---> [False,True,True,False]