对一个文件中的指定信息进行过滤输出结果

来源:互联网 发布:网络教育 概念股 编辑:程序博客网 时间:2024/05/21 06:29

最近因为工作的需要一直在写脚本,今天用到了一个简单的小程序,因为指定目录下的文件中有一部分是无法使用的,在程序对该路径下面的数据进行处理的时候需要去除这些数据,因此尝试写了一个小脚本,如下:

# !/usr/bin/python
#-*-coding:utf-8-*-


import random
import os
import json


def get_random(p_filepath='p_filter.json', c_filepath='c_filter.json'):
'''
p_filter.json中为需要处理的文件,c_filter.json中是需要去除的文件,
程序的目的是过滤掉p_filter.json中的需要去除的文件
随机生成20个任务,每个任务由616个记录组成
'''

p_file = open(p_filepath)
c_file  = open(c_filepath)
p_filelist = json.load(p_file)
print len(p_filelist), p_filelist[0]
c_filelist = json.load(c_file)
print len(c_filelist), c_filelist[0], 
file_num = len(p_filelist)
os.mkdir('random_filter_task')
for i in range(20):
result_list = []
final_list = []
n = 1
while n <= 600:
lin = random.randint(0, int(file_num)-1)
result_list.append(p_filelist[lin])
n += 1
for line in c_filelist:
result_list.append(line)
fp = open('random_filter_task/test_task' + str(i) + '.json', 'w')
data = json.dumps(result_list)
fp.write(data)

get_random()

0 0