如何将列表保存到excel电子表格中

Question

我有一个有效的库存程序。我正试图保存一个名为库存的列表中的数据(商品名称和数量(。因此，它们可以在下次启动程序时重复使用，并在以后进行编辑。

我的库存程序代码：

import os
class Inventory:
def __init__(self):
self.item = []
self.qty = []
def remove(self, name):
ix = self.item.index(name)
self.item.pop(ix)
self.qty.pop(ix)
def add(self, name, qty):
self.item.append(name)
self.qty.append(qty)
def update(self, name, update):
if update >= 0:
self.qty[self.item.index(name)] += update
elif update <= -1:
self.qty[self.item.index(name)] += update
def search(self, name):
pos = self.item.index(name) if name in self.item else -1
if pos >= 0:
return self.item[pos], self.qty[pos]
else:
return None
def __str__(self):
out = ""
zipo = list(zip(self.item, self.qty))
for foobar in zipo:
out += f"Item : {foobar[0]} nQuantity : {foobar[1]}n"
out += "----------n"
return out

def menuDisplay():
"""Display the menu"""
print('=============================')
print('= Inventory Management Menu =')
print('=============================')
print('(1) Add New Item to Inventory')
print('(2) Remove Item from Inventory')
print('(3) Update Inventory')
print('(4) Search Item in Inventory')
print('(5) Print Inventory Report')
print('(99) Quit')

def add_one_item(inventory):
print('Adding Inventory')
print('================')
while True:
try:
new_name = input('Enter the name of the item: ')
assert new_name.isalpha(), "Only letters are allowed!"
new_qty = int(input("Enter the quantity of the item: "))
inventory.add(new_name, new_qty)
break
except Exception as e:
print("Invalid choice! try again! " + str(e))
print()

def remove_one_item(inventory):
print('Removing Inventory')
print('==================')
removing = input('Enter the item name to remove from inventory: ')
inventory.remove(removing)

def ask_exit_or_continue():
return int(input('Enter 98 to continue or 99 to exit: '))

def update_inventory(inventory):
print('Updating Inventory')
print('==================')
item = input('Enter the item to update: ')
update = int(input(
"Enter the updated quantity. Enter 5 for additional or -5 for less: "))
inventory.update(item, update)

def search_inventory(inventory):
print('Searching Inventory')
print('===================')
search = input('Enter the name of the item: ')
result = inventory.search(search)
if result is None:
print("Item not in inventory")
else:
name, qty = result
print('Item:     ', name)
print('Quantity: ', qty)
print('----------')

def print_inventory(inventory):
print('Current Inventory')
print('=================')
print(inventory)

def main():
inventory = Inventory()
while True:
try:
menuDisplay()
CHOICE = int(input("Enter choice: "))
if CHOICE in [1, 2, 3, 4, 5]:
if CHOICE == 1:
add_one_item(inventory)
elif CHOICE == 2:
remove_one_item(inventory)
elif CHOICE == 3:
update_inventory(inventory)
elif CHOICE == 4:
search_inventory(inventory)
elif CHOICE == 5:
print_inventory(inventory)
exit_choice = ask_exit_or_continue()
if exit_choice == 99:
exit()
elif CHOICE == 99:
exit()
except Exception as e:
print("Invalid choice! try again!"+str(e))
print()
# If the user pick an invalid choice,
# the program will come to here and
# then loop back.

main()

该程序名为Cleancopy.py，我正试图将数据保存到名为Inventory的TextEdit文件或excel文档中。

假设导出名为inventory的列表中的名称和数量的代码。

我该如何解决这个问题？这是我第一次尝试保存这些数据。谢谢你抽出时间。

Answer 1

保存和检索列表的最快方法是将它们保存到文本文件中，然后使用eval读取列表。对于库存，每次添加或删除项目时都会更新数据文件。如果首次实例化库存对象，则加载数据。

尝试此更新：

class Inventory:
def __init__(self):
.....
self.load()

def remove(self, name):
....
self.save()
def add(self, name, qty):
.....
self.save()
def update(self, name, update):
.....
self.save()
.....

def save(self):
with open('inventory.dat','w') as f:
f.write(str(self.item) + 'n' + str(self.qty))
def load(self):
from os import path
if path.exists('inventory.dat'):
with open('inventory.dat','r') as f:
lns = f.readlines()
self.item = eval(lns[0])
self.qty = eval(lns[1])

对于更多的数据，您应该使用csv文件。