使用Python计算矩阵中的单元数

Question

我有一个包含矩阵的文件，第一行显示第3行，第5列。在

    3 5 
    TTTTT
    TMMMT
    TTTTT

我正在编写一个python程序来划分列并求出单元格的数量

^{pr2}$

第二个

    TT
    MM
    TT
    total number= 6

最后一个呢

    T
    T
    T

    total number = 3

到目前为止我掌握的密码

 rows = []
with open('example.in') as fp: #the file name is example.in
    for line in fp:
     rows.append(line.strip()) #removes white spaces 

    for i , rowi in enumerate(rows[1:]): #slices out the first row
      print(len(rowi))

输出

    5
    5
    5

它是将每行的单元格数相加。 我是python编程新手。在

Answer 1

也许像这样？在

def count_cells_row(file_in):
    """
    generator to count cells per row

    file_in   fileobject containing matrix
    """
    for row in file_in:
        row = row.strip()
        if row:
            yield len(row)

with open('/path/to/file') as file_in:
    ncells = sum([i for i in count_cells_row(file_in)])

print (ncells)

所以我首先制作一个生成器，它生成每次迭代的行的长度。 然后我通过列表理解列出这些值（[I for I in count_cells_row（file_in）]，这是另一种循环方法。 结果将出现在你的案例中[5，5，5]。在

我使用python函数sum来计算这些值的总和。在