如何返回逗号之间的数字?

Question

我有一个文本文件，其中包含一个素数列表，但是，它们在每行的两个逗号之间，如下所示:

1, 2, 2
2, 3, 1
3, 5, 2
4, 7, 2
...

如何编写代码返回第n个素数，或者等于第n行两个逗号之间的素数?

Answer 1

def n_prime(n):
with open('file.txt','r') as f:
for i, line in enumerate(f):
if i == n:
return line.split(',')[1].strip()
print(n_prime(0)) # 2
print(n_prime(1)) # 3
print(n_prime(2)) # 5
print(n_prime(3)) # 7

输入:file.txt

1, 2, 2
2, 3, 1
3, 5, 2
4, 7, 2