将列表中的值按行和列写入CSV文件

37 阅读2分钟
  1. 在使用Python编写一个简单的食谱输入和输出程序时,遇到了一个问题:无法将食谱中的配料按行和列正确写入CSV文件中。目标是将列表中的每个元素写入一个逗号分隔的文件中,但实际结果却不是想要的样子。

  2. 解决方案: 为了解决这个问题,需要使用csv.writer()函数的writerow()方法,而不是write()方法。writerow()方法可以将列表中的值按行写入CSV文件中,这样就能实现按行和列正确写入CSV文件。

# 使用csv.writer()函数的writerow()方法
with open('recipes/' + recipeName + '.csv', 'w', newline='') as csvfile:
    recipewriter = csv.writer(csvfile)
    recipewriter.writerow([ingredientList[0]])
    recipewriter.writerow([ingredientList[1]])
    recipewriter.writerow([ingredientList[2]])

在此示例中,ingredientList是一个包含三个元素的列表,使用writerow()方法将它们写入CSV文件中,并且添加了一个newline=''参数,以便在不同的操作系统中正确处理换行符。

通过使用writerow()方法,可以将列表中的值按行和列正确写入CSV文件中,从而达到预期的效果。

# Python 3.3.3

import sys #Allows use of the 'exit()' function
import csv #Allows use of the CSV File API

def mainMenu():
    print("########################################################")
    print("# Welcome to the recipe book, please select an option: #")
    print("# 1. Add new recipe                                    #")
    print("# 2. Lookup existing recipe                            #")
    print("# 3. Exit                                              #")
    print("########################################################")

    selectedOption = None
    inputtedOption = input()

    try:
        inputtedOption = int(inputtedOption)
    except ValueError:
        print("Invalid option entered")

    if inputtedOption == 1:
        selectedOption = inputtedOption
    elif inputtedOption == 2:
        selectedOption = inputtedOption
    elif inputtedOption == 3:
        print("Exiting...")
        sys.exit(0)

    return selectedOption

def validateInput(inputtedData):
    try: #Test if data is an integer greater than 1
        inputtedData = int(inputtedData)
        if int(inputtedData) < 1: #Recipes cannot contain less than 1 ingredient
            print("Sorry, invalid data entered.\n('%s' is not valid for this value - positive integers only)" % inputtedData)
            return False
        return int(inputtedData)
    except ValueError:
        print("Sorry, invalid data entered.\n('%s' is not valid for this value - whole integers only [ValueError])\n" % inputtedData)
        return False

def addRecipe():
    print("Welcome to recipe creator! The following questions will guide you through the recipe creation process.\nPlease enter the name of your recipe (e.g. 'Pizza'):")
    recipeName = input()
    print("Recipe Name: %s" % recipeName)
    print("Please enter the amount of people this recipe serves (e.g. '6'):")
    recipeServingAmount = input()
    if validateInput(recipeServingAmount) == False:
        return
    else:
        recipeServingAmount = validateInput(recipeServingAmount)
    print("Recipe serves: %s" % recipeServingAmount)
    print("Please enter the number of ingredients in this recipe (e.g. '10'):")
    recipeNumberOfIngredients = input()
    if validateInput(recipeNumberOfIngredients) == False:
        return
    else:
        recipeNumberOfIngredients = validateInput(recipeNumberOfIngredients)
    print("Recipe contains: %s different ingredients" % recipeNumberOfIngredients)
    ingredientList = {}
    i = 1
    while i <= recipeNumberOfIngredients:
        nthFormat = "st"
        if i == 2:
            nthFormat = "nd"
        elif i == 3:
            nthFormat = "rd"
        elif i >= 4:
            nthFormat = "th"
        ingredientNumber = str(i) + nthFormat
        print("Please enter the name of the %s ingredient:" % ingredientNumber)
        ingredientName = input()
        print("Please enter the quantity of the %s ingredient:" % ingredientNumber)
        ingredientQuantity = input()
        print("Please enter the measurement value for the %s ingredient (leave blank for no measurement - e.g. eggs):"  % ingredientNumber)
        ingredientMeasurement = input()
        print("%s ingredient: %s%s %s" % (ingredientNumber, ingredientQuantity, ingredientMeasurement, ingredientName))
        finalIngredient = [ingredientName, ingredientQuantity, ingredientMeasurement]
        ingredientList[i] = finalIngredient

        with open('recipes/' + recipeName + '.csv', 'w', newline='') as csvfile:
            recipewriter = csv.writer(csvfile)
            recipewriter.writerow([ingredientList[0]])
            recipewriter.writerow([ingredientList[1]])
            recipewriter.writerow([ingredientList[2]])

        i = i + 1

def lookupRecipe():
    pass  # To-do: add CSV reader and string formatter

#Main flow of program
while True:
    option = mainMenu()

    if option == 1:
        addRecipe()
    elif option == 2:
        lookupRecipe()