代码
import argparse
import ast
import json
def count_word_num_txt(txt_path, word):
fishing_count = 0
with open(txt_path, 'r') as file:
for line in file:
sentences = ast.literal_eval(line)
sections = str(sentences).split("[")
for section in sections:
if "'{}'".format(word) in section:
fishing_count += 1
return fishing_count
def count_word_num_json(json_path, word):
count = 0
with open(json_path, 'r') as file:
data = json.load(file)
for image_path, detections in data.items():
for detection in detections:
detection_type = detection['type']
if detection_type == word:
count += 1
return count
if __name__ == '__main__':
parser = argparse.ArgumentParser(description="Count the occurrences of a word in a text or JSON file.")
# "../result/fish/ori_data.txt" "../result/fish/fish_result.json"
parser.add_argument("--file", type=str, default="../result/fish/ori_data.txt", help="Path to the input file (txt or json).")
parser.add_argument("--word", type=str,default="fishing", help="Target word to count.")
parser.add_argument("--format", type=str, choices=["txt", "json"], default="txt", help="File format (txt or json).")
args = parser.parse_args()
if args.format == "txt":
count_function = count_word_num_txt
elif args.format == "json":
count_function = count_word_num_json
else:
print("Invalid file format. Please choose 'txt' or 'json'.")
exit(1)
if args.format == "txt":
counts = count_function(args.file, args.word)
elif args.format == "json":
counts = count_function(args.file, args.word)
print("{}文件: 出现 '{}' 的次数:".format(args.file, args.word), counts)
逻辑
- 导入了argparse、ast和json模块，用于处理命令行参数、将字符串转换为Python对象以及处理JSON文件。
- 定义了两个函数，count_word_num_txt和count_word_num_json，用于分别统计文本文件和JSON文件中某个目标单词的出现次数。
- 在count_word_num_txt函数中，它接受文本文件路径(txt_path)和目标单词(word)作为参数，然后初始化一个计数器fishing_count为0。它打开文本文件，逐行读取文件内容，将每一行使用ast.literal_eval转换为Python对象，然后将对象转换为字符串，并通过split(“[”)来分割成多个部分。接着，它遍历每个部分，检查目标单词是否出现在部分中（在单引号内），如果是，就将计数器增加1，最后返回计数器的值。
- 在count_word_num_json函数中，它接受JSON文件路径(json_path)和目标单词(word)作为参数，然后初始化一个计数器count为0。它打开JSON文件，使用json.load将JSON数据加载为Python对象，然后遍历数据，逐个检查每个图片的检测内容中是否包含目标单词，如果包含，就将计数器增加1，最后返回计数器的值。
- 在if name == ‘main’:块中，它使用argparse来解析命令行参数，包括输入文件路径(–file)、目标单词(–word)和文件格式(–format)。默认的文件路径是"…/result/fish/ori_data.txt"，默认的目标单词是"fishing"，默认的文件格式是"txt"。
- 根据文件格式的选择，确定要使用的统计函数（count_function）是count_word_num_txt还是count_word_num_json。
- 最后，根据所选择的文件格式和指定的文件路径和目标单词，调用相应的统计函数来统计目标单词的出现次数，并打印出结果。
这个代码允许用户通过命令行参数来指定要统计的文件、目标单词以及文件格式，然后执行相应的统计操作，并打印出结果。
结果
