Python批量查询企业经营范围实现流程
为了帮助这位刚入行的小白实现Python批量查询企业经营范围,我们可以按照以下流程进行操作:
flowchart TD
A[准备数据] --> B[导入必要的库]
B --> C[读取待查询企业列表]
C --> D[循环查询企业经营范围]
D --> E[保存查询结果]
E --> F[结束]
下面展开每个步骤的详细说明。
1. 准备数据
在开始之前,我们需要准备两个关键的数据:
-
待查询企业列表:这是一个包含多个企业名字的文本文件,每行一个企业名字。可以在Excel或文本编辑器中创建并保存为文本文件(如
company_list.txt
)。 -
查询结果保存文件:这是一个用于保存查询结果的文本文件,每行一个企业名字和对应的经营范围。我们将在后续的步骤中逐个查询并将结果写入该文件(如
result.txt
)。
2. 导入必要的库
在Python中,我们可以使用requests
库发送HTTP请求,使用csv
库读写CSV文件。首先,我们需要安装这两个库:
# 安装requests库
pip install requests
# 安装csv库(Python自带库,无需安装)
安装完成后,我们需要在代码中导入这两个库:
import requests
import csv
3. 读取待查询企业列表
我们可以使用csv
库来读取待查询企业列表。以下是读取企业列表并存储到列表中的示例代码:
# 打开待查询企业列表文件
with open('company_list.txt', newline='', encoding='utf-8') as f:
reader = csv.reader(f)
company_list = list(reader)
4. 循环查询企业经营范围
接下来,我们需要循环遍历企业列表,并使用接口查询每个企业的经营范围。这里我们假设有一个名为get_company_scope
的自定义函数,用于发送查询请求并返回经营范围。
def get_company_scope(company_name):
# 构造查询请求的URL
url = f"
# 发送HTTP GET请求
response = requests.get(url)
# 解析并返回查询结果
if response.status_code == 200:
return response.json()['scope']
else:
return '查询失败'
在循环中,我们可以逐个查询企业的经营范围,并将结果存储到一个字典中。以下是循环查询的示例代码:
company_scope_dict = {} # 用于存储企业经营范围的字典
for company in company_list:
company_name = company[0]
scope = get_company_scope(company_name)
company_scope_dict[company_name] = scope
5. 保存查询结果
最后,我们需要将查询结果保存到文件中。这可以通过使用csv
库写入CSV文件来实现。
# 打开结果保存文件
with open('result.txt', 'w', newline='', encoding='utf-8') as f:
writer = csv.writer(f)
# 逐行写入查询结果
for company in company_list:
company_name = company[0]
scope = company_scope_dict.get(company_name, '查询失败')
writer.writerow([company_name, scope])
6. 完整代码
import requests
import csv
def get_company_scope(company_name):
# 构造查询请求的URL
url = f"
# 发送HTTP GET请求
response = requests.get(url)
# 解析并返回查询结果
if response.status_code == 200:
return response.json()['scope']
else:
return '查询失败'
# 打开待查询企业列表文件
with open('company_list.txt', newline='', encoding='utf-8') as f:
reader = csv.reader(f)
company_list = list(reader)
company_scope_dict = {} # 用于存储企业经