python课堂实例_Python 练习实例38

可以输入任意阶方阵，并优化输入效果，输入时支持回车。

# encoding:utf-8

import re

stopword = ''

str = ''

num = []

result = 0

print '请输入一个n阶方阵:数字之间用空格隔开，回车输入下一行元素，双回车表示结束\n' \

'例如： 11 12 13\n' \

' 21 22 23\n' \

' 31 32 33\n'

for line in iter(raw_input, stopword):#输入方阵，将回车作为字符输入

str = str + line + '\n'

row = len(re.findall(r'\n', str))

for x in re.finditer(r'\d+', str):#将字符串的数字取出并存入到num数列

num.append(int(x.group()))

i = 0

for j in range(row):#将主对角线元素进行相加

result += num[i]

i = i + row + 1

print '主对角线元素和为:',result

chengxuyuan

chengxuyuan

hdw***taoyuan@foxmail.com2年前 (2017-08-18)