import os import pandas as pd from hotel_main import hotel_main

folder_path = 'G:/python/booking情感分析/hotels_comments_txt/' attributes = ['员工素质', '设施服务', '清洁程度', '舒适度', '性价比', '位置']

获取所有酒店评论文件名

filenames = os.listdir(folder_path)

创建空表格

df = pd.DataFrame(columns=['酒店名称']+attributes)

遍历每个酒店评论文件

for filename in filenames: # 获取酒店名称 hotel_name = filename.split('.')[0]

# 获取酒店评论属性值
hotel_attrs_values = hotel_main(folder_path+filename)

# 如果酒店评论属性值为"无"，则将属性值都设置为"无"
if hotel_attrs_values == "无":
    row_values = [hotel_name] + ["无"]*len(attributes)
# 如果酒店评论属性值不为"无"，则将属性值写入表格
else:
    row_values = [hotel_name] + [hotel_attrs_values.get(attr, "无") for attr in attributes]

# 将一行数据添加到表格中
df.loc[len(df)] = row_values

将表格保存为csv文件

df.to_csv('hotel_attrs_values.csv', index=False