根据用户输入更改现有CSV文件

时间:2018-03-18 16:54:46

标签: python python-3.x csv input

我会尽量保持这个尽可能简短,因为我试图找到答案,但我无法做到。我正在尝试创建一个简单的直到系统。我使用CSV文件存储产品的描述,价格和库存。如果用户以管理员身份登录,他们将能够更改特定产品的库存。我发现很难理解如何更换新库存。我忘了提到我使用列表来存储每个类别 我们说我有以下文件

apples,1,11
grape,1.2,1
chocolate,0.75,15
bananas,1.35,27

以下代码创建列表:

Products = []
Prices = []
Prices= []
import csv
with open('listlist.csv') as csvfile:
    readCSV = csv.reader(csvfile,delimiter=',')
    for row in readCSV:
        print(row)
        item = row[0]
        price = row[1]
        stock = row[2]
        Products.append(item)
        Prices.append(price)
        Stock.append(int(stock))

如果经理想要更改商品的库存' grape'从1到11,我该如何以最简单的方式处理这个问题?

2 个答案:

答案 0 :(得分:1)

由于这似乎是家庭作业或练习,这不是一个完整的答案,但应该让你开始。

使用词典列表读取文件以存储项目:

items_in_store = []
import csv
with open('listlist.csv') as csvfile:
    readCSV = csv.reader(csvfile,delimiter=',')
    for row in readCSV:
        item = dict(name=row[0], price=row[1], stock=row[2])
        items_in_store.append(item)

循环结果列表,更改特定目标项:

tgt_item = 'apple'
tgt_stock = 5
for item in items_in_store:
    if item['name'] == tgt_item:
         # we found our item, change it
         item['stock'] = tgt_stock

persist your changes到文件(注意这次我们以写模式打开文件):

with open('listlist.csv', 'w') as csvfile:
    writeCSV = csv.writer(csvfile, delimiter=',')
    for item in items_in_store:
        row = [item['name'], item['price'], item['stock']]
        writeCSV.writerow(row)

替代方法,再次读取文件,但这次存储在包含字典的字典中:

items_in_store = {} # create empty dictionary
import csv
with open('listlist.csv') as csvfile:
    readCSV = csv.reader(csvfile,delimiter=',')
    for row in readCSV:
        # use item name as key in first dictionary
        # and store a nested dictionary with price and stock
        items_in_store[row[0]] = dict(price=row[1], stock=row[2])

以这种方式存储我们的数据,我们不需要循环更改库存,我们可以立即使用其键访问所需的项目:

tgt_item = 'apple'
tgt_stock = 5
items_in_store[tgt_item]['stock'] = tgt_stock

在上述所有示例代码段中,您可以要求用户输入以填充tgt_itemtgt_stock

答案 1 :(得分:0)

你可以使用熊猫来做,你不需要处理不同的列表

       0         1       2
0   apples      1.00    11
1   grape       1.20    1
2   chocolate   0.75    15
3   bananas     1.35    27


import pandas
df =  pandas.read_csv(csv_file_path, header=None)
df.loc[df[0] == "apples", 2] = new_stock

如果将列名添加到文件中,可以通过列名更改[0]和2