如果想要将生成的随机财富数据保留小数点后两位,可以在写入文件之前使用格式化字符串的方法来控制小数点的位数。以下是修改后的代码示例:
import random
# 生成随机财富数据
total_people = 100
average_wealth = 200 * 10**8 # 平均财富 200亿
# 5个人的财富超过5亿
high_wealth_individuals = [round(random.uniform(5, 10) * 10**8, 2) for _ in range(5)]
# 75个人的财富不超过1万元
low_wealth_individuals = [round(random.uniform(0, 10000), 2) for _ in range(75)]
# 合并数据
wealth_data = high_wealth_individuals + low_wealth_individuals
# 创建并保存为文本文件
with open('/storage/emulated/0/文件/代码print相关/结构图/wealth_data.txt', 'w') as file:
for wealth in wealth_data:
file.write(f'{wealth:.2f}\n') # 保留两位小数
print('随机财富数据已生成并保存为文本文件。')
在上述修改后的代码中,我在生成财富数据时使用了 round()
函数来保留小数点后两位。然后,将格式化字符串 {wealth:.2f}
用于写入文件,以确保数据保留两位小数。
现在,生成的财富数据将以两位小数的形式保存到文本文件中。如果你不想设置只保存2位小数数据,请看: