python - 如何返回逗号之间的数字？

Question

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

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

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

标签： pythonlistsearch

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