使用input()选择文件在python中创建绘制excel数据的程序

import openpyxl

# 获取用户输入的文件名
filename = input("请输入要创建的Excel文件名：")

# 创建一个新的Excel文件
workbook = openpyxl.Workbook()

# 获取默认的工作表
sheet = workbook.active

# 写入数据到Excel文件
data = [
    ['姓名', '年龄', '性别'],
    ['张三', 20, '男'],
    ['李四', 25, '女'],
    ['王五', 30, '男']
]

for row in data:
    sheet.append(row)

# 保存Excel文件
workbook.save(filename)

print("Excel文件创建成功！")