-
在使用Python编写一个简单的食谱输入和输出程序时,遇到了一个问题:无法将食谱中的配料按行和列正确写入CSV文件中。目标是将列表中的每个元素写入一个逗号分隔的文件中,但实际结果却不是想要的样子。
-
解决方案: 为了解决这个问题,需要使用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()