用 Python 写了一个能生成10万数据的餐饮评分数据生成器
时间: 2024-02-11 14:07:24 浏览: 97
好的,下面是一个能够生成10万条餐饮评分数据的 Python 代码示例:
```python
import random
import csv
restaurant_names = ["麦当劳", "肯德基", "必胜客", "星巴克"]
review_contents = ["味道好", "服务态度好", "价格合理", "环境优美"]
num_ratings = 100000
min_rating = 0
max_rating = 5
with open("ratings.csv", "w", newline="") as csvfile:
writer = csv.writer(csvfile)
writer.writerow(["restaurant_name", "rating", "review_content"])
for i in range(num_ratings):
restaurant_name = random.choice(restaurant_names)
rating = round(random.uniform(min_rating, max_rating), 1)
review_content = random.choice(review_contents)
writer.writerow([restaurant_name, rating, review_content])
```
与之前的示例相比,我们主要做了两个修改:
1. `num_ratings` 变量的值从 1000 修改为 100000,以生成更多的评分数据。
2. 在生成评分数据时,我们使用了 `round` 函数对评分值进行了一定的精度控制,以保证评分值只有一位小数。
这样,我们就可以使用这个程序生成10万条餐饮评分数据,并将其保存到 CSV 文件中。
阅读全文