python保存文件为txt_如何在Python- .txt文件中保存列表并访问列表中的值

我想创建一个模块来检查数字是否为素数,并列出所有素数.

如果必须在每次创建素数时生成素数列表,模块中的函数将花费一些时间.

如何打开与程序所在文件夹相同的.txt文件,然后让它从中访问数据并根据该文件创建列表？该函数完成后,.txt文件将保存在列表中.我还需要一些关于如何在给定位置实际保存文件的指导.

任何建议或提示都会很棒.谢谢！

解决方法:

我会挑选一个set对象,你可以这样做：

import cPickle as pickle

primes_set = set([2,3,5])

pickle.dump(primes_set, open('primes_set.pkl', 'wb'))

my_primes = pickle.load(open('primes_set.pkl', 'rb')) # to load the object

创建pickle文件后,您可以像这样使用它：

def get_primes(my_nums):

'''

Return only prime numbers from inputted list.

'''

my_primes = pickle.load(open('primes_set.pkl', 'rb'))

return [num for num in my_nums if num in my_primes]

标签：python,list,text

来源： https://codeday.me/bug/20190723/1511240.html