帮我用Python写个记账小程序

好的，下面是一个简单的记账小程序的代码示例：

import csv
import datetime

def add_expense(amount, category):
    with open('expenses.csv', mode='a', newline='') as file:
        writer = csv.writer(file)
        date = datetime.datetime.now().strftime('%Y-%m-%d %H:%M:%S')
        writer.writerow([date, amount, category])

def view_expenses():
    with open('expenses.csv', mode='r') as file:
        reader = csv.reader(file)
        for row in reader:
            print(row)

while True:
    print("1. Add Expense")
    print("2. View Expenses")
    print("3. Quit")
    choice = int(input("Enter your choice: "))

    if choice == 1:
        amount = float(input("Enter amount: "))
        category = input("Enter category: ")
        add_expense(amount, category)
    elif choice == 2:
        view_expenses()
    elif choice == 3:
        break
    else:
        print("Invalid choice. Try again.")

这个程序使用了Python内置的csv模块来读写CSV文件，并且使用了datetime模块来获取当前日期和时间。用户可以选择添加新的支出或查看已有的支出。支出数据以日期、金额和类别的形式存储在一个CSV文件中。