python缩进和二维列表的解包

太阳不下山 发表于 2021-6-24 09:59:42

这两天博主遇见了一道题目如下：
现在给你N个数（0<N<1000），现在要求你写出一个程序，找出这N个数中的所有素数，并求和。
　　输入

　　第一行给出整数M(0<M<10)代表多少组测试数据
每组测试数据第一行给你N，代表该组测试数据的数量。
接下来的N个数为要测试的数据，每个数小于1000
　　输出

　　每组测试数据结果占一行，输出给出的测试数据的所有素数和
　　输入
3
5
1 2 3 4 5
8
11 12 13 14 15 16 17 18
10
21 22 23 24 25 26 27 28 29 30
　　输出
10
41
52
　　博主在写的过程中是这么想的既然要素数求和那就要先求素数用一些代码求素数（1不是素数）
然后进行第二步规范输入输出
这里的多组输入我选择用map直接给列表a传参这个方法我是从学长那学到的。传参，筛选出素数后求和把最后结果当成一个列表存进大列表中。这里博主想用一个二维列表，这样就可以存储上多组数据并一起输出，最后用解包的办法把二维列表里的一维列表遍历出来。
（我对于解包了解不是太全面欢迎大神补充）
这是我第一次的代码
from math import sqrt
m = 0
allnum = []
M = int(input())
a = []# 进来的所有数
while M > 0:
M -= 1
N = int(input())
a = map(int, input().split())
b = []# 每一组求出的素数
for r in a:
if r == 1:

 continue
for i in range(2, int(sqrt(r)) + 1):
 if r % i == 0:
 break
else:
 b.append(r)allnum.append()for r in allnum: print(r)　　但是我在这里遇见了一个问题我在测试过程中先一组一组测试发现没有问题当我两组一起测试时发现二维列表里只有一组数据就是最后一组数据第一组数据被自动覆盖了我以为是存储这些数据的b列表位置的错误但我更改b列表以后还是无法达到题目格式输出。最后我做了如图调整
from math import sqrt
m = 0
allnum = []
M = int(input())
a = []# 进来的所有数
while M > 0:
M -= 1
N = int(input())
a = map(int, input().split())
b = []# 每一组求出的素数
for r in a:

 if r == 1:
 continue
 for i in range(2, int(sqrt(r)) + 1):
 if r % i == 0:
 break
 else:
 b.append(r)
allnum.append([sum(b)])for r in allnum: print(r)　　再测试时已经可以格式化输出了。
这里，在素数求和时我对代码进行了一个优化，引入了数学开平方根的方法
　　总结 :
1.之前只听python缩进严格这是我第一次亲身经历 python缩进不同结果不同这是第一次让我切身体会到python缩进的严格以后一定会多加注意
2.其实这里可以不用多组输出直接计算完毕后输出也可以我这么做有些多此一举了当时以为必须最后输入完毕才允许输出其实改成如下也可以通过
from math import sqrt
m = 0
allnum = []
M = int(input())
a = []# 进来的所有数
while M > 0:
M -= 1
N = int(input())
a = map(int, input().split())
b = []# 每一组求出的素数
for r in a:

 if r == 1:
 continue
 for i in range(2, int(sqrt(r)) + 1):
 if r % i == 0:
 break
 else:
 b.append(r)
sumnum=sum(b)
print(sumnum)　　
但是很尴尬优化代码的目的就是减少时间或者空间或者两者但是这样改了之后时间空间都增多了一丢丢不知道为什么希望有大神来解答一下。

页: [1]

CodeAE代码之家-专为程序员打造的技术家园！-网站地图

python缩进和二维列表的解包